61. 旋转链表:
给你一个链表的头节点 head
,旋转链表,将链表每个节点向右移动 k
个位置。
样例 1:
输入:
head = [1,2,3,4,5], k = 2
输出:
[4,5,1,2,3]
样例 2:
输入:
head = [0,1,2], k = 4
输出:
[2,0,1]
提示:
- 链表中节点的数目在范围 [0, 500] 内
- -100 <= Node.val <= 100
- 0 <= k <= 2 * 109
分析:
- 面对这道算法题目,二当家的再次陷入了沉思。
- 首先节点向右移动的位置
k
为0,我们什么都不需要做,直接返回原来的链表即可。 - 如果想要旋转链表,就必须知道链表的长度,所以我们先从头遍历一次链表,计算一下链表的长度,如果长度是0,同样我们什么都不需要做,也直接返回原来的链表即可。
- 在知道链表的长度
n
之后,我们就可以旋转链表了,但是节点向右移动位置k
的值可能大于链表的长度n
,这时候我们并不需要真移动k
个位置,因为每移动n
个位置,链表就会恢复原状,所以我们只需要移动k % n
个位置,而如果这个值是0时,我们依然可以什么都不做直接返回原来的链表。 - 移动之后,链表的头和尾会发生变化,所以需要确定新的尾和头然后将他们断开,并且把旧的尾和头相连。
题解:
rust:
// Definition for singly-linked list.
// #[derive(PartialEq, Eq, Clone, Debug)]
// pub struct ListNode {
// pub val: i32,
// pub next: Option<Box<ListNode>>
// }
//
// impl ListNode {
// #[inline]
// fn new(val: i32) -> Self {
// ListNode {
// next: None,
// val
// }
// }
// }
impl Solution {
pub fn rotate_right(mut head: Option<Box<ListNode>>, k: i32) -> Option<Box<ListNode>> {
if k == 0 || head.is_none() {
return head;
}
// 链表的长度
let mut n = 0;
let mut tail = &head;
while let Some(node) = tail {
n += 1;
tail = &node.next;
}
let add = n - k % n;
if add == n {
return head;
}
// 寻找新尾
let mut tail = &mut head;
for _ in 1..add {
tail = &mut tail.as_mut().unwrap().next;
}
// 新头和新尾断开
let mut ans = tail.as_mut().unwrap().next.take();
// 找到旧尾
let mut tail = &mut ans;
while tail.is_some() && tail.as_ref().unwrap().next.is_some() {
tail = &mut tail.as_mut().unwrap().next;
}
// 旧尾和旧头相连
tail.as_mut().unwrap().next = head;
return ans
}
}
go:
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func rotateRight(head *ListNode, k int) *ListNode {
if k == 0 || head == nil || head.Next == nil {
return head
}
// 链表的长度
n := 1
tail := head
for tail.Next != nil {
tail = tail.Next
n++
}
add := n - k%n
if add == n {
return head
}
// 成环
tail.Next = head
// 寻找新尾
for add > 0 {
tail = tail.Next
add--
}
// 新头
ans := tail.Next
// 新尾和新头断开
tail.Next = nil
return ans
}
c++:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* rotateRight(ListNode* head, int k) {
if (k == 0 || head == nullptr || head->next == nullptr) {
return head;
}
// 链表的长度
int n = 1;
ListNode *tail = head;
while (tail->next != nullptr) {
tail = tail->next;
n++;
}
int add = n - k % n;
if (add == n) {
return head;
}
// 旧尾连旧头成环
tail->next = head;
// 寻找新尾
while (add--) {
tail = tail->next;
}
// 新头
ListNode *ans = tail->next;
// 新尾和新头断开
tail->next = nullptr;
return ans;
}
};
python:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def rotateRight(self, head: Optional[ListNode], k: int) -> Optional[ListNode]:
if k == 0 or not head or not head.next:
return head
# 确定链表的长度
n = 1
tail = head
while tail.next:
tail = tail.next
n += 1
if (add := n - k % n) == n:
return head
# 旧尾连旧头成环
tail.next = head
# 寻找新尾
while add:
tail = tail.next
add -= 1
# 新头
ans = tail.next
# 新尾和新头断开
tail.next = None
return ans
java:
/**
* 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 (k == 0 || head == null || head.next == null) {
return head;
}
// 链表的长度
int n = 1;
ListNode tail = head;
while (tail.next != null) {
tail = tail.next;
n++;
}
int add = n - k % n;
if (add == n) {
return head;
}
// 旧尾连旧头成环
tail.next = head;
// 寻找新尾
while (add-- > 0) {
tail = tail.next;
}
// 新头
ListNode ans = tail.next;
// 新尾和新头断开
tail.next = null;
return ans;
}
}
非常感谢你阅读本文~
欢迎【点赞】【收藏】【评论】三连走一波~
放弃不难,但坚持一定很酷~
希望我们大家都能每天进步一点点~
本文由 二当家的白帽子:https://le-yi.blog.csdn.net/ 博客原创~