为何用clap::Parser读取大数计算比硬编码数值慢很多？

我写了个简单的Rust程序，用clap::Parser接收命令行整数参数做计算，但传大数（比如999999999）时跑了快10秒；把数值硬编码、去掉clap依赖后，只需要2秒左右。

clap版本代码 #

use clap::Parser ;

#[derive(Parser,Default)]
struct Args {
    #[arg(short)]
    number: u128
}

fn sum(n: u128) -> u128 {
    let mut result: u128 = 0;
    for i in 1..n { result += n/i; }
    result 
}

fn main() {
    let args = Args::parse() ;
    println!("{}", sum(args.number)) ;
}

clap版本运行耗时 #

$ time ./target/release/main -n 999999999
20877697533

real    0m9.442s
user    0m9.370s
sys     0m0.030s

硬编码版本代码 #

fn sum(n: u128) -> u128 {
    let mut result: u128 = 0;
    for i in 1..n { result += n/i; }
    result 
}

fn main() {
    let n: u128 = 999999999 ;
    println!("{}", sum(n)) ;
}

硬编码版本运行耗时 #

$ time ./target/release/main
20877697533

real    0m2.398s
user    0m2.236s
sys     0m0.013s

原因分析 #

根本原因不是clap的锅，是编译器对编译期常量和运行期变量的优化力度完全不同：

• 当n是硬编码的固定值时，Rust的编译器（LLVM）能看穿整个循环逻辑，对n/i这个除法做极致优化：因为除数n固定，编译器会提前算出乘法逆元，把除法转成更快的乘法+移位操作，还能对循环做展开、向量化，让循环跑起来快很多。
• 当n是clap解析来的运行期变量时，编译器不知道n的值，只能用通用的u128除法指令。这种通用除法的速度比优化后的乘法慢得多，再加上循环要跑近10亿次，整体耗时自然就上去了。

解决办法 #

推荐：换用O(√n)的高效算法 #

原循环是O(n)复杂度，哪怕编译器优化了，10亿次循环还是慢。可以用数论里的优化方法，把计算复杂度降到O(√n)，循环次数直接从10亿降到3万左右，毫秒级就能出结果：

fn sum(n: u128) -> u128 {
    if n == 0 {
        return 0;
    }
    let sqrt_n = (n as f64).sqrt() as u128;
    let mut result = 0;
    // 计算1到√n的商之和
    for k in 1..=sqrt_n {
        result += n / k;
    }
    // 每个商q对应两个数k和n/k，所以乘以2
    result *= 2;
    // 减去重复计算的√n*√n（因为k=√n时只算一次）
    result -= sqrt_n * sqrt_n;
    // 原函数是算1到n-1的和，所以减去最后一项n/n=1
    result - 1
}

不推荐：手动优化除法（仅作原理演示） #

如果非要用原循环，可以针对运行期的n预计算乘法逆元来加速除法，但这种方法只适用于奇数n，而且复杂度还是O(n)，实际意义不大，这里就不展开了。

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