检查Rust中所有向量的长度是否相同

Question

给出一些值为T的向量矢量，即。 Vec<Vec<T>>。

检查内部向量是否具有相同长度的惯用方法是什么？ （没有外部依赖）

也就是说，如果所有内部向量具有相同的长度，则为true，否则为false。

Answer 1

您可以使用all方法检查迭代器的所有元素是否与谓词匹配。然后只比较列表中的第一个元素。

fn main() {
    let vec_of_vecs = vec![
        vec![1, 2, 3],
        vec![1, 2, 3],
        vec![1, 2, 3],
        vec![1, 2, 3],
        vec![1, 2, 3],
        vec![1, 2, 3, 4], // remove this to prove that it works for both cases
    ];
    let all_same_length = vec_of_vecs
        .iter()
        .all(|ref v| v.len() == vec_of_vecs[0].len());

    if all_same_length {
        println!("They're all the same");
    } else {
        println!("They are not the same");
    }
}

Answer 2

在我看来，另一个更通用和惯用的解决方案：

fn all_eq_len<'a, T, E: 'a>(collection: T) -> bool
where
    T: IntoIterator<Item = &'a Vec<E>>,
{
    let mut iter = collection.into_iter();
    if let Some(first) = iter.next() {
        let len = first.len();
        iter.all(|v| v.len() == len)
    } else {
        true
    }
}

当然还有使用itertools：

use itertools::Itertools;

vec_of_vecs.iter().map(|v| v.len()).all_equal()