题目:
Given a list, rotate the list to the right by k places, where k is non-negative.
For example:
Given 1->2->3->4->5->NULL
and k = 2
,
return 4->5->1->2->3->NULL
.
Given [0,1,2]
, rotate 1 steps to the right -> [2,0,1]
.
Given [0,1,2]
, rotate 2 steps to the right -> [1,2,0]
.
Given [0,1,2]
, rotate 3 steps to the right -> [0,1,2]
.
Given [0,1,2]
, rotate 4 steps to the right -> [2,0,1]
.
方法:先成环,再断开链表。
思路:由于旋转的节点数k可能为0,也可能超过链表的长度,所以我们需要对k取余。游标指针从头节点向后移动,当指向尾节点时,得到链表长度len;然后将链表头尾相连,接着游标再向后移动 k % len 步得到结果链表的尾节点。最后将head指向结果尾部节点的下一个节点,再断开链表(尾部节点->next置NULL)
Attention:
1. 由于旋转后,我们并没有改变相对节点顺序(如1-2-3-4,k=2,变为3-4-1-2),所以可以先形成链表环,之后再要求位置断开。这种方法十分巧妙,应该学习。(此时,p指向原链表最后一个节点)
//(2) 让链表头尾相连 p->next = head;
2. 需要考虑k的不同取值,k = 0, k<=len, k> len, 取余得到相对旋转位置。
//(3) 找到结果链表的尾节点 k %= len; int step = len - k;3. 注意初始时,如果链表无节点,返回head, 排除特殊情况。
if(head == NULL) return head;复杂度:O(N)
AC Code:
/** * 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 == NULL) return head; ListNode* p = head; int len = 1; //(1) 计算链表长度 while(p->next != NULL) { p = p->next; len++; } //(2) 让链表头尾相连 p->next = head; //(3) 找到结果链表的尾节点 k %= len; int step = len - k; while(step > 0) { p = p->next; step--; } //(4) 断开链表,从结果链表的尾节点位置 新的头结点,是结果链表尾节点的下一个结点。结果链表尾节点->next 置NULL head = p->next; p->next = NULL; return head; } };