Leetcode—61. 旋转链表【中等】

源代码•宸

已于 2024-02-28 15:32:49 修改

阅读量391

点赞数 1

分类专栏： LeetCode刷题文章标签： leetcode 链表算法 c++ 数据结构经验分享

于 2024-02-07 15:11:28 首次发布

本文链接：https://blog.csdn.net/qq_44631615/article/details/136069737

版权

LeetCode刷题专栏收录该内容

398 篇文章 5 订阅

订阅专栏

2024每日刷题（114）

Leetcode—61. 旋转链表

在这里插入图片描述

C++实现代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* rotateRight(ListNode* head, int k) {
        if(!head || head->next == nullptr || k == 0) {
            return head;
        }
        ListNode* tail = head;
        int len = 1;
        for(; tail->next; tail = tail->next) {
            len++;
        }
        // 成环
        tail->next = head;
        int t = len - k % len;
        for(int i = 0; i < t; i++) {
            tail = tail->next;
        }
        ListNode* newNode = tail->next;
        // 断链
        tail->next = nullptr;
        return newNode;
    }
};

运行结果

在这里插入图片描述

C语言实现代码

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
struct ListNode* rotateRight(struct ListNode* head, int k) {
    if(!head || head->next == NULL || k == 0) {
        return head;
    }
    int lcn = 0;
    struct ListNode* p = head;
    struct ListNode* q = p;
    while(p) {
        lcn++;
        q = p;
        p = p->next;
    }
    q->next = head;
    int cnt = lcn - k % lcn - 1;
    p = head;
    while(cnt--) {
        p = p->next;
    }
    q = p->next;
    p->next = NULL;
    return q;
}