目录
- leetcode 234
- leetcode 19
leetcode 234
题目在这https://leetcode.cn/problems/palindrome-linked-list/,leetcode 234的回文链表,思路很简单,就是fast和slow两个指针,fast一次移动两个、slow一次一个,最后slow指向的链表反转后和head比较就行了。
很简单一题,但考虑一个问题:slow和fast是什么形式?由于rust有所有权机制,所以slow和fast只能是借用,并且还不能是可变借用(可变借用只能有一个、可变借用和不可变借用不能同时存在)。
所以slow和fast只能是两个不可变借用,直接放上代码:
// 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 reverse(mut head: Option<Box<ListNode>>) -> Option<Box<ListNode>> {
let mut prev = None;
while let Some(mut node) = head {
head = node.next;
node.next = prev;
prev = Some(node);
}
prev
}
pub fn is_palindrome(head: Option<Box<ListNode>>) -> bool {
// 快慢指针,fast一次移动2个,slow一次移动1个,slow会停留在中间偏后的位置
// 反转slow为头结点的链表
// 比较head和slow两个链表,直到head的长度达到即停止
let p = head.as_ref().unwrap();
if p.next.is_none() {
return true;
}
let mut head = head;
let mut slow = &head;
let mut fast = &head;
while slow.is_some() && fast.is_some() {
slow = &(slow.as_ref().unwrap().next);
fast = &(fast.as_ref().unwrap().next);
if fast.is_none() {
break;
}
fast = &(fast.as_ref().unwrap().next);
}
// let s = slow as *const Option<Box<ListNode>> as * mut Option<Box<ListNode>>;
let s = slow as *const Option<Box<ListNode>> as *mut Option<Box<ListNode>>;
let mut head2 = unsafe {
(*s).take()
};
head2 = Solution::reverse(head2);
let mut flag = true;
while let (Some(node1), Some(node2)) = (head.as_ref(), head2.as_ref()) {
if node1.val != node2.val {
flag = false;
break;
}
head = head.unwrap().next;
head2 = head2.unwrap().next;
}
flag
}
}
主要注意代码中的
let s = slow as *const Option<Box<ListNode>> as *mut Option<Box<ListNode>>;
let mut head2 = unsafe {
(*s).take()
};
这里我的本意是写成这样:
let mut head2 = slow.take();
但是slow是不可变借用,这里就会报错:
cannot borrow `*slow` as mutable, as it is behind a `&` reference
`slow` is a `&` reference, so the data it refers to cannot be borrowed as mutable
大意就是slow不是可变借用,take()的作用是拿走物主对某个东西的所有权,然后将所有权交给另一个人。你借一个人的东西,并且不被允许改变这个东西,那么你肯定不能把这个东西的所有权扔给别人对吧。
这个时候就需要裸指针的概念了,如果不会请移步:
https://kaisery.github.io/trpl-zh-cn/ch19-01-unsafe-rust.html
链接中截图关键内容:
注意*const
和*mut
是不可变、可变两种裸指针,星号不是解引用,而是类型的一部分。⚠️注意是将不可变引用变成*const
类型的裸指针,可变引用变成*mut
类型的裸指针,所以前面的代码里写的是:
let s = slow as *const Option<Box<ListNode>> as *mut Option<Box<ListNode>>;
slow是不可变引用&Option<Box<ListNode>>
,所以先转换为*const Option<Box<ListNode>>
,再转换为*mut Option<Box<ListNode>>
类型。
然后要在unsafe代码块中使用它,记得写法是(*s).take()
拿到所有权赋给head2
let mut head2 = unsafe {
(*s).take()
};
至此head2就是slow引用的那个节点了。
leetcode 19
题目是删除链表中倒数第n个节点,要求一趟遍历。
这里也可以使用裸指针,只要是如下场景都可以考虑裸指针:
(1)&和&mut同时出现,又不可避免。那么如果已经有了&,还需要一个&mut的话,可以创建裸指针mut;
(2)需要通过&不可变引用进行改变,那么可以将&转换为const,再转换为*mut,此时就和&mut作用一致了。
// 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 remove_nth_from_end(head: Option<Box<ListNode>>, n: i32) -> Option<Box<ListNode>> {
// 从头结点开始向后走n - 1步
let mut dummy = Some(Box::new(ListNode::new(0)));
dummy.as_mut().unwrap().next = head;
let mut p_tail = &(dummy.as_ref().unwrap().next);
let mut cnt = 0; // 向后走了几步
while cnt < n - 1 {
cnt += 1;
p_tail = &(p_tail.as_ref().unwrap().next);
}
let mut delete_next = &dummy;
while p_tail.as_ref().unwrap().next.is_some() {
p_tail = &(p_tail.as_ref().unwrap().next);
delete_next = &(delete_next.as_ref().unwrap().next);
}
// 至此,我们只需要删除delete_next节点的下一个节点
// 然后返回dummy的下一个节点即可
// 通过裸指针拿到delete_next节点的下一个的下一个节点的所有权
let mut need_take = &(delete_next.as_ref().unwrap().next);
need_take = &(need_take.as_ref().unwrap().next);
// need_take是不可变引用,先拿到*mut类型的裸指针,便可拿到need_take引用的节点所有权
let mut temp1 = need_take as *const Option<Box<ListNode>> as *mut Option<Box<ListNode>>;
let mut temp = unsafe {
(*temp1).take()
};
// 我们需要将need_take的所有权赋给delete_next节点的next,所有权现在给了temp
// delete_next是不可变引用,我们要修改它引用的节点的next,就可以通过可变裸指针
// 不可变引用需要先转换为*const再转换为*mut
let mut delete_next_temp = delete_next as *const Option<Box<ListNode>> as *mut Option<Box<ListNode>>;
unsafe {
(*delete_next_temp).as_mut().unwrap().next = temp;
}
dummy.unwrap().next
}
}