引入#
最近 b 站上很多小伙伴对何同学的错误代码进行了一些很有意思的二创,一度登上了 Github Trending:
其中看到有人使用 C++ 中的宏实现了何同学的 “或运算”,于是受到启发,使用 Rust 中的宏实现了一下。
Source Code#
fn parse_to_vec(nums: &str) -> Vec<usize> {
let nums = nums.replace(' ', "");
nums.split('|').map(|x| x.parse().unwrap()).collect()
}
fn power_con(powers: &mut [u8], nums: &[usize], power: u8) {
for &num in nums {
powers[num] = power;
}
}
#[macro_export]
macro_rules! powerCon {
($powers: expr, $nums: expr, $force: expr) => {
power_con($powers, &parse_to_vec(stringify!($nums)), $force)
};
}
fn main() {
let mut powers = [0_u8; 68];
powerCon!(&mut powers, 0, 100);
powerCon!(&mut powers, 1 | 2 | 6 | 7 | 11 | 52 | 57 | 58 | 65, 10);
println!("{:?}", powers)
}