如何实现支持String、&str与原始类型的泛型素数判断函数？

你的思路方向是对的——利用ToPrimitive trait统一处理不同类型的输入转换，但原代码存在几个关键问题需要修正，以下是可行的优化实现：

原代码的问题点 #

• panic风险：to_i64().unwrap()会在输入无法转换为有效整数时直接panic（比如输入字符串"abc"）
• 边界逻辑错误：
  • 输入值≤1时，函数错误返回true，但这类数不是素数
  • 当输入为2时，循环从2开始直接返回false，但2是最小的素数
• 效率问题：循环遍历范围和终止条件逻辑冗余，没有跳过偶数（除2外）

优化后的实现 #

use num_traits::ToPrimitive;

pub fn is_prime<T>(val: T) -> bool
where
    T: ToPrimitive,
{
    // 1. 尝试转换为i64，转换失败直接返回false
    let Some(val) = val.to_i64() else {
        return false;
    };

    // 2. 处理边界情况：≤1的数不是素数
    if val <= 1 {
        return false;
    }

    // 3. 2是唯一的偶素数
    if val == 2 {
        return true;
    }

    // 4. 偶数直接排除
    if val % 2 == 0 {
        return false;
    }

    // 5. 遍历奇数，只需要检查到平方根即可
    let sqrt_val = (val as f64).sqrt() as i64;
    for i in (3..=sqrt_val).step_by(2) {
        if val % i == 0 {
            return false;
        }
    }

    true
}

关键改进说明 #

• 安全的类型转换：用if let Some(val) = val.to_i64()替代unwrap()，避免panic，转换失败的输入（如非数字字符串）直接判定为非素数
• 完善的边界处理：覆盖了≤1、2、偶数等特殊情况
• 效率优化：只遍历奇数到输入值的平方根，减少循环次数

使用示例 #

fn main() {
    // 测试不同类型输入
    println!("{}", is_prime(2));          // 输出: true
    println!("{}", is_prime(10u32));      // 输出: false
    println!("{}", is_prime(String::from("17"))); // 输出: true
    println!("{}", is_prime("abc"));      // 输出: false
    println!("{}", is_prime(-5));         // 输出: false
}

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