力扣刷题 - 61. 旋转链表

题目:
给你一个链表的头节点 head ,旋转链表,将链表每个节点向右移动 k 个位置。

示例 1:
力扣刷题 - 61. 旋转链表_第1张图片
输入:head = [1,2,3,4,5], k = 2
输出:[4,5,1,2,3]

示例 2:
力扣刷题 - 61. 旋转链表_第2张图片
输入:head = [0,1,2], k = 4
输出:[2,0,1]

提示:

  • 链表中节点的数目在范围 [0, 500] 内
  • -100 <= Node.val <= 100
  • 0 <= k <= 2 * 109

简单描述一下我的思路:

  • 将链表的首尾连接起来,进行旋转。

要注意的是有个点很重要需要计算实际的旋转次数,当结点数小于题目给定的旋转次数时,显然中间有旋转回原来的链表的操作,这些操作是不需要的,我的做法是 k %= 结点数。还有就是首尾连接之后记得断开连接。详细的在代码中有注释

class Solution {
    public ListNode rotateRight(ListNode head, int k) {
        // 排除特殊情况
        if (head == null || head.next == null || k == 0) return head;
        // 统计当前链表的结点数
        int count = 1;
        ListNode tem = head;
        while(tem.next != null){
            count++;
            tem = tem.next;
        }
        // 计算实际需要旋转次数
        k %= count;
        // k 为 0,说明经过旋转后变回原来的链表
        if(k == 0) return head;

        // 将链表结点首尾连接
        tem.next = head;
        // 开始旋转
        for(int i = 0;i < count - k;i++){
            tem = tem.next;
        }
        // 去除首尾连接
        ListNode newNode = tem.next;
        tem.next = null;
        
        return newNode;
    }
}

力扣刷题 - 61. 旋转链表_第3张图片

你可能感兴趣的:(算法学习,链表,leetcode,数据结构,java,算法)