Skip to content

Commit 1571e2e

Browse files
committed
增加新文章:「使用Rust的Iterator解决FizzBuzz问题」
1 parent 2dd1e08 commit 1571e2e

File tree

1 file changed

+237
-0
lines changed

1 file changed

+237
-0
lines changed
Lines changed: 237 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,237 @@
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

Comments
 (0)