|
| 1 | ++++ |
| 2 | +title = "使用Rust的Iterator优雅解决FizzBuzz问题" |
| 3 | +date = 2024-09-18T22:46:00-07:00 |
| 4 | +lastmod = 2024-09-18T23:53:02-07:00 |
| 5 | +tags = ["rust"] |
| 6 | +categories = ["rust"] |
| 7 | +draft = false |
| 8 | +toc = true |
| 9 | ++++ |
| 10 | + |
| 11 | +## <span class="section-num">1</span> 前言 {#前言} |
| 12 | + |
| 13 | +按照维基百科的说法,[FizzBuzz问题](https://en.wikipedia.org/wiki/Fizz_buzz) 是一个简单但是常见的面试编程问题(可能以前常见,现在都是考Leetcode了,这种连Easy 都不算了),这个问题的要求如下: <br/> |
| 14 | + |
| 15 | +1. 写一个程序,输出从1到100的数字 <br/> |
| 16 | +2. 对于3的倍数,不输出数字,而是输出 "Fizz" <br/> |
| 17 | +3. 对于5的倍数,不输出数字,而是输出 "Buzz" <br/> |
| 18 | +4. 对于即是3的倍数又是5的倍数的数字(即15的倍数),打印 "FizzBuzz" <br/> |
| 19 | + |
| 20 | + |
| 21 | +## <span class="section-num">2</span> 常规解法 {#常规解法} |
| 22 | + |
| 23 | +问题非常简单,刚学编程的学生都可以写出符合要求的代码,下面是 Rust 的常规解法: <br/> |
| 24 | + |
| 25 | +```rust |
| 26 | +fn main() { |
| 27 | + for i in 0..=100 { |
| 28 | + if i % 3 == 0 && i % 5 == 0 { |
| 29 | + println!("FizzBuzz"); |
| 30 | + } else if i % 3 == 0 { |
| 31 | + println!("Fizz"); |
| 32 | + } else if i % 5 == 0 { |
| 33 | + println!("Buzz"); |
| 34 | + } else { |
| 35 | + println!("{i}"); |
| 36 | + } |
| 37 | + } |
| 38 | +} |
| 39 | +``` |
| 40 | + |
| 41 | +这个没有什么太多可说的,就是直接按需求翻译代码了。 <br/> |
| 42 | + |
| 43 | + |
| 44 | +## <span class="section-num">3</span> Iterator 解法 {#iterator-解法} |
| 45 | + |
| 46 | +如果现在给 FizzBuzz 问题再加一个限制,不能使用乘法,除法,或者取模操作,那么又要怎么实现呢? <br/> |
| 47 | + |
| 48 | +Rust 标准库中的各式 `Iterator` 可以算是Rust零开销抽象(Zero Cost Abstraction)与表达能力的最佳体现了。 <br/> |
| 49 | + |
| 50 | +最近在读 Programming Rust, 2nd edition, 里面就有使用各种 Iterator 组合,不使用除法或者取模操作来解决 FizzBuzz 问题的实现, 可以说是把 `iterator` 玩得非常花了: <br/> |
| 51 | + |
| 52 | +```rust |
| 53 | +use std::iter::{once, repeat}; |
| 54 | + |
| 55 | +fn main() { |
| 56 | + let fizzes = repeat("").take(2).chain(once("fizz")).cycle(); |
| 57 | + let buzzes = repeat("").take(4).chain(once("buzz")).cycle(); |
| 58 | + let fizzes_buzzes = fizzes.zip(buzzes); |
| 59 | + |
| 60 | + let fizz_buzz = (1..=100).zip(fizzes_buzzes).map(|tuple| match tuple { |
| 61 | + (i, ("", "")) => i.to_string(), |
| 62 | + (_, (fizz, buzz)) => format!("{}{}", fizz, buzz), |
| 63 | + }); |
| 64 | + |
| 65 | + for line in fizz_buzz { |
| 66 | + println!("{line}") |
| 67 | + } |
| 68 | +} |
| 69 | +``` |
| 70 | + |
| 71 | +看起来是否不知道所云呢? 现在可以把每个 `iterator` 的作用逐一拆解。 <br/> |
| 72 | + |
| 73 | + |
| 74 | +### <span class="section-num">3.1</span> repeat + take {#repeat-plus-take} |
| 75 | + |
| 76 | +`repeat` 的作用就是无限重复某个传入的元素, 例如 `repeat(4)` 就是生成无限个数字4, `repeat("")` 就是生成无限个空白字符. <br/> |
| 77 | + |
| 78 | +虽然 `repeat` 能生成无限个指定的元素,但是我只想要若干个元素,怎么整呢? `take` 就可以满足这个要求,所以 `repeat(4).take(4)` 就是生成4个数字4的意思,而 `repeat("").take(2)` 就是生成2个空字符 <br/> |
| 79 | + |
| 80 | +```rust |
| 81 | +use std::iter; |
| 82 | + |
| 83 | +// that last example was too many fours. Let's only have four fours. |
| 84 | +let mut four_fours = iter::repeat(4).take(4); |
| 85 | + |
| 86 | +assert_eq!(Some(4), four_fours.next()); |
| 87 | +assert_eq!(Some(4), four_fours.next()); |
| 88 | +assert_eq!(Some(4), four_fours.next()); |
| 89 | +assert_eq!(Some(4), four_fours.next()); |
| 90 | + |
| 91 | +// ... and now we're done |
| 92 | +assert_eq!(None, four_fours.next()); |
| 93 | +``` |
| 94 | + |
| 95 | + |
| 96 | +### <span class="section-num">3.2</span> once {#once} |
| 97 | + |
| 98 | +有生成无限个元素的 `iterator=,自然就有只生成一个元素的 =iterator`, 那就是 `once()`, 这个 `iterator` 只会返回一个指定的元素。 <br/> |
| 99 | + |
| 100 | +所以 `once("fizz")` 就是创建一个只会返回一个 "fizz" 的 `iterator` : <br/> |
| 101 | + |
| 102 | +```rust |
| 103 | +use std::iter; |
| 104 | + |
| 105 | +// one is the loneliest number |
| 106 | +let mut one = iter::once(1); |
| 107 | + |
| 108 | +assert_eq!(Some(1), one.next()); |
| 109 | + |
| 110 | +// just one, that's all we get |
| 111 | +assert_eq!(None, one.next()); |
| 112 | +``` |
| 113 | + |
| 114 | + |
| 115 | +### <span class="section-num">3.3</span> chain {#chain} |
| 116 | + |
| 117 | +顾名思义,就是把两个 iterator 像链子一样串起来, 合并成一个 iterator: <br/> |
| 118 | + |
| 119 | +```rust |
| 120 | +use std::iter::chain; |
| 121 | + |
| 122 | +let a = [1, 2, 3]; |
| 123 | +let b = [4, 5, 6]; |
| 124 | + |
| 125 | +let mut iter = chain(a, b); |
| 126 | + |
| 127 | +assert_eq!(iter.next(), Some(1)); |
| 128 | +assert_eq!(iter.next(), Some(2)); |
| 129 | +assert_eq!(iter.next(), Some(3)); |
| 130 | +assert_eq!(iter.next(), Some(4)); |
| 131 | +assert_eq!(iter.next(), Some(5)); |
| 132 | +assert_eq!(iter.next(), Some(6)); |
| 133 | +assert_eq!(iter.next(), None); |
| 134 | +``` |
| 135 | + |
| 136 | + |
| 137 | +### <span class="section-num">3.4</span> circle {#circle} |
| 138 | + |
| 139 | +`circle` 就比较有趣了,它的作用是无限循环一个 `iterator`, `repeat` 循环一个元素,而 `circle` 是循环一个 iterator: <br/> |
| 140 | + |
| 141 | +```rust |
| 142 | +let dirs = ["North", "East", "South", "West"]; |
| 143 | +let mut spin = dirs.iter().cycle(); |
| 144 | +assert_eq!(spin.next(), Some(&"North")); |
| 145 | +assert_eq!(spin.next(), Some(&"East")); |
| 146 | +assert_eq!(spin.next(), Some(&"South")); |
| 147 | +assert_eq!(spin.next(), Some(&"West")); |
| 148 | +assert_eq!(spin.next(), Some(&"North")); |
| 149 | +assert_eq!(spin.next(), Some(&"East")); |
| 150 | +``` |
| 151 | + |
| 152 | +把4个 iterator 组合起来的 `repeat("").take(2).chain(once("fizz")).cycle();` 表达式的意思就是, 返回一个 iterator, 这个 iterator 无限循环: `"" "" "fizz" "" "" "fizz" ...` <br/> |
| 153 | + |
| 154 | + |
| 155 | +### <span class="section-num">3.5</span> zip {#zip} |
| 156 | + |
| 157 | +`zip` iterator 的含义就是 "zips up", 翻译过来就是拉上拉链,它的作用就是把两个 `iterator` 像拉链一样拉起来,返回一个 iterator,用代码来解释会更直观: <br/> |
| 158 | + |
| 159 | +```rust |
| 160 | +let a1 = [1, 2, 3]; |
| 161 | +let a2 = [4, 5, 6]; |
| 162 | + |
| 163 | +let mut iter = a1.iter().zip(a2.iter()); |
| 164 | + |
| 165 | +assert_eq!(iter.next(), Some((&1, &4))); |
| 166 | +assert_eq!(iter.next(), Some((&2, &5))); |
| 167 | +assert_eq!(iter.next(), Some((&3, &6))); |
| 168 | +assert_eq!(iter.next(), None); |
| 169 | +``` |
| 170 | + |
| 171 | +`zip` 就是把 `a1` 和 `a2` 两个iterator 「拉起来」了,每次返回一对的元素. 所以 =fizzes.zip(buzzes)=,就是合并了两个 iterator : <br/> |
| 172 | + |
| 173 | +```rust |
| 174 | +// fizzes: "" "" "fizz" "" "" "fizz" "" "" "fizz" .. |
| 175 | +// buzzes: "" "" "" "" "buzz" "" "" "" "" "buzz" |
| 176 | +// fizzes_buzzes: ("" "") ("" "") ("fizz" "") ("" "") ("" "buzz") ... |
| 177 | +``` |
| 178 | + |
| 179 | +而 `(1..=100).zip(fizzes_buzzes)` 就是创建一个包含三个元素的 tuple: <br/> |
| 180 | + |
| 181 | +```rust |
| 182 | +// (1..=100): 1 2 3 4 5 6 7 ... |
| 183 | +// fizzes_buzzes: ("" "") ("" "") ("fizz" "") ("" "") ("" "buzz") ... |
| 184 | +// (1..=100).zip(fizzes_buzzes): (1 ("" "")) (2 ("" "")) (3 ("fizz" "")) (4 ("" "")) (5 ("" "buzz")) .. |
| 185 | +``` |
| 186 | + |
| 187 | + |
| 188 | +### <span class="section-num">3.6</span> map {#map} |
| 189 | + |
| 190 | +`map` 这个 iterator 在其他语言也有相同的实现,入参是一个闭包函数,然后把每个元素作为入参,调用闭包函数,在新的迭代返回函数的调用结果. <br/> |
| 191 | + |
| 192 | +```rust |
| 193 | +.map(|tuple| match tuple { |
| 194 | + (i, ("", "")) => i.to_string(), |
| 195 | + (_, (fizz, buzz)) => format!("{}{}", fizz, buzz), |
| 196 | +}) |
| 197 | +``` |
| 198 | + |
| 199 | +最核心的是Rust的 pattern matching, 用来匹配不同的值, `(i, ("", ""))` 就是匹配所有 fizz 和 buzz为 `("", "")` 的值,什么情况下 `fizz` 和 `buzz` 会都为 "" 呢,无法整除3以及无法整除5的时候,那么就直接返回数字 `i`; <br/> |
| 200 | + |
| 201 | +`(_, (fizz,buzz))`, `_` 就是通配符,就是匹配掉所有其他的情况,无论是 fizz = "", fizz = "fizz", buzz `"" 或者 buzz = "buzz", 都把返回 "{fizz}{buzz}", 也就是 =(_, (fizz,buzz))` 匹配了4种情况. <br/> |
| 202 | + |
| 203 | +`map` 迭代器返回的是一个 String, 最后再加 String 打印出来. <br/> |
| 204 | + |
| 205 | +同样是解决问题,这个版本的解法肯定是看起来「高大上」得多,说不定能让面试官眼前一亮,又或者是把自己绕晕。 <br/> |
| 206 | + |
| 207 | + |
| 208 | +## <span class="section-num">4</span> Zero Cost Abstraction {#zero-cost-abstraction} |
| 209 | + |
| 210 | +所谓的是零开销抽象(Zero Cost Abstraction),用C++之父的话来解释就是: <br/> |
| 211 | + |
| 212 | +> In general, C++ implementations obey the zero-overhead principle: What you don’t use, you don’t pay for. And further: What you do use, you couldn’t hand code any better. <br/> |
| 213 | +
|
| 214 | +概括来说,就是使用 Iterator 写出来的代码,和你自己 for-loop 手写是性能是一样的,并不会有额外的抽象开销。 <br/> |
| 215 | + |
| 216 | +换个角度讲,你手写的代码也没法实现得比 Iterator 更快,表达力还可能没有那么强。 <br/> |
| 217 | + |
| 218 | +如果看上面的 Iterator 实现觉得着实难以理解,我们可以再来一版兼具优雅与简洁的实现: <br/> |
| 219 | + |
| 220 | +```rust |
| 221 | +fn main() { |
| 222 | + for i in 1..=100 { |
| 223 | + match (i % 3, i % 5) { |
| 224 | + (0, 0) => println!("FizzBuzz"), |
| 225 | + (0, _) => println!("Fizz"), |
| 226 | + (_, 0) => println!("Buzz"), |
| 227 | + (_, _) => println!("{}", i), |
| 228 | + } |
| 229 | + } |
| 230 | +} |
| 231 | +``` |
| 232 | + |
| 233 | + |
| 234 | +## <span class="section-num">5</span> Reference {#reference} |
| 235 | + |
| 236 | +- Programming Rust, 2nd edition <br/> |
| 237 | + |
0 commit comments