给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
示例 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* rotateRight(struct ListNode* head, int k){
if (head == NULL) return head;
struct ListNode *newhead = head, *tail = head, *temp = head;
int len = 0;
while (temp != NULL) {
if (temp->next == NULL)
tail = temp;
len++;
temp = temp->next;
}
int move_len = len - k%len;
if (move_len == len) return head;
for (int i = 0; i < move_len-1; i++)
newhead = newhead->next;
temp = newhead;
newhead = newhead->next;
temp->next = NULL;
tail->next = head;
return newhead;
}