[经典面试题]k节点一组旋转链表

来源：互联网发布：u盘安装ubuntu工具包编辑：程序博客网时间：2024/05/01 05:35

【题目】

给出一个链表和一个数k，比如链表1→2→3→4→5→6，k=2，则翻转后2→1→4→3→6→5，若k=3,翻转后3→2→1→6→5→4，若k=4，翻转后4→3→2→1→5→6。

如果节点的数量是不k的倍数则最终留出节点应该保持原样，每K个一反转，不到k个不用反转。用程序实现。

------美团校招

来自LeetCode ：Reverse Nodes in k-Group

【代码】

#include <iostream>#include <list>using namespace std;struct ListNode{    int val;    ListNode *next;    ListNode(int x):val(x),next(NULL){}};// head不带头结点// k 旋转个数// 最后不满k个不用旋转ListNode *ReverseKGroup(ListNode *head, int k) {    // 容错处理    if(head == NULL || k < 2){        return head;    }    //添加头结点    ListNode *dummy = new ListNode(0);    dummy->next = head;    ListNode *pre,*cur,*tail;    pre = dummy;    // 分组旋转的第一个节点即旋转后的尾节点    tail = head;    // 当前节点    cur = head;    int count = 0;    // 统计节点个数    while(cur != NULL){        cur = cur->next;        count++;    }    // 旋转次数    int rCount = count / k;    // 分组旋转下标    int index = 0;    // 旋转    while(rCount){        // 分组旋转        // k节点只需旋转k-1节点        index = k-1;        while(index){            //先删除            cur = tail->next;            tail->next = cur->next;            //再插入            cur->next = pre->next;            pre->next = cur;            index--;        }//while        pre = tail;        tail = tail->next;        rCount--;    }//while    return dummy->next;}int main(){    int A[] = {1,2,3,4,5};    ListNode *head = new ListNode(0);    head->next = NULL;    ListNode *node;    ListNode *pre = head;    for(int i = 0;i < 5;i++){        node = new ListNode(A[i]);        node->next = NULL;        pre->next = node;        pre = node;    }    head = ReverseKGroup(head->next,3);    while(head != NULL){        cout<<head->val<<" ";        head = head->next;    }    cout<<endl;    return 0;}

0 0