首页
关于
Search
1
2023 年保研经验贴 [计算机-末九Rank50%-入北计清深-北计直博]
1,127 阅读
2
FreeBSD Kernel-编译环境搭建
434 阅读
3
Linux Kernel-THP阅读分析[Hang up]
424 阅读
4
Linux Kernel-编译调试环境搭建
335 阅读
5
Linux Kernel-源码阅读环境搭建
322 阅读
内核
源码分析
阅读笔记
Rust
语法
习题
读书笔记
深入理解计算机系统
论文
技术之外
开发
rcore-arm
登录
Search
yyi
累计撰写
49
篇文章
累计收到
2
条评论
首页
栏目
内核
源码分析
阅读笔记
Rust
语法
习题
读书笔记
深入理解计算机系统
论文
技术之外
开发
rcore-arm
页面
关于
搜索到
21
篇与
的结果
2024-04-07
Rustlings Solution
可以参见我的Repo: Github Repo
2024年04月07日
171 阅读
0 评论
0 点赞
2024-02-24
Rust算法练习-LC2092/997/787/980/2646
LC2646. Minimize the Total Price of the Trips Hard袜,这题还是挺有挑战的。给定一棵树,每个点有权值,给定M组点对 (ai, bi),求Sum(dis[ai, bi])。如果只到这,好像还没啥,树剖LCA就能解决,别说N=50了,N=10^5也行但是多了个条件,可以把一些不相邻的点的权值减半,让我们求最小的结果我目测得把路径离线,先统计出来每个点要走多少次。树可以染色吗,所有点一定是可以分成两组的么,如果是,那就简单了,无非是选组1或者组2,但是显然不是(考虑链:1-2-3-4,我可以染1和4)考虑树形dp,一个节点能染色要求它的子节点都没有染色,那好像就好做了,我们预先确定一个根rootdpi表示到节点i,节点有半价/无半价的最小值dpfa = Sum(min(dp[son])) + price[fa] * freq[fa]dpfa = Sum(dpson) + price[fa]/2 * freq[fa]边界是叶子节点,直接处理自己的值就行,考虑这样是不是能覆盖所有状态。不相邻的话应该是恰好覆盖0ms,一遍过,讲真,现在一遍过的概率高很多,我还是蛮高兴的,证明自己下手越来越有准头了。impl Solution { pub fn minimum_total_price( n: i32, edges: Vec<Vec<i32>>, price: Vec<i32>, trips: Vec<Vec<i32>>, ) -> i32 { let n = n as usize; let mut fa = vec![0; n]; let mut freq: Vec<i32> = vec![0; n]; let mut dep: Vec<i32> = vec![0; n]; let edges: Vec<Vec<i32>> = edges.iter().fold(vec![vec![]; n], |mut acc, e| { acc[e[0] as usize].push(e[1]); acc[e[1] as usize].push(e[0]); acc }); let mut dp: Vec<Vec<i32>> = vec![vec![0; 2]; n]; let root = 0; dfs(root, &mut fa, &mut dep, &edges); for trip in trips.iter() { let mut x = trip[0] as usize; let mut y = trip[1] as usize; while x != y { if dep[x] < dep[y] { std::mem::swap(&mut x, &mut y); } freq[x] += 1; x = fa[x] as usize; } freq[x] += 1; } dfs2(root, &edges, &freq, &mut dp, -1, &price); dp[root as usize][0].min(dp[root as usize][1]) } } fn dfs(p: i32, fa: &mut Vec<i32>, dep: &mut Vec<i32>, edges: &Vec<Vec<i32>>) { for &to in edges[p as usize].iter() { if to == fa[p as usize] { continue; } fa[to as usize] = p; dep[to as usize] = dep[p as usize] + 1; dfs(to, fa, dep, edges); } } fn dfs2( p: i32, edges: &Vec<Vec<i32>>, freq: &Vec<i32>, dp: &mut Vec<Vec<i32>>, fa: i32, price: &Vec<i32>, ) { let mut res0 = 0; let mut res1 = 0; for &to in edges[p as usize].iter() { if to == fa { continue; } dfs2(to, edges, freq, dp, p, price); res0 += dp[to as usize][0]; res1 += std::cmp::min(dp[to as usize][0], dp[to as usize][1]); } dp[p as usize][0] = res1 + freq[p as usize] * price[p as usize]; dp[p as usize][1] = res0 + freq[p as usize] * (price[p as usize] >> 1); }LC62. Unique Paths/63. Unique Paths II/980. Unique Paths III Med Med Hard62是从左上走到右下的不重复路径数,63是带障碍的,980是从起点到终点,且走过全部空格子的路径数,难度陡增啊62、63都是比较简单的dp,dpi表示走到(i, j)的方案数,dp[unreachable] = -1, ans = dpm, dpi = dpi-1 + dpi; 常规到我不打算贴代码了,主要关注980数据范围 20*20, 这个能直接搜么,搜的复杂度是多少?最讨厌算搜索的复杂度了。不过看上去400个格子,4^400肯定不可做的。但是应该也不是dp,状态太多了,也压不了。我还是倾向于搜索+剪枝,先搜一发试试有个小插曲:你觉得这好看么,我望着这排迭代器沉思了几秒,决定还是乖乖遍历impl Solution { pub fn unique_paths_iii(grid: Vec<Vec<i32>>) -> i32 { let target = grid .iter() .map(|row| row.iter().filter(|&&x| x == 0).count()) .sum::<usize>(); let (x, y) = grid .iter() .enumerate() .flat_map(|(i, row)| { row.iter() .enumerate() .find(|&(_, &x)| x == 1) .map(|(j, _)| (i, j)) }) .next() .unwrap(); } }没想到0ms是真的。。impl Solution { fn dfs(grid: &mut Vec<Vec<i32>>, x: usize, y: usize, now_step: i32, target_step: i32) -> i32 { if grid[x][y] == 2 { return match now_step == target_step { true => 1, false => 0, }; } let mut res = 0; grid[x][y] = -1; let dx = [0, 0, 1, -1]; let dy = [1, -1, 0, 0]; for i in 0..4 { let nx = x as i32 + dx[i]; let ny = y as i32 + dy[i]; if nx >= 0 && nx < grid.len() as i32 && ny >= 0 && ny < grid[0].len() as i32 { let nx = nx as usize; let ny = ny as usize; if grid[nx][ny] == 0 || grid[nx][ny] == 2 { res += Self::dfs(grid, nx, ny, now_step + 1, target_step); } } } grid[x][y] = 0; res } pub fn unique_paths_iii(mut grid: Vec<Vec<i32>>) -> i32 { let mut target_steps = 1; let (mut sx, mut sy) = (0, 0); for i in 0..grid.len() { for j in 0..grid[0].len() { if grid[i][j] == 0 { target_steps += 1; } else if grid[i][j] == 1 { sx = i; sy = j; } } } Self::dfs(&mut grid, sx, sy, 0, target_steps) } }LC787. Cheapest Flights Within K Stops Med求两点之间最短路,但是有个限制,最多走K条边,感觉挺难的,一看N范围100。EMM是不是BFS K层就可以了。这个复杂度是啥样的,应该就是KM能搞,好像也能用类似Dij的思路堆优化,但是优化起来考虑的更多。先写个BFS试试,就是贝尔曼福德吧impl Solution { pub fn find_cheapest_price(n: i32, flights: Vec<Vec<i32>>, src: i32, dst: i32, k: i32) -> i32 { let mut dis = vec![0x3f3f3f3f; n as usize]; dis[src as usize] = 0; let mut now_dis = Vec::with_capacity(n as usize); for _ in 0..=k { now_dis.clear(); now_dis = dis.clone(); for flight in &flights { let u = flight[0] as usize; let v = flight[1] as usize; let w = flight[2]; if now_dis[u] != 0x3f3f3f3f { now_dis[v] = now_dis[v].min(dis[u] + w); } } std::mem::swap(&mut dis, &mut now_dis); } match dis[dst as usize] { 0x3f3f3f3f => -1, _ => dis[dst as usize], } } }LC997. Find the Town Judge Ez找到一个入度n-1,没有出度的点,都懒得开IDE本期每日一题将在今天开始变成每日一题+一道Hard我发现每日一题总刷Ez,一点用没有,Ez等级能用到的语法我基本上已经熟悉了。并将会在近期开rust kernel的impl Solution { pub fn find_judge(n: i32, trust: Vec<Vec<i32>>) -> i32 { let n = n as usize; let (mut ind, mut oud) = (vec![0;n+1], vec![0;n+1]); trust.iter().for_each(|t| { oud[t[0] as usize] += 1; ind[t[1] as usize] += 1; }); (1..=n as i32).find(|&x| oud[x as usize] == 0 && ind[x as usize] == n-1).unwrap_or(-1) } }LC2092. Find All People With Secret Hard10^5个人,10^5的时间,最多10^5的会,N^2是没戏了,应该是哪一维要log这东西要我想的话肯定是一个并查集,按时间就把人塞到并查集里了。哎好像能直接做奥,主要就是同一时间的传播怎么处理这个问题。先不考虑同时开会传播的问题。把会议按时间排个序。对每对x、y,看看是不是有一边持有Secret,如果是,把没有那边并进firstPerson这个集合里。好像考虑的话也没问题。。对同一时间的遍历两遍就行了。既然不能只遍历两遍,考虑这块怎么优化。一个比较显然的优化是,如果能把当前时间的人也划分联通块,就不会是链式传播了。但是这样要写一个离散化并查集,好麻烦。。。(不然初始化是N的,如果离散化,应该是均摊N)但是我考虑一个暴力,在sys里也常用。我不扫两遍,我扫到没有更新为止。这样复杂度肯定是不对的,但是没准能骗struct Union { parent: Vec<i32>, } impl Union { fn new(n: usize) -> Self { Union { parent: (0..n as i32).collect(), } } fn find(&mut self, x: i32) -> i32 { if self.parent[x as usize] != x { self.parent[x as usize] = self.find(self.parent[x as usize]); } self.parent[x as usize] } fn union(&mut self, x: i32, y: i32) { // println!("merge {} {}", x, y); let x = self.find(x); let y = self.find(y); if x != y { self.parent[x as usize] = y; } } } impl Solution { pub fn find_all_people(n: i32, mut meetings: Vec<Vec<i32>>, first_person: i32) -> Vec<i32> { let mut union = Union::new(n as usize); union.union(0, first_person); let mut meetings_by_time = vec![vec![]; 100001]; for i in 0..meetings.len() { let m = &meetings[i]; meetings_by_time[m[2] as usize].push((m[0], m[1])); } // println!("{:?} {:?} {:?}", meetings_by_time[5], meetings_by_time[8], meetings_by_time[10] ); for i in 0..meetings_by_time.len() { let mut changed = true; while changed { changed = false; for j in 0..meetings_by_time[i].len() { let m = meetings_by_time[i][j]; let root = union.find(first_person); let (dx, dy) = (union.find(m.0), union.find(m.1)); if dx == dy || (dx != root && dy != root) { continue; } union.union(m.0, m.1); changed = true; } } } let mut ans = vec![]; let root = union.find(first_person); for i in 0..n { if union.find(i) == root { ans.push(i); } } ans } }
2024年02月24日
68 阅读
0 评论
0 点赞
2024-02-24
Rust难点笔记 2. 链表
为一个基础数据结构开一个新篇?链表这么有牌面么?今天在Leetcode随机到了跳表,于是我准备学习下Rust链表先。乍一看链表应该不难啊,值得放进难点?看一下course.rs:人家开了一整章,好吧,我们知道Rust不允许悬垂指针,那我们的链表可咋整,且听我慢慢来学1 实现链表从简单的定义开始pub enum List { Empty, Elem(i32, List), }当尝试build这个的时候,会出现当前这个类型并不是定长的,Rust编译器认为栈上类型必须有固定长度,因此,我们按照提示,把List通过Box封装到堆上,栈上只留下一个定长的指针。更改为box封装后,我们的链表过编了,但是这还有点问题,一点都不优雅[] = Stack () = Heap [Elem A, ptr] -> (Elem B, ptr) -> (Empty, *junk*)链表头并不是一个指针,而是一个栈上元素,而后面的尾节点却在堆上,而我们期望的(显然,你应该已经了解C的链表了)[head] -> (val A, ptr) -> (val B, null)是这样的链表这是有意义的,前者在分割、合并链表的时候,都会有Copy到栈上、Copy到堆里的操作,但是后者只需要动动指针,也和我们最初的期望保持一致。由此,我们的定义改版为如下:struct Node { elem: i32, next: List, } pub enum List { Empty, More(Box<Node>), }在叙述中,这种enum会遵循Rust的编译器优化,More中包含了一个非0指针,而Empty是无数据的(More、Empty都是自定义的,叫啥都行),在这种情况下不会产生enum_tag的开销,即Rust不需要记住具体是哪种tag,非0就是More,为0就是Empty还有一点小问题,封装的可见性。材料里报的是error,我本地报的warning,但是不管哪种,我们都希望尽可能符合最小暴露原则,所以定义改为如下:pub struct List { head: Link, } enum Link { Empty, More(Box<Node>), } struct Node { elem: i32, next: Link, }接下来,我们为这个链表加上操作impl List { pub fn new() -> Self { List { head: Link::Empty } } pub fn push(&mut self, elem: i32) { // 在链表头插入数据 } }push会改变链表,所以需要一个可变借用,我们新建一个Node,然后把她的next指向旧head,把head指向他 pub fn push(&mut self, elem: i32) { let new_node = Box::new(Node { elem: elem, next: self.head, }); self.head = Link::More(new_node); }看起来很美好,但是估计没有这么简单Link没有实现Copy,我们也不能简单粗暴的实现一个Clone,如果直接赋值,会导致Link的所有权被移走,head悬垂了我们用一个不太优雅的小trick:mem::replace,在从head中偷出值的同时塞回去一个新值变成这样:pub fn push(&mut self, elem: i32) { let new_node = Box::new(Node { elem: elem, next: std::mem::replace(&mut self.head, Link::Empty), }); self.head = Link::More(new_node); }好吧,看着奇奇怪怪,在逻辑中引入了一些不逻辑的东西,但是我们没有什么办法了接着实现Pop,在弹出的同时返回一个Optionpub fn pop(&mut self) -> Option<i32> { match &self.head { Link::Empty => { None; } Link::More(node) => { self.head = node.next; Some(node.elem) } } }不需要编译,都能猜到肯定 self.head = node.next;有点问题,这里又产生move了我本来想说,那就不给大家演示了,结果一编译发现还有个问题两个愿望,一次满足:pub fn pop(&mut self) -> Option<i32> { match std::mem::replace(&mut self.head, Link::Empty) { Link::Empty => None, Link::More(node) => { self.head = node.next; Some(node.elem) } } }把head变成empty,同时换到node里,但是和push一样使用了一点trick链表的释放rust为绝大多数类型实现了drop,并且如果结构的每个字段都有drop,也会自动为结构实现drop但是材料给了一种问题情况:#[test] fn long_list() { let mut list = List::new(); for i in 0..100000 { list.push(i); } drop(list); }虽然这看上去是一个尾递归(递归发生在尾部,应当复用栈帧),但是在rust实现里并非如此impl Drop for List { fn drop(&mut self) { // NOTE: 在 Rust 代码中,我们不能显式的调用 `drop` 方法,只能调用 std::mem::drop 函数 // 这里只是在模拟编译器! self.head.drop(); // 尾递归 - good! } } impl Drop for Link { fn drop(&mut self) { match *self { Link::Empty => {} // Done! Link::More(ref mut boxed_node) => { boxed_node.drop(); // 尾递归 - good! } } } } impl Drop for Box<Node> { fn drop(&mut self) { self.ptr.drop(); // 糟糕,这里不是尾递归! deallocate(self.ptr); // 不是尾递归的原因是在 `drop` 后,还有额外的操作 } } impl Drop for Node { fn drop(&mut self) { self.next.drop(); } } 对于Box\<Node>来说,编译器的行为并不是尾递归,会爆栈材料给出一种方法impl Drop for List { fn drop(&mut self) { let mut cur_link = mem::replace(&mut self.head, Link::Empty); while let Link::More(mut boxed_node) = cur_link { cur_link = mem::replace(&mut boxed_node.next, Link::Empty); } } } 手动的换出每个节点,然后在while结束的时候就drop了优化类型定义和代码复杂度刚刚的代码中,除了replace很怪,还有一大坨的类型声明也很怪,毕竟我们C的链表声明一个结构体就够了有一个很显然的优化是enum Link和Option十分的类似pub struct List { head: Link, } type Link = Option<Box<Node>>; struct Node { elem: i32, next: Link, } Option提供了一个take函数,让我们不需要那个冗长的replacepub fn push(&mut self, elem: i32) { let new_node = Box::new(Node { elem: elem, next: self.head.take(), }); self.head = Some(new_node); }Option让我们的match可以变为mappub fn pop(&mut self) -> Option<i32> { self.head.take().map(|node| { self.head = node.next; node.elem }) }加入泛型我们会在签名处加入 <T\>,这样这个链表即时就支持泛型了pub struct List<T> { head: Link<T>, } type Link<T> = Option<Box<Node<T>>>; struct Node<T> { elem: T, next: Link<T>, } impl<T> List<T> { pub fn new() -> Self { List { head: None } } pub fn push(&mut self, elem: T) { let new_node = Box::new(Node { elem: elem, next: self.head.take(), }); self.head = Some(new_node); } pub fn pop(&mut self) -> Option<T> { self.head.take().map(|node| { self.head = node.next; node.elem }) } } impl<T> Drop for List<T> { fn drop(&mut self) { let mut cur_link = self.head.take(); while let Some(mut boxed_node) = cur_link { cur_link = boxed_node.next.take(); } } }
2024年02月24日
92 阅读
0 评论
0 点赞
2024-02-22
Rust难点笔记 1. 生命周期 [UPD2. 24.2.24]
Rust难点笔记 1. 生命周期说在最前面:这是一篇笔记,而不是教程,难免有不周到的、比较主观的想法,敬请大家批评指正和理解这个系列是支撑我写rCore学习时边写边学产生的笔记,周期会比较长远期计划会包括[生命周期, 宏, 异步, 智能指针, 链表]01 生命周期为什么要有生命周期? Rust的内存机制我们已经很熟悉了。生命周期和引用是强绑定的,就一句话:引用还在的时候,不能让引用的对象消失,从而造成悬垂指针问题因此,我们在编译的时候如果出现一个被引用的对象在其引用前被销毁,就会报 xxx does not live so long的错误,这就是对象存活时间小于引用的生命周期,Rust不允许02 生命周期标注编译器有的时候没办法推断生命周期,需要我们手动标注元素的存活周期。在了解生命周期标注的刚开始,最困扰我的问题就是:我标注的东西,只是一个标注,真的能保证程序安全么带着这个问题开始学看一下最常见的生命周期例子fn max(a: &i32, b: &i32) -> &i32 { if *a > * b { a } else { b } }看到这个函数的第一个结论是:返回值的&i32必须来自于a或b,否则,任何max函数体内创建的变量都会在离开函数时销毁。问题是编译器在编译时无法确定到底来自a还是b(程序员应该预期的是a和b都有可能),此时我们要声明a、b的返回值都是一样的(其实是,至少活的一样长,换句话说,我理解a是三者之间最短的那个的生命周期)在函数后声明生命周期fn max<'a>(a: &'a i32, b: &'a i32) -> &'a i32 { if *a > * b { a } else { b } }此时,我们声明了该函数的两个参数应当和返回值生命周期一样长。考虑如果不是max,是firstfn call_max(a: &i32) -> &i32 { let b = 2; max(a, &b) } fn max<'a>(a: &'a i32, b: &'a i32) -> &'a i32 { a }此时我们只需要要求a的生命周期和返回值一样长了,但是因为标注了a、b都要符合生命周期,导致call_max报错如果改为:fn max<'a, 'b>(a: &'a i32, b: &'b i32) -> &'a i32 { a }call_max就不再会报错了回到报错的case,我们看看为什么因为call_max中,原则上是不能返回b的引用的,b会在call_max退出时销毁。由于调用的时候,a、&b和max的返回值遵循同一个生命周期,rust会取其中最小的,(也就是&b)作为我们标注的生命周期而作为call_max的返回值,其在caller处还可能会被用,但是由于标注最短的原则,它的生命周期只覆盖到call_max函数结尾,这肯定是不行的。而把生命周期改为'b的时候,和'a解绑了,'a和传入call_max的a的生命周期一致,和call_max的返回值一致,而不依赖'b。所以可以通过03 生命周期的函数省略生命周期本来标的就乱七八糟的,编译器会在能推导的情况下生成默认的生命周期https://doc.rust-lang.org/reference/lifetime-elision.html文档里其实介绍了结构和特征的生命周期,但是目测是一法通万法通,先从函数开始可省略的函数参数都有独立生命周期所有参数共用生命后期,生命周期会被直接作用到返回值如果是方法,返回值会和&self的生命周期一致来看看例子第一个例子里是第一条规则,没填就分配独立的第二个例子同理,lvl不是引用,没有生命周期第三个例子,用到了一二两条,给输入参数放了个'a,而入参只有'a 一个生命周期第四个例子还是第一条规则这个用到了一三,如果有&Self,会和&Self的生命周期一致第一个例子无法为输出推断生命周期,因为没有输入第二个例子无法推断生命周期,因为s和t的生命周期默认独立,不能推断返回值的04 结构体生命周期类比函数生命周期,给出一个例子就好懂了struct ImportantExcerpt<'a> { part: &'a str, } fn main() { let novel = String::from("Call me Ishmael. Some years ago..."); let first_sentence = novel.split('.').next().expect("Could not find a '.'"); let i = ImportantExcerpt { part: first_sentence, }; }这个例子说明part必须和持有它的结构体活的至少一样久而这个代码就过不了#[derive(Debug)] struct ImportantExcerpt<'a> { part: &'a str, } fn main() { let i; { let novel = String::from("Call me Ishmael. Some years ago..."); let first_sentence = novel.split('.').next().expect("Could not find a '.'"); i = ImportantExcerpt { part: first_sentence, }; } println!("{:?}",i); }因为结构体本身活到了println处,而part引用只能活到前一个作用域05 方法生命周期struct ImportantExcerpt<'a> { part: &'a str, } impl<'a> ImportantExcerpt<'a> { fn level(&self) -> i32 { 3 } }注意,这里ImportantExcerpt的完整签名是ImportantExcerpt<'a>,生命周期声明也是结构体的一部分所以这里语法类似泛型06 几个不太聪明的例子例子来自course.rsRust的生命周期面对某些特定情况,可能不够聪明,比如:#[derive(Debug)] struct Foo; impl Foo { fn mutate_and_share(&mut self) -> &Self { &*self } fn share(&self) {} } fn main() { let mut foo = Foo; let loan = foo.mutate_and_share(); foo.share(); println!("{:?}", loan); }传入的虽然是可变借用,但是返回时实际上是一个不可变借用,所以理论上来说最终是不可变的share处又进行了一次不可变借用,语义上说,两个不可变借用是没问题的,但是显然编译器不会那么听我的话mut函数的生命周期标注实际上是这样的: fn mutate_and_share<'a>(&'a mut self) -> &'a Self { &'a *self }代表返回值loan和foo的可变借用 &mut self 生命周期相同,所以loan存续期间, &mut self也一直存续虽然本身是个安全的东西,但是现在还没有机制支持,只能按我们标注的情况判定为不合法了。再来看一个#![allow(unused)] fn main() { use std::collections::HashMap; use std::hash::Hash; fn get_default<'m, K, V>(map: &'m mut HashMap<K, V>, key: K) -> &'m mut V where K: Clone + Eq + Hash, V: Default, { match map.get_mut(&key) { Some(value) => value, None => { map.insert(key.clone(), V::default()); map.get_mut(&key).unwrap() } } } } 第一个map.get_mut(&key) 调用完成后,map的可变借用就可以结束了,但是因为我们的声明,这个借用一直存续到函数结束因此map.insert() 会报错,因为产生了另一个可变借用。
2024年02月22日
89 阅读
0 评论
0 点赞
2024-02-22
Rust入门笔记 6. 迭代器方法
在这段时间练习Rust过程中,大量的简单易用的迭代器方法让我觉得非常优美(尽管这么写起来可能一点都不好读),决定单开一篇整理一下迭代器的方法迭代器是什么所有实现了特征Iterator的类型都算迭代器,即使还没用过迭代器,但是只要用过for循环,就隐式的使用过迭代器了。Rust的for循环是一个语法糖,in后面的对象需要是一个迭代器,for循环反复调用迭代器的next,直到Next返回一个None。为什么要用迭代器而不直接用下标?其实Cpp的STL也大量的使用迭代器,在Rust里,我觉得主要原因是迭代器保证了不会出现越界问题。Rust保证安全性问题是有开销的,Vec每次下标访问都会检查边界,如果超出边界则panic我们平时常用的 a..b, a..=b 也都是迭代器,或者说,它们都是Range,同时实现了Iterator迭代器迭代器方法进入正题,以下列举迭代器方法首先是取元素以及遍历方法nextnext方法是迭代器最基本的功能,让迭代器指向第一个对象,返回一个 Option,它不需要参数,只需要一个Selflet a = vec![1, 2]; let mut iter = a.iter(); assert_eq!(Some(&1), iter.next()); assert_eq!(Some(&2), iter.next()); assert_eq!(None, iter.next());enumerate常用python的话对这个也不会陌生,enumerate同时返回迭代次数和对象,返回值为Option<(usize, Self::Item)>(的一个迭代器),相当于同时帮我们记录了迭代次数。比较常用的情况就是同时需要数组值和下标。fn main() { let vec = vec![222, 546, 231, 666, 10]; for (i, val) in vec.iter().enumerate() { println!("{} : {}", i, val); } } 0 : 222 1 : 546 2 : 231 3 : 666 4 : 10for_each()for_each实际上调用的是fold,它接受一个闭包作为参数,并把每次迭代出的值传递给闭包,作为闭包的参数,它没有返回值(或者说返回值是单元Unit())fn main() { (1..=3).for_each(|x| { print!("{}, ", x); }); } // 1, 2, 3, take()take接受一个参数k,表示取迭代器的前k个元素,返回的是一个被截断过的迭代器这个方法会获取所有权fn main() { (1..=10).take(4).for_each(|x| { print!("{}, ", x); }); } // 1, 2, 3, 4, nth()、nth_back()返回的是剩余元素中第k个(倒数第k)位置的元素,并跳到(k的)下一个位置,中间所有的元素都会被消费fn main() { let vec = vec![222, 546, 231, 666, 10]; let mut iter = vec.iter(); println!("{:?}", iter.nth(2)); println!("{:?}", iter.nth(0)); assert_eq!(iter.nth(0), Some(&10)); assert_eq!(iter.nth(0), None); } // Some(231) // Some(666)fn main() { let vec = vec![222, 546, 231, 666, 10]; let mut iter = vec.iter(); println!("{:?}", iter.nth_back(2)); println!("{:?}", iter.nth(1)); assert_eq!(iter.nth(0), None); } // Some(231) , 此时231, 666, 10都被消费了,iter还在开头 // Some(546)last()只取迭代器的最后一个元素,会消费整个迭代器。当然,返回的也是Option\<Self::Item\>以下是变换方法rev()翻转整个迭代器vec![0, 1, 2, 3, 4].iter().rev().for_each(|x|print!("{x}, ")); // 4, 3, 2, 1, 0, skip(k)跳过k个元素,相当于take的补assert_eq!(vec![2,3], (0..4).skip(2).collect::<Vec<_>>());step_by(k)调整迭代器步长//0, 2, 4, 6, 8, 10, (0..=10).step_by(2).for_each(|x| print!("{x}, "));zip()把两个迭代器合并为一个元组的迭代器let it = [1,3,5].iter().zip([2,4,6].iter()); it : vec![(&1,&2),(&3,&4),(&5,&6)]filter()接受一个闭包,把元素一个个传给闭包,该闭包返回一个个bool,留下所有闭包返回true的元素,组成一个迭代器(0..=10).filter(|x| x % 3 == 0) // 还是个迭代器,[0, 3, 9]collect()我意识到我要先说collect,不然只能用for说明结果还是迭代器collect把一个迭代器的元素组成一个新的集合,具体组合成啥由程序员指定或者由编译器推导(Rust的推导能力我觉得还是很强大的)let vec = (0..=100).collect::<Vec<_>>(); // 默认i32,0-100的一个vecmax()、sum()、min()、count()找到迭代元素里的最大值和求和(最小值、元素个数),sum需要指定结果类型let vec = vec![1, 5, 3, 4, 2]; let max = vec.iter().max().unwrap(); // 5 let sum = vec.iter().sum()::<i32>(); // 15all()、any()接受一个闭包,和filter一样的,区别是这两个是返回一个bool而不是一个迭代器,分别代表是否所有元素都满足闭包、是否由元素满足闭包find()接受一个和filter一样的闭包,区别是filter返回全部的迭代器,find返回第一个满足条件的元素map()、filter_map()map()接受一个闭包,把元素逐个传进闭包,把闭包的返回值组成一个迭代器。对于filter_map()来说,不会把None的东西压进迭代器,实现filter的能力fn main() { let numbers = vec![1, 2, 3, 4, 5]; // 使用 filter_map 方法将所有偶数映射为它们的平方根 let results: Vec<_> = numbers.iter().filter_map(|&x| { if x % 2 == 0 { Some((x, (x as f64).sqrt())) } else { None } }).collect(); println!("Filtered and mapped: {:?}", results); }flat_map()flat_map也是接受一个闭包,不过返回的是一系列迭代器,并把他们拍扁成一个迭代器fn main() { let words = vec!["hello", "world", "rust"]; let chars: Vec<_> = words.iter().flat_map(|&word| word.chars()).collect(); println!("{:?}", chars); // ['h', 'e', 'l', 'l', 'o', 'w', 'o', 'r', 'l', 'd', 'r', 'u', 's', 't'] }partition()根据闭包的布尔条件分拆成两个集合fn main() { let numbers = vec![1, 2, 3, 4, 5]; // 使用 partition 方法将奇数和偶数分开 let (even, odd): (Vec<_>, Vec<_>) = numbers .into_iter() .partition(|&x| x % 2 == 0); println!("Even numbers: {:?}", even); // [2, 4] println!("Odd numbers: {:?}", odd); // [1, 3, 5] } scan()、fold()scan和fold都接受两个参数,分别是初始值和一个闭包,传入闭包两个参数:累加值和当前元素区别是scan会把每次闭包返回的结果放一起,最后生成一个迭代器,而fold返回最终的结果fn main() { let numbers = vec![1, 2, 3, 4, 5]; // 使用 scan 方法计算累加和 let cumulative_sum: Vec<_> = numbers.iter().scan(0, |acc, &x| { *acc += x; Some(*acc) }).collect(); println!("{:?}", cumulative_sum); // [1, 3, 6, 10, 15] } fn main() { let numbers = vec![1, 2, 3, 4, 5]; let sum = numbers.iter().fold(0, |acc, &x| acc + x); println!("Sum: {}", sum); // 输出 15 }
2024年02月22日
71 阅读
0 评论
0 点赞
1
2
...
5