LeetCode 61. 旋转链表
给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
示例 1:
输入: 1->2->3->4->5->NULL, k = 2
输出: 4->5->1->2->3->NULL
解释:
向右旋转 1 步: 5->1->2->3->4->NULL
向右旋转 2 步: 4->5->1->2->3->NULL
示例 2:
输入: 0->1->2->NULL, k = 4
输出: 2->0->1->NULL
解释:
向右旋转 1 步: 2->0->1->NULL
向右旋转 2 步: 1->2->0->NULL
向右旋转 3 步: 0->1->2->NULL
向右旋转 4 步: 2->0->1->NULL
算法:此题与删除倒数第K个结点类似,我们需要注意的是K有可能大于链表表长,所以我们要做取模处理。
/** * Definition for singly-linked list. * struct ListNode { * int val; * ListNode *next; * ListNode(int x) : val(x), next(NULL) {} * }; */ class Solution { public: ListNode* rotateRight(ListNode* head, int k) { if(!head)return NULL; auto l=new ListNode(-1); l->next=head; int n=0; auto r=head; while(r)n++,r=r->next; k%=n; int cnt=0; auto p=l,q=l; while(p->next){ if(cnt<k)cnt++; else q=q->next; p=p->next; } p->next=l->next; l->next=q->next; q->next=NULL; return l->next; } };