给你一个链表的头节点 head ,旋转链表,将链表每个节点向右移动 k 个位置。
示例 1:
输入:head = [1,2,3,4,5], k = 2
输出:[4,5,1,2,3]
示例 2:
输入:head = [0,1,2], k = 4
输出:[2,0,1]
提示:
链表中节点的数目在范围 [0, 500] 内
-100 <= Node.val <= 100
0 <= k <= 2 * 109
思路:贪心算法 |
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if(!head) return head; //为空特判
int n=0;
ListNode* tail;
for(auto p=head;p;p=p->next){ //先求出总结点数
tail=p; //标记出尾结点
n++;
}
k%=n; //k可能很大,将其缩小到n以内
if(!k) return head;
auto p=head;
for(int i=0;i<n-k-1;i++){ //找出旋转后的尾结点
p=p->next;
}
tail->next=head;
head=p->next;
p->next=nullptr;
return head;
}
};