HiD*_*der 4 sorting floating-point rust
我正在根据两个标准对向量进行排序。第一个是浮点数NaN
,第二个是字符串,用于按字典顺序断开关系。
vec.sort_by(|a, b| {
match (foo(a) as f64 / bar(a) as f64).partial_cmp(&(foo(b) as f64 / bar(b) as f64)) {
Some(x) => {
Ordering::Equal => name(a).cmp(name(b)),
other => other,
}
None() => {
//Not sure what to put here.
}
}
}
Run Code Online (Sandbox Code Playgroud)
foo(a)
返回 int > 0,
bar(a)
返回 int >= 0,
name(a)
返回& String
.
如何排序NaN
,使其大于任何其他数字,并等于任何其他数字NaN
(字典顺序决胜局)?
您已经知道如何处理平局,您所需要的只是以所需的方式比较浮点数。只需...编写您所描述的代码:
use std::cmp::Ordering;
use std::f32;
fn main() {
let mut vec = [91.0, f32::NAN, 42.0];
vec.sort_by(|&a, &b| {
match (a.is_nan(), b.is_nan()) {
(true, true) => Ordering::Equal,
(true, false) => Ordering::Greater,
(false, true) => Ordering::Less,
(false, false) => a.partial_cmp(&b).unwrap(),
}
});
println!("{:?}", vec);
}
Run Code Online (Sandbox Code Playgroud)
您可以奇特地将其包装在一个也代表密钥的结构中:
use std::cmp::Ordering;
use std::f32;
fn main() {
let mut vec = [91.0, f32::NAN, 42.0];
vec.sort_by_key(|&a| MyNanKey(a));
println!("{:?}", vec);
}
#[derive(Debug, Copy, Clone, PartialEq)]
struct MyNanKey(f32);
impl Eq for MyNanKey {}
impl PartialOrd for MyNanKey {
fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
Some(self.cmp(other))
}
}
impl Ord for MyNanKey {
fn cmp(&self, other: &Self) -> Ordering {
match (self.0.is_nan(), other.0.is_nan()) {
(true, true) => Ordering::Equal,
(true, false) => Ordering::Greater,
(false, true) => Ordering::Less,
(false, false) => self.0.partial_cmp(&other.0).unwrap(),
}
}
}
Run Code Online (Sandbox Code Playgroud)
我没有考虑这是否适用于各种无穷大或非规范化浮点值,所以要小心。