这是一道来自Leetcode的题目,如下所示。其中,K可以大于链表的长度,如果大于链表的长度,则旋转的大小就是 size % K。
思路:
1)遍历链表,求出链表长度size,并得到指向链表末尾结点的指针end;
2)设置两个指针pHead和pEnd,一前一后。根据K的大小,pEnd往前走K+1步。之后,两个指针同步移动,直到pEnd==NULL;
3)此时,pHead的下一个结点就是新链表的开头。例如,1->2->3->4->5->6,K=3。此时,pHead指向的是3,根据题目意思是要将4、5、6移到链表的前端。我们用result指针先保存好pHead->next,这就是要返回的结果,然后把末尾结点的next指向开头就好了;
4)最后,需要讲pHead->next置空,消除链表的内部环。
/**
* 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 (k <= 0 || !head)
return head;
ListNode* pHead = head;
ListNode* pEnd = head;
ListNode* end = NULL;
int size = 0;
while (pHead) { //记录最后一个节点和计算链表长度
size++;
end = pHead;
pHead = pHead->next;
}
int mod = k % size;
pHead = head; // pHead回到开始位置
for (int i=0; i<=mod; i++)
pEnd = pEnd->next;
while (pEnd) {
pEnd = pEnd->next;
pHead = pHead->next;
}
end->next = head; // end接到链表头,pHead下一个就是新链表的开始
ListNode* result = pHead->next; // 然后再把pHead->next置空断环
pHead->next = NULL;
return result;
}
};