什么是最简单的链式比较方法,产生第一个不相等的方法?

时间:2016-08-23 17:09:58

标签: syntax comparison rust

将数据与优先级进行比较,对于具有可以比较的多个成员的结构,或者对于sort_by回调来说,这是很常见的。

// Example of sorting a: Vec<[f64; 2]>, sort first by y, then x,
xy_coords.sort_by(
    |co_a, co_b| {
        let ord = co_a[1].cmp(&co_b[1]);
        if ord != std::cmp::Ordering::Equal {
            ord
        } else {
            co_a[0].cmp(&co_b[0])
        }
    }
);

是否有更简单的方法来执行多个cmp函数,其中只返回第一个不相等的结果?

1 个答案:

答案 0 :(得分:6)

  

执行多个cmp函数,其中只返回第一个不相等的结果

这基本上是为元组定义Ord的方式。创建一个函数,将您的类型转换为元组并比较它们:

fn main() {
    let mut xy_coords = vec![[1, 0], [-1, -1], [0, 1]];

    fn sort_key(coord: &[i32; 2]) -> (i32, i32) {
        (coord[1], coord[0])
    }

    xy_coords.sort_by(|a, b| {
        sort_key(a).cmp(&sort_key(b))
    });
}

由于这很常见,因此有一种方法:

xy_coords.sort_by_key(sort_key);

这对你的情况没有帮助,因为floating point doesn't implement Ord

许多可能性之一是杀死NaN上的程序:

xy_coords.sort_by(|a, b| {
    sort_key(a).partial_cmp(&sort_key(b)).expect("Don't know how to handle NaN")
});

另见

  

有些时候您可能不想创建一个大元组来比较将被忽略的值,因为较高优先级的值会提前退出比较。

Guava's ComparisonChain窃取页面,我们可以创建一个小型构建器,允许我们使用闭包来避免额外的工作:

use std::cmp::Ordering;

struct OrdBuilder<T> {
    a: T,
    b: T,
    ordering: Ordering,
}

impl<T> OrdBuilder<T> {
    fn new(a: T, b: T) -> OrdBuilder<T> {
        OrdBuilder {
            a: a,
            b: b,
            ordering: Ordering::Equal,
        }
    }

    fn compare_with<F, V>(mut self, mut f: F) -> OrdBuilder<T>
        where F: for <'a> FnMut(&'a T) -> V,
              V: Ord,
    {
        if self.ordering == Ordering::Equal {
            self.ordering = f(&self.a).cmp(&f(&self.b));
        }
        self
    }

    fn finish(self) -> Ordering {
        self.ordering
    }
}

这可以像

一样使用
struct Thing {
    a: u8,
}

impl Thing {
    fn b(&self) -> u8 {
        println!("I'm slow!");
        42
    }
}

fn main() {
    let a = Thing { a: 0 };
    let b = Thing { a: 1 };

    let res = OrdBuilder::new(&a, &b)
        .compare_with(|x| x.a)
        .compare_with(|x| x.b())
        .finish();

    println!("{:?}", res);
}