如何将Rust中的[u8; 32]类型转换为二进制并统计前导零

问题分析 #

你通过Sha256哈希函数得到了[u8;32]类型的哈希值，需要统计其二进制形式的前导零个数，但之前的转换代码存在问题——用{:b}格式化u8时，不会自动补全8位（比如0x00会被转为"0"而非"00000000"），导致拼接后的二进制字符串长度不足256位，统计结果错误。

解决方案 #

方法一：生成完整二进制字符串后统计 #

先修正二进制转换函数，确保每个字节都被格式化为8位的二进制字符串，再统计前导零：

pub fn to_bits(arr: &[u8; 32]) -> String {
    let mut final_str = String::with_capacity(256); // 预分配256位容量提升效率
    for &byte in arr {
        final_str.push_str(&format!("{:08b}", byte)); // {:08b}确保补全8位
    }
    final_str
}

// 统计前导零
pub fn count_leading_zeros(arr: &[u8; 32]) -> usize {
    let bits = to_bits(arr);
    bits.chars().take_while(|&c| c == '0').count()
}

方法二：直接操作字节（更高效） #

无需生成完整字符串，直接遍历字节计算前导零：

1. 遍历哈希数组，遇到全零字节则累加8个零位
2. 遇到第一个非零字节时，计算该字节的前导零个数并累加，随后终止遍历（后续字节无需处理）

pub fn count_leading_zeros(arr: &[u8; 32]) -> usize {
    let mut zero_count = 0;
    for &byte in arr {
        if byte == 0 {
            zero_count += 8;
        } else {
            // 利用u8原生方法leading_zeros获取当前字节的前导零个数
            zero_count += byte.leading_zeros() as usize;
            break;
        }
    }
    zero_count
}

第二种方法避免了字符串的内存分配与拼接，性能更优，适合处理大量哈希值的场景。

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