leetcode 92 反转链表区间

题目：

给定一个链表和需要反转的区间上下界，用one-pass完成反转操作。

 

思路：

1. 定位 2. 反转 3.链接

 

class Solution {
    public ListNode reverseBetween(ListNode head, int m, int n) {
        ListNode pre = null;
        ListNode cur = head;
        
        //定位
        int start = 1;
        while (cur != null && start < m) {
            pre = cur;
            cur = cur.next;
            start++;
        }
        
        //反转
        ListNode pre2 = cur;
        ListNode cur2 = null;
        if (cur != null) {
            cur2 = cur.next;
            start++;
        }
        ListNode tmp = null;
        while (cur2 != null && start <= n) {
            tmp = cur2;
            cur2 = cur2.next;
            tmp.next = pre2;
            pre2 = tmp;
            start++;
        }
        
        //链接
        if (pre != null) {
            pre.next = pre2;  
        }
        if (cur != null) {
            cur.next = cur2;
        }
        
        return pre == null ?  pre2 : head;
    }
}

 

实现上尽管已经将能做的做到极致了，各种判断null和记录next node。但倒在了最后的return上。

第一次是忘了写return（又犯了老毛病），加了 return head; 后测试发现还是有问题。问题在于如果区间m是从1开始的，则新的链表头并不在head处，而是在pre2处。

 

然后逛讨论区的时候发现了一个巨强的写法，甘拜下风。