根据题目我们首先想到的是如果k的值大于链表的长度时,这时应该用取模来解决。再进一步,我们需要将链表的头指针向右移动k个位置,由于需要移动整个数组,所以我们可以考虑将其闭环,将尾指针指向头结点,这样只需将尾指针再移动n - k个位置,再将其下一个next结点作为头结点,然后将其指向空结点,这样就完成了运动。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if(head == nullptr || head -> next == nullptr || k == 0)return head;
ListNode* newhead = head;
int n = 1;
while(newhead -> next != nullptr){
newhead = newhead -> next;
n++;
}
k = k % n;//取模
if(k == 0)return head;
newhead -> next = head;//头尾相连
for(int i = 0;i < n - k;i++){
newhead = newhead -> next;//遍历到第n - k个结点
}
ListNode* anshead = newhead -> next;//将下一个结点作为头结点
newhead -> next = nullptr;//指向空
return anshead;
}
};
当然还有一种快慢指针的解法,前面的步骤和上面的一样,后面我们用快指针先移动k个位置,再和慢指针一起移动到尾部,这时慢指针指向的就是第n - k个位置,将其下个结点作为头结点,快指针指向head头结点,再将慢指针指向空,即完成了移动。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if(head == nullptr || head -> next == nullptr || k == 0)return head;
ListNode* cur = head;
int len = 0;
while(cur != nullptr){
cur = cur -> next;
len++;
}
k = k % len;
if(k == 0)return head;
ListNode* fast = head;//快指针
while(k > 0){
fast = fast -> next;
k--;
}
ListNode* slow = head;//慢指针
while(fast -> next != nullptr){
fast = fast -> next;
slow = slow -> next;
}
ListNode* newHead = slow -> next;
slow -> next = nullptr;
fast -> next = head;
return newHead;
}
};