61. 旋转链表
解题思路:
这题原来打算使用比较笨的方法,就是用unordered_map去映射每一个节点,把新的索引作为map的索引,但是发现最后结果超时了。。
所以新的思路就是,首先把整个链表拉成一个环,然后通过规律找到旋转后,新的第一个元素节点的位置,从这个位置把环断开。
假设整个链表长度为len,要右移旋转的偏移量为k(就是题里给的那个k),则第一个元素被偏移到: ( 0 + k ) % l e n (0+k) \% len (0+k)%len,而新的第一个元素的原来的索引就是:
l e n − ( 0 + k ) % l e n len - (0+k) \% len len−(0+k)%len,这样我们就找到了断开环的位置,让head等于这个位置的元素,把之前元素的next置为NULL即可。
源代码:
/**
* 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 || k == 0) return head;
int len = 0;
ListNode *p = head;
while(p){
len ++;
if(!p->next){
p->next = head;
break;
}
p = p->next;
}
p = head;
// 记录第一个元素的位置
int c_index = len - (k % len), i = 0;
while(p && i < c_index-1){
p = p->next;
i ++;
}
head = p->next;
p->next = NULL;
return head;;
}
};