LeetCodeGo/0092.ReverseLinkedListII.go at master · hanxuema/LeetCodeGo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package main


 func reverseBetween(head *ListNode, m int, n int) *ListNode {
    if head == nil {
        return nil
    }
    var pre *ListNode
    cur := head

    for m > 1{
        pre = cur
        cur = cur.Next
        m--
        n--
    }
    p2 := pre
    c2 := cur
    for n > 0 {
            next := cur.Next
            cur.Next = pre
            pre = cur
            cur = next
        n--
    }
    if p2 == nil {
        head = pre
    }else{
        p2.Next = pre
    }

    c2.Next = cur
    return head
}

// my implementation
// func reverseBetween(head *ListNode, m int, n int) *ListNode {
//     if head == nil || m == n {
//         return head
//     }

//     length := n - m
//     var tail *ListNode
//     if m > 1 {
//         tail = &ListNode{-1,head}
//     }
//     cur := head

//     for m > 1 {
//         tail = tail.Next
//         cur = cur.Next
//         m--
//     }
//     newCur := cur
//     var pre *ListNode
//     next := cur.Next
//     for length >= 0 {
//         next = cur.Next
//         cur.Next = pre
//         pre = cur
//         cur = next
//         length--
//     }
//     // next := cur.Next
//     if tail != nil {
//         tail.Next = pre
//         newCur.Next = next
//     } else {
//         head = pre
//         newCur.Next = next
//     }
//     return head
// }