leetcode错题记录（5）

最新推荐文章于 2024-05-31 17:17:25 发布

@xiaohe

最新推荐文章于 2024-05-31 17:17:25 发布

阅读量106

点赞数

文章标签： leetcode 链表数据结构

本文链接：https://blog.csdn.net/weixin_50813307/article/details/126797710

版权

61. 旋转链表https://leetcode.cn/problems/rotate-list/

这是难度中等的题，也是单链表相关的，觉得最近比较做链表类的题，哈哈，可能是觉得找到了解这类题的方法，比较容易。这道题目的是将最后的节点移到头节点前面，以此来实现链表整体右移的效果。同样的，首先想到的也是栈。

思想：先遍历一遍链表，将所有节点放入栈，然后将将栈顶节点的next指向头节点，再删除栈顶节点，删除之后将新的栈顶节点的next设置为null，否则就成循环链表了，一直重复，直到执行到n次为止（n为输入参数），这样就可以实现链表右移。

class Solution {
    public ListNode rotateRight(ListNode head, int k) {
        if(head==null||head.next==null) return head;
        ListNode temp=head;
        Stack<ListNode> stack = new Stack<>();
        while(temp!=null){
            stack.push(temp);
            temp=temp.next;
        }
        for(int i=1; i<=k; i++){
            temp=stack.peek();
            temp.next=head;
            stack.pop();
            stack.peek().next=null;
            head=temp;
        }
        return head;
    }
}

测试用例通过，然后提交，结果报的是空栈异常，后面检查发现这里输入的k并没有说明是小于链表长度的，意思是这里的k可以无限大，当k=链表长度时，则相当于右移了0位，所以我们这里要对k求余得到steps，我们只需要右移steps就可以了。

修改后的代码如下：

class Solution {
    public ListNode rotateRight(ListNode head, int k) {
        if(head==null||head.next==null) return head;
        ListNode temp=head;
        Stack<ListNode> stack = new Stack<>();
        while(temp!=null){
            stack.push(temp);
            temp=temp.next;
        }
        //这里是关键，我们要对k求余，否则当k大于链表长度时会报空栈异常
        int steps = k%stack.size();
        for(int i=1; i<=steps; i++){
            temp=stack.peek();
            temp.next=head;
            stack.pop();
            stack.peek().next=null;
            head=temp;
        }
        return head;
    }
}

提交结果：