Rust函数返回值元组时出现类型不匹配错误

你的函数代码：

fn get_two_bytes(data: &[u8]) -> (Vec<(u8, u8)>, BTreeSet<(u8, u8)>) {
    let two_byte_vec = data
        .chunks(2)
        .map(|x| (x[0], x[1]))
        .collect::<Vec<_>>();

    let mut two_byte_set : BTreeSet<&(u8, u8)> = BTreeSet::new();
    for n in &two_byte_vec {
        two_byte_set.insert(n);
    }

    return (two_byte_vec, two_byte_set);
}

运行时出现的类型不匹配错误：

error[E0308]: mismatched types
  --> src/lib.rs:13:27
   |
13 |     return (two_byte_vec, two_byte_set);
   |                           ^^^^^^^^^^^^ expected tuple, found `&(u8, u8)`
   |
   = note: expected struct `BTreeSet<(u8, u8)>`
              found struct `BTreeSet<&(u8, u8)>`

解决方法 #

问题核心是two_byte_set的类型是BTreeSet<&(u8, u8)>（存储元组的引用），但函数签名要求返回BTreeSet<(u8, u8)>（存储拥有所有权的元组）。由于(u8, u8)实现了Copy trait，只需在插入时解引用元素即可复制出拥有所有权的元组：

use std::collections::BTreeSet;

fn get_two_bytes(data: &[u8]) -> (Vec<(u8, u8)>, BTreeSet<(u8, u8)>) {
    let two_byte_vec = data
        .chunks(2)
        .map(|x| (x[0], x[1]))
        .collect::<Vec<_>>();

    let mut two_byte_set = BTreeSet::new();
    // 解引用&(u8,u8)得到(u8,u8)，利用Copy特性复制元素
    for n in &two_byte_vec {
        two_byte_set.insert(*n);
    }

    (two_byte_vec, two_byte_set)
}

也可以用更简洁的链式调用生成集合，避免手动循环：

use std::collections::BTreeSet;

fn get_two_bytes(data: &[u8]) -> (Vec<(u8, u8)>, BTreeSet<(u8, u8)>) {
    let two_byte_vec = data
        .chunks(2)
        .map(|x| (x[0], x[1]))
        .collect::<Vec<_>>();

    let two_byte_set = two_byte_vec.iter().copied().collect();

    (two_byte_vec, two_byte_set)
}

这里iter()生成引用迭代器，copied()会自动将&(u8,u8)转换为(u8,u8)（因为类型实现了Copy），最后直接collect成BTreeSet<(u8, u8)>。

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