这道题乍一想觉得很简单,写完程序觉得自己的方法细节处理太繁琐了,改了两个比较隐蔽的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;