旋转链表

最新推荐文章于 2021-09-28 15:12:35 发布

原创最新推荐文章于 2021-09-28 15:12:35 发布 · 122 阅读

0 ·

CC 4.0 BY-SA版权

leetcode C++ 专栏收录该内容

116 篇文章

订阅专栏

本文详细解析了链表旋转算法的实现过程，通过实例展示了如何将链表的节点向右移动k个位置，包括计数、重设k值、定位倒数第k个节点以及进行链表的重新链接。

给定一个链表，旋转链表，将链表每个节点向右移动 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

分析：

就是把后k个节点拼接到最前面

/**
 * 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) {
        // 就是把最后k个节点拿到前面来嘛
        
        // 计数,使得k的值小于链表的长度，方便后续使用
        ListNode *curr = head;
        int count = 0;
        while(curr){
            curr = curr->next;
            count ++;
        }
        
        // 0判断
        if(k == 0 || count == 0 || k % count == 0)
            return head;
        
        // 重设k值
        k = k % count;
        
        // 先找到倒数第k个节点
        ListNode *left = head;
        ListNode *right = head;
        // 0 1 2            4
        while(k>0){
            if(right){
                right = right->next;
            }else{
                right = head;
            }
            k --;
        }
        while(right->next){
            right = right->next;
            left = left->next;
        }
        
        // 链接
        ListNode *new_head = left->next;
        left->next = NULL;
        right->next = head;
        return new_head;
    }
};