rust的迭代器方法——collect

概述: 它能够将一个迭代器转换为各种集合类型,例如向量、哈希集、哈希映射等,能够将迭代器的所有元素收集到一个集合中

常见的用法:

将迭代器转换为Vec:

rust 复制代码
let Vec: Vec<i32> = (1..5).collect();

将迭代器转换为HashSet:

rust 复制代码
use std::collections::HashSet;

let set: HashSet<i32> = vec![1,2,3,2].into_iter().collect();

将迭代器转换为HashMap:

rust 复制代码
use std::collections::HashMap;

let map: HashMap<&str, i32> = vec![("a", 1), ("b", 2), ("c", 3)].into_iter().collect();

collect的类型推导:

collect是泛型的,通常需要通过上下文推导出类型目标,但是有时,如果没有足够的上下文,rust无法推导出目标类型,可能需要显式指定。

rust 复制代码
let numbers = (1..4).collect::<Vec<i32>>();

内部工作原理:

collect之所以通用,是因为它依赖于FromIterator trait,该trait定义了如何从一个迭代器构建某个集合类型。Vec、HashMap、HashSet等类型都实现了FromIterator

FromIterator定义如下:

rust 复制代码
pub trait FromIterator<A> {
    fn from_iter<T: IntoIterator<Item = A>>(iter: T) -> Self;
}

常见的使用场景:

  1. 可以使用filter_map结合collect,从一个可能包含None的迭代器中收集有效的值

    rust 复制代码
    let values = vec![Some(1), None, Some(2), Some(3)];
    let result: Vec<i32> = values.into_iter().filter_map(|x| x).collect();
    println!("{:?}", result); // 输出: [1, 2, 3]
  2. 去重和排序:将元素收集到HashMap中去重,再收集到Vec来排序

    rust 复制代码
    use std::collections::HashSet;
    
    let vec = vec![3, 1, 2, 1, 3, 4];
    let mut set: HashSet<_> = vec.into_iter().collect();
    let mut vec: Vec<_> = set.into_iter().collect();
    vec.sort();
    println!("{:?}", vec); // 输出: [1, 2, 3, 4]
  3. 将字符迭代器收集为字符串:

    rust 复制代码
    let chars = vec!['R', 'u', 's', 't'];
    let s: String = chars.into_iter().collect();
    println!("{}", s); // 输出: Rust

性能考虑:

collect通常是高效的,因为它能够预知迭代器的大小并进行优化,提前分配足够的空间,避免不必要的内存重新分配。此外,对于较大的数据集,rust还可以通过并行化迭代器(rayon crate)来进一步提升性能。

相关推荐
Gofarlic_oms13 小时前
利用API实现ANSYS许可证管理自动化集成
运维·服务器·开发语言·matlab·自动化·负载均衡
AI+程序员在路上4 小时前
VS Code 完全使用指南:下载、安装、核心功能与 内置AI 编程助手实战
开发语言·人工智能·windows·开源
invicinble5 小时前
这里对java的知识体系做一个全域的介绍
java·开发语言·python
catchadmin5 小时前
使用 PHP TrueAsync 改造 Laravel 协程异步化的可行路径
开发语言·php·laravel
wbs_scy5 小时前
【Linux 线程进阶】进程 vs 线程资源划分 + 线程控制全详解
java·开发语言
AI人工智能+电脑小能手5 小时前
【大白话说Java面试题】【Java基础篇】第15题:JDK1.7中HashMap扩容为什么会发生死循环?如何解决
java·开发语言·数据结构·后端·面试·哈希算法
郑州光合科技余经理6 小时前
同城O2O海外版二次开发实战:从支付网关到配送算法
开发语言·前端·后端·算法·架构·uni-app·php
南子北游7 小时前
Python学习(基础语法1)
开发语言·python·学习
张健11564096487 小时前
使用信号量限制并发数量
开发语言·c++
糯米团子7497 小时前
Web Worker
开发语言·前端·javascript