LeetCode面试题Day18|LC61 旋转链表
题目1:
指路:
. - 力扣(LeetCode)61 旋转链表
思路与分析:
如果我没记错的话这个题应该在一次周赛上出现过,但我做题的网站太杂忘记哪一次了。对于旋转链表我的第一思路是将其做成环形链表,返回原链表中最右边的节点,剩下的依次返回,那么第一次旋转也就是原链表中倒数第一个节点旋转后的位置应该是新链表的第一个,原链表中倒数第二个节点旋转后的位置应该是倒数第二个……以此类推我们得到旋转k次以后,原链表中的首节点最后旋转完成后的位置应该是正数第k+1个,而原链表中的尾节点最后旋转完成的位置应该是正数第k个……再看样例,当n=5,k=2时从第4个节点开始返回,而n=3,k=1从第三个节点开始返回(大家也可以再写几个案例),得到一个规律那就是原链表完成k次旋转后,首节点应该是原链表的第n-k+1个位置。在这里需要注意的一点是并不是完全旋转k次,当k>n时,旋转n次即为原链表,因此我们可以直接旋转k%n次,结合上面的推导我们可以直接用一个首节点指向原链表中n-k+1的位置,最后修改新链表的首尾节点指向。
代码:
/**
* 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;
int cnt = 1;
ListNode* dummyhead = head; // dummyhead = ori_tail
while (dummyhead->next != nullptr) { // 保证不是最后一个,即向后遍历有意义
cnt += 1;
dummyhead= dummyhead->next;
} // 遍历全部节点找到全部节点个数
k %= cnt; // 找到最少遍历的次数
if ((k %= cnt )== 0) return head;
// 怎么让尾节点的下一个节点指向首节点?
// 直接做第二种解法
ListNode* new_tail = head;
for (int i = 0; i < cnt - k - 1; i++) {
new_tail = new_tail->next;
}
ListNode* new_head = new_tail->next;
// 修改链表的头尾节点指向
dummyhead->next = head;
new_tail->next = nullptr;
return new_head;
}
};