61. 旋转链表

101 阅读2分钟

题目介绍

力扣61题:leetcode-cn.com/problems/ro…

image.png

image.png

分析

根据题目意思,给你一个链表的头节点 head ,然后将链表每个节点向右移动 k 个位置。

image.png 如样例所示,head = [1,2,3,4,5],k = 2,我们输出[4,5,1,2,3]。下面来讲解模拟的做法。

假设链表的长度为n,为了将链表每个节点向右移动 k 个位置,我们只需要将链表的后 k % n个节点移动到链表的最前面,然后将链表的后k % n个节点和前 n - k个节点连接到一块即可。

具体过程如下:

1、首先遍历整个链表,求出链表的长度n,并找出链表的尾节点tail。

image.png

2、由于k可能很大,所以我们令 k = k % n,然后再次从头节点head开始遍历,找到第n - k个节点p,那么1 ~ p是链表的前 n - k个节点,p+1 ~ n是链表的后k个节点。

image.png 3、接下来就是依次执行,

//将n-k之后的节点连接第1~n-k节点
tail->next = head;
//旋转之后新的头节点
head = p->next;
//旋转之后p节点就是尾结点
p->next = null;

将链表的后k个节点和前 n - k个节点拼接到一块,并让head指向新的头节点(p->next),新的尾节点即p节点的next指针指向null

image.png

4、最后返回链表的新的头节点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 || k == 0)  {
            return head;
        }
        int n = 0;//记录链表的长度
        ListNode tail = null;  //尾节点
        for(ListNode p = head; p != null ; p = p.next){
            tail = p;//记录链表的尾节点
            n++;//计算链表的长度
        }
        //防止k比链表的长度n还大
        k %= n;
        ListNode p = head;
        for(int i = 0; i < n - k - 1; i++)  {
            p = p.next;   //找到链表的第n-k个节点
        }
        tail.next = head;//将n-k之后的节点连接第1~n-k节点
        //旋转之后新的头节点
        head = p.next;
        //旋转之后p节点就是尾结点
        p.next = null;
        return head;  //返回新的头节点
    }
}

时间复杂度分析:  

链表一共被遍历两次,因此总的时间复杂度为O(n),n是链表的长度。