LeetCode 92:翻转链表 2

 这道题乍一想觉得很简单,写完程序觉得自己的方法细节处理太繁琐了,改了两个比较隐蔽的Bug。对LeetCode测试样例可以通过,但是[3, 5],1,2这样的case就过不了,仔细推了一下,可能是prev没有在rear前面一个节点,而是和它同一个节点导致的。

class Solution {
public:
    ListNode* reverseBetween(ListNode* head, int m, int n) {
        if(!head || m>=n) return head;

        ListNode* rear = head, *prev = head; //Bug2: 没有在定义时初始化prev导致m=1时prev无法赋值,会访问随机的地址运行出错
        for(int i=1; i<m; i++) {
            prev = rear;
            rear = rear->next;
        }

        ListNode* tail = rear;
        for(int i=m; i<n; i++) {
            tail = tail->next;
        }
        ListNode* behind = tail->next; //prev, behind保存了连接待翻转序列的两个节点的指针
        //将rear-->tail闭区间节点反转链表,再接回原序列即可
        ListNode* p1 = nullptr, *p2 = rear, *p3 = p2;
        while(p2 != behind) { //花了很大力气找到的Bug: (tail->next)换成behind就不会死循环了,因为tail指向的节点的next已经翻转过了!!!所以会造成这段序列反复翻转?去掉 p2 && 应该也没有问题
            p3 = p2->next;
            p2->next = p1;
            p1 = p2;
            p2 = p3;
        }
        //翻转结束的这段链表再接回去
        prev->next = tail;
        rear->next = behind;

        return head;
    }
};

 为了在有些case中,解决prev不在rear前面的问题,引入了dummy节点,返回dummy节点的next而不是head,改了这两个地方就可以通过所有case

        ListNode dummy(0);
        dummy.next = head;

        return dummy.next;
发布了97 篇原创文章 · 获赞 11 · 访问量 2476

猜你喜欢

转载自blog.csdn.net/chengda321/article/details/104126742