首页 > 其他 > 详细

Leetcode: Reverse Nodes in k-Group

时间:2014-12-14 00:44:24      阅读:294      评论:0      收藏:0      [点我收藏+]

链表的倒转,

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
   ListNode* reverseKGroup(ListNode* head, int k){
    if(!head || !head->next || k <= 1)
        return head;
    ListNode* pFirst = head;
    ListNode* pBefore = NULL;
    ListNode* pKth = pFirst;
    while(1) {
        int cnt = k -1;
        while(cnt && pKth){
            pKth = pKth->next;
            cnt--;
        }
        if(cnt > 0 || pKth == NULL) break;
        reverse(pBefore,pFirst,pKth);
        if(pBefore == NULL)
            head = pFirst;
        pBefore = pKth;
        pFirst = pKth->next;
        pKth = pFirst;
    }
    return head;
   }
   void reverse(ListNode* pBefore, ListNode*& pFirst, ListNode*& pKth){
    ListNode* p = pFirst;
    while(p != pKth) {
        ListNode* tmp = p->next;
        p->next = pKth->next;
        pKth->next = p;
        p = tmp;
    }
    ListNode* tmp = pFirst;
    pFirst = pKth;
    pKth = tmp;
      if(pBefore)
        pBefore->next = pFirst;
   }
};


Leetcode: Reverse Nodes in k-Group

原文:http://blog.csdn.net/sina012345/article/details/41919427

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!