[LeetCode]61.旋转链表

博客探讨了如何优化链表旋转算法,从最初的实现导致超时问题,到通过计算链表长度和取模减少循环次数实现AC,最后采用连接成环的方法在O(1)时间内完成旋转,显著提高了效率。

[OT]My Answer

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode rotateRight(ListNode head, int k) {
        if(head == null || head.next == null){
            return head;
        }
        int store = 0;
        ListNode p;
        for(int i = 0; i < k; i++){
            p = head;
            store = p.val;
            while(p.next != null){
                int tmp = p.next.val;
                p.next.val = store;
                store = tmp;
                p = p.next;
            }
            head.val = store;
        }
        return head;

    }
}

第一次写出来,能够解出符合要求的结果,但是超时了,因为循环的次数等于右移次数,若是右移次数相当大,则耗时很高。后考虑到移动次数较高时移动是循环进行的,可以先得到链表的长度,后让右移次数对长度取余,修改后的结果如下:

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode rotateRight(ListNode head, int k) {
        if(head == null || head.next == null){
            return head;
        }
        int store = 0;
        ListNode p = head;
        int num = 1;
        while(p.next != null){
            num++;
            p = p.next;
        }
        int times = k % num;
        for(int i = 0; i < times; i++){
            p = head;
            store = p.val;
            while(p.next != null){
                int tmp = p.next.val;
                p.next.val = store;
                store = tmp;
                p = p.next;
            }
            head.val = store;
        }
        return head;

    }
}

修改后 AC 了,然而时间复杂度只超过了 8 % 的人,说明效率很低。

题解

连接成环

class Solution {
    public ListNode rotateRight(ListNode head, int k) {
        if (k == 0 || head == null || head.next == null) {
            return head;
        }
        int n = 1;
        ListNode iter = head;
        while (iter.next != null) {
            iter = iter.next;
            n++;
        }
        int add = n - k % n;
        if (add == n) {
            return head;
        }
        iter.next = head;
        while (add-- > 0) {
            iter = iter.next;
        }
        ListNode ret = iter.next;
        iter.next = null;
        return ret;
    }
}

将单链表连接成环后,只需要移动指针到合适位置,然后断开环即可。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值