Leetcode 25 Reverse Nodes in k-Group

这道题和上面两道相比,是反转链表中更加普通的情况。

解决这道题需要两次循环,第一次求得长度,通过长度分段进行反转。第二次循环分段进行反转,分段反转就是Reverse Linked List II的情况了。

在第二次循环中有两个循环,一个是外层循环,一个是内层循环。内层的逻辑和之前的代码完全一致。外层循环的任务是改变prev(prev = cur)和循环变量。

具体代码如下:

class Solution {
public:
      ListNode* reverseKGroup(ListNode* head, int k) {
          if (head == NULL || head->next == NULL || k == 1) {
             return head;
          }
          
          ListNode* dummyHead = new ListNode(0);
          dummyHead->next = head;
          
          ListNode* prev = dummyHead;
          ListNode* cur = prev->next;
          ListNode* nex = cur->next;
          
          int len = 0;
          
          while(cur) {
            ++len;
            cur = cur->next;  
          }
          
          while(len >= k) {
              cur = prev->next;
              nex = cur->next;
              
              for(int i=0; i<k-1; ++i) {
                  cur->next = nex->next;
                  nex->next = prev->next;
                  prev->next = nex;
                  nex = cur->next;
              }
              
              prev = cur;
              len -= k;
          }
          
          ListNode* retHead = dummyHead->next;
          delete dummyHead;
          return retHead;
          
      } 
};

 

posted @ 2018-09-04 21:09  ly-bnu  阅读(83)  评论(0编辑  收藏  举报