如何向Rust编译器动态告知指定变量为非零值？

针对你的需求，有两种核心方案可以在确保变量非零的前提下，让编译器消除不必要的边界检查，且不会引入额外运行时开销：

方案1：使用std::hint::assert_unchecked（Rust 1.70+） #

这个函数是给编译器的无运行时检查断言，直接告诉编译器某个条件必然成立，编译器会基于此优化代码，不会生成任何检查逻辑。

use std::hint::assert_unchecked;

pub fn f(n: u64) -> u32 {
    // 必须确保n绝对不为0，否则会触发未定义行为（UB）
    unsafe {
        assert_unchecked(n != 0);
    }
    n.trailing_zeros()
}

方案2：无检查转换为NonZeroU64 #

利用NonZeroU64::new_unchecked完成无检查的类型转换，后续使用NonZeroU64的方法时，编译器会默认其值非零，从而跳过边界检查。

use std::num::NonZeroU64;

pub fn f(n: u64) -> u32 {
    // 同样必须保证n≠0，否则会触发UB
    let nz_n = unsafe { NonZeroU64::new_unchecked(n) };
    nz_n.trailing_zeros()
}

关键注意事项 #

• 两种方案都属于unsafe操作，必须100%确保代码路径中变量非零，一旦实际值为0，会导致未定义行为（程序崩溃、内存损坏等不可预测后果）。
• 不要用普通assert!或debug_assert!，它们会生成运行时检查代码，违背你消除检查的初衷。
• 如果变量存在为零的可能，仅在确定非零的代码分支中使用这些unsafe逻辑，比如：

  pub fn process(n: u64) -> Option<u32> {
      if n == 0 {
          None
      } else {
          Some(unsafe {
              NonZeroU64::new_unchecked(n).trailing_zeros()
          })
      }
  }
  

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