创建绑定到整数类型的泛型函数

问题描述 投票:1回答:1

为了学习防锈,我从一些基本练习开始。我写了一个简单的函数,希望能用惯用的锈来计数整数中设置的位数。

fn bit_count(x: u32) -> u32 {
    (0..32).into_iter().map(|i| (x >> i) & 1).sum()
}

fn main() {
    println!("{} has {} set bits.", 5, bit_count(5));
}

现在,我想使函数通用,以便可以传递任何整数类型:i32u32i64u64 ...等

我对C ++中的tmp非常熟悉,但是我对rust泛型的尝试失败了,到目前为止,我有这个:

extern crate num;

fn bit_count<T>(x: T) -> T
where
    T: num::Integer + std::ops::BitAnd + std::ops::Shr + num::NumCast,
    std::ops::Range<T>: std::iter::IntoIterator,
{
    (T::zero()..num::NumCast::from(32).unwrap())
        .into_iter()
        .map(|i| (x >> num::NumCast::from(i)) & T::one())
        .sum()
}

fn main() {
    println!("{} has {} set bits.", 5, bit_count(5));
}

[我看到了num条板箱做广告,看起来很合适。我原本希望有T: num::Integer并完成,但是我觉得我在这里钻一个兔子洞,似乎无法获得正确的边界组合。

任何建议都会很棒!以及使我的代码更加惯用的任何技巧也将有所帮助,谢谢。

generics rust traits
1个回答
0
投票

最后到达那里。事实证明,我需要使用num::PrimInt特性作为绑定,因为它包含所有按位运算和强制转换。 num::Integer的约束较少,仅在数学意义上为整数建模,因此无需按位运算。

我拥有的最终代码如下:

extern crate num;

fn bit_count<T>(x: T) -> T
where
    T: num::PrimInt + std::iter::Sum,
{    
    let n_bits = std::mem::size_of::<T>() * 8;
    (0..n_bits).into_iter().map(|i| (x >> i) & T::one()).sum()

}

fn main() {
    println!("{} has {} set bits.", 5, bit_count(5u32));
    println!("{} has {} set bits.", 5, bit_count(5i32));
    println!("{} has {} set bits.", 5, bit_count(5i64));
}

不需要T::one()很好,但是似乎没有办法解决。另外,在我的范围内需要std::iter::Sum特性以允许正常工作流程。

num板条箱实际上也具有对设置的位数num::PrimInt::count_ones进行计数的功能。

© www.soinside.com 2019 - 2024. All rights reserved.