Rust中无查表无分支高效生成n个连续置1低位整数的方法

原实现的报错根源 #

在固定位宽的整数上做移位操作时，Rust会严格校验移位偏移量：N位宽度的整数，偏移量大于等于自身位宽就会直接触发溢出panic——这是对硬件层面未定义移位行为的主动防护，不管左移还是右移，只要偏移量越界，就会触发同样的校验逻辑。
最初写的(1 << n) -1、尝试替代的u8::MAX >> (8 -n)，本质都是在u8的位宽内执行偏移量等于位宽（n=0或n=8时偏移量为8）的操作，自然会触发panic。

无分支、无查表的最简实现 #

完全不需要依赖checked_shl的分支判断，只要把移位操作放到更宽的整数类型上执行，就能自然规避移位越界的问题：

fn conv(n: u8) -> u8 {
    ((1u16 << n) - 1) as u8
}

实现原理 #

• 移位操作的执行载体换成16位宽度的u16，题目中n的有效取值范围是0到8，对应的左移偏移量最大为8，远小于u16的16位宽度上限，完全符合Rust的移位校验规则，不会触发任何运行时panic。
• 计算完成后将16位的结果截断为u8类型：16位整数计算出的低8位，正好就是需要的低n位全1的掩码值，高位的无效部分会在类型转换时直接丢弃，不影响结果正确性。

如果需要兼容任意u8类型的输入、不依赖调用方传入合规的n值，只要把普通左移替换为包装移位即可，全程依然保持无分支特性：

fn conv(n: u8) -> u8 {
    ((1u16.wrapping_shl(n as u32)) - 1) as u8
}

边界值验证 #

对照题目给出的转换规则，逐值验证结果完全匹配：

• n=0: (1u16 << 0) -1 =0 → 00000000
• n=1: (1u16 <<1) -1 =1 →00000001
• n=2: 3 →00000011
• n=3: 7 →00000111
• n=4: 15 →00001111
• n=5: 31 →00011111
• n=6: 63 →00111111
• n=7: 127 →01111111
• n=8: (1u16 <<8) -1 =255 →11111111

方案优势 #

• 全程没有分支判断，没有查表，编译器可以直接将代码优化为和硬件原生移位、减法、类型截断等价的指令，执行效率远高于带match分支的checked_shl实现。
• 对于超出有效范围的n值（比如n>8），计算结果会自动截断为u8的最大值255，符合位掩码场景下的直觉预期，不会因为输入超界出现非预期的错误值。

内容的提问来源于stack exchange，提问作者puritii