Reverse Nodes in k-Group

Reverse Nodes in k-Group

Given a linked list, reverse the nodes of a linked list k at a time and return its modified list.

If the number of nodes is not a multiple of k then left-out nodes in the end should remain as it is.

You may not alter the values in the nodes, only nodes itself may be changed. Only constant memory is allowed.

样例

Given this linked list: 1->2->3->4->5

For k = 2, you should return: 2->1->4->3->5

For k = 3, you should return: 3->2->1->4->5

 

 

嗯。。最近在刷lintcode上的题。然后做到这道题，难度标的是困难但是挺简单的嘛。想到好久没发随笔了就发一个把。

用C++写的，代码如下喽。思路就是一个指针先往前k个然后将两个指针之间的用头插法倒置就好。因为没有说有头结点，所以p == head时要特殊处理。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    /**
     * @param head a ListNode
     * @param k an integer
     * @return a ListNode
     */
    ListNode *reverseKGroup(ListNode *head, int k) {
        // Write your code here
        if(k <= 1 || head == NULL) return head;
        ListNode *p, *p1, *pre, *pre1, *t;
        p = head;
        p1 = head;
        while(p != NULL) {
            int flag = 0;
            for(int i=0;i<k;i++) {
                p1 = p1->next;
                if(p1 == NULL && i < k-1) {
                    flag = 1;
                    break;
                }
            }
            if(flag) break;
            
            pre1 = p;
            if(p == head) {
                head = p1;
                while(p != p1) {
                    t = head;
                    head = p;
                    p = p->next;
                    head->next = t;
                }
            }else {
                pre->next = p1;
                while(p != p1){
                    t = pre->next;
                    pre->next = p;
                    p = p->next;
                    pre->next->next = t;
                }
            }
            pre = pre1;
        }
        return head;
    }
};