给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
先通过循环遍历,计算出链表的长度num,然后用k%num,因为如果k%num=0,我们就不需要去旋转链表,如果k>num我们只需要去旋转取余次,定义一个指针变量temp,使tem->next=头节点,然后开始循环找到倒数第二个节点也就是p1和最后一个节点p2,把倒数第二个节点p1->next赋值为null,把最后一个节点p2->next赋值给第一个节点temp->next,最后把temp指向p2,这样就OK了。
示例 1:
输入: 1->2->3->4->5->NULL, k = 2
输出: 4->5->1->2->3->NULL
解释:
向右旋转 1 步: 5->1->2->3->4->NULL
向右旋转 2 步: 4->5->1->2->3->NULL
示例 2:
输入: 0->1->2->NULL, k = 4
输出: 2->0->1->NULL
解释:
向右旋转 1 步: 2->0->1->NULL
向右旋转 2 步: 1->2->0->NULL
向右旋转 3 步: 0->1->2->NULL
向右旋转 4 步: 2->0->1->NULL
struct ListNode {
int val;
struct ListNode *next;
};
struct ListNode* rotateRight(struct ListNode* head, int k){
if((k == 0) || (head == NULL)) {
return head;
}
struct ListNode temp, *p1, *p2, *p3;
int num = 0;
p3 = head;
while(p3 != NULL) {
p3 = p3->next;
num++;
}
k %= num;
temp.next = head;
while(k--) {
p1 = &temp;
p2 = p1->next;
while(p2->next != NULL) {
p1 = p1->next;
p2 = p1->next;
}
p1->next = NULL;
p2->next = temp.next;
temp.next = p2;
}
return temp.next;
}