[LeetCode] 61. Rotate List

思路:

又是个链表小花式操作, 一共轮两遍链表, 第一次计算出链表总节点个数, 第二遍进行拼接. 第一遍算出个数后要更新k的值, 因为k有可能大于等于节点个数, 为了不做多余动作, 就要这么处理. 然后新建一个虚节点做头, 很常用的技巧了, 一直找找到新链表的尾, 记住此时这个尾的下一个节点就要做新链表的头节点, 记录这个尾巴是为了最后好把它的next置为空. 然后再向下挪动一位, 把新链表的头结点接到虚节点后, 然后再遍历到旧链表尾端, 拿这个尾巴接上旧链表的头, 最后把新尾端的next置空即可.

ListNode* rotateRight(ListNode* head, int k) {
if (! head || ! head->next || ! k) return head;
// 数个数
int size = 0;
ListNode* temp = head;
while (temp) {
size++;
temp = temp->next;
}
// 重置k
if (k >= size) k %= size;
if (! k) return head;

ListNode dummy(0);
ListNode* pDummy = &dummy;
temp = head;
for (int i = 0; i < size - k - 1; i++)
temp = temp->next;
// 记录新尾节点
ListNode* tail = temp;
temp = temp->next;
// 将新头节点接到虚节点后面
pDummy->next = temp;

while (temp->next)
temp = temp->next;
// 将旧头节点接到旧尾节点后面
temp->next = head;
// 将新尾节点的next置空
tail->next = NULL;

return dummy.next;
}