题目描述:
给定一个链表,旋转链表,将链表每个节点向右移动 k 个位置,其中 k 是非负数。
示例1:
输入: 1->2->3->4->5->NULL, k = 2 输出: 4->5->1->2->3->NULL 解释: 向右旋转 1 步: 5->1->2->3->4->NULL 向右旋转 2 步: 4->5->1->2->3->NULL 来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/rotate-list 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例2:
输入: 0->1->2->NULL, k = 4 输出: 2->0->1->NULL 解释: 向右旋转 1 步: 2->0->1->NULL 向右旋转 2 步: 1->2->0->NULL 向右旋转 3 步: 0->1->2->NULL 向右旋转 4 步: 2->0->1->NULL 来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/rotate-list 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目难度:中等
分析:
题目比较好理解,每旋转一次,就是把链表最后的一个元素拿到链表开头而已,可以简化的地方也就是当旋转了一个周期(链表长度)以后链表相当于没变化。也就是从链表尾部向前数k的元素,在此时截断链表,分为两部分,然后把后半部分拿到前面即可。
代码如下:
java:
/** * Definition for singly-linked list. * public class ListNode { * int val; * ListNode next; * ListNode(int x) { val = x; } * } */ class Solution { public ListNode rotateRight(ListNode head, int k) { // 如果链表为空,或者旋转次数为0时,可以直接返回原链表 if (head == null || k == 0) { return head; } // 定义一个指向链表开头的指针 ListNode current = head; // 最后返回结果的指针 ListNode res = new ListNode(0); // 用来指向结果的指针 ListNode p = res; // 链表的长度 int length = 0; // 计算链表的长度 while (current != null) { length++; current = current.next; } // 这里很容易知道:转了一圈又回到原点等于没动,所以取模简化 if (k % length == 0) { return head; } // 计算完长度,指针回到头节点 current = head; // 用来存储指针移动后最后所处的位置,也就是需要head链表的前几个元素 int falg = 0; // 这里要注意了,我刚开始以为旋转的意思就是指针向后移动几次而已 //实际上是要把尾部的元素放到开头才对,这样就会有个差值,也就是计算长度的目的 // 只需要对length取模即可。然后找到current应该处在的位置,就是该在什么地方切断 for (int i = 0; i < length - k % length; i++) { if (current != null) { current = current.next; falg++; } } // 从标记位往后,添加到新链表中 while (current != null) { p.next = new ListNode(current.val); p = p.next; current = current.next; } // 最后把标记位之前到所有元素放到res最后面即可 current = head; for (int i = 0; i < falg; i++) { p.next = new ListNode(current.val); p = p.next; current = current.next; } return res.next; } }
python:
# Definition for singly-linked list. # class ListNode: # def __init__(self, x): # self.val = x # self.next = None class Solution: def rotateRight(self, head: ListNode, k: int) -> ListNode: if head == None or k == 0: return head current, res = head, ListNode(0) p, length, flag = res, 0, 0 while current != None: length += 1 current = current.next if k % length == 0: return head current = head for i in range(length - k % length): if current != None: flag += 1 current = current.next while current != None: p.next = ListNode(current.val) p = p.next current = current.next current = head for i in range(flag): p.next = ListNode(current.val) p = p.next current = current.next return res.next
总结:
属于链表中比较常规的一道题。可配合官方的动画来看,这样就比较好理解。只要理解了题目意思,也就很简单了。