1 struct ListNode { 2 int val; 3 ListNode *next; 4 ListNode(int x) : val(x), next(NULL) {} 5 }; 6 7 class Solution { 8 public: 9 ListNode* reverseBetween(ListNode* head, int m, int n) { 10 int change_len = n - m + 1; 11 ListNode *pre_head = NULL;//须要逆置的头节点的前驱 12 ListNode *result = head; 13 14 while (head && --m) {//m-1 次 15 pre_head = head;//记录当前头节点 16 head = head->next; 17 }//head >>须要逆置的头节点 18 ListNode *modify_list_tail = head;//记录初始逆置的头节点,也就是逆置后的尾部 19 ListNode *new_head = NULL;//逆置后的头节点 20 while (head && change_len) { 21 ListNode *next = head->next; 22 head->next = new_head; 23 new_head = head; 24 head = next;//最后一次 head 指向逆置段的后一个节点 25 --change_len;//只逆置 change_len 个节点 26 } 27 modify_list_tail->next = head;//逆置段的尾部指向下一个节点 28 if (pre_head) { 29 pre_head->next = new_head;//逆置段的前一个节点指向 30 } else { 31 // pre_head = new_head; ❌ 32 // 特殊状况: m=1 从第一个节点开始逆置,result存放的就是头节点,因此 result 须要变成逆置后的头节点 33 result = new_head; 34 } 35 return result; 36 } 37 };
1 int main(int argc, const char * argv[]) { 2 ListNode a(1); 3 ListNode b(2); 4 ListNode c(3); 5 ListNode d(4); 6 ListNode e(5); 7 a.next = &b; 8 b.next = &c; 9 c.next = &d; 10 d.next = &e; 11 Solution solve; 12 ListNode *head = solve.reverseBetween(&a, 2, 4); 13 while (head) { 14 cout <<head->val<<endl; 15 head = head->next; 16 } 17 18 return 0; 19 }