[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;
}

你可能感兴趣的:(LeetCode,leetcode,链表)