如何用切片模式或向量查询BTreeMap中基于元组首元素的范围

你使用BTreeMap<(String, Vec<i32>), i32>存储数据，希望仅通过元组的第一个String元素进行范围查询，忽略第二个Vec<i32>的内容，现有代码的range写法无法实现需求，以下是几种可行方案：

方案一：利用元组排序特性构造边界 #

由于Rust中元组的比较是按元素顺序依次对比，当第一个元素相等时才会比较第二个元素。因此要获取所有第一个元素为指定字符串的键值对，可以构造左边界为目标字符串+空Vec，右边界为目标字符串的"下一个"字符串+任意Vec（因为目标字符串的"下一个"字符串会比所有以目标字符串为第一个元素的元组更大）。

示例代码：

use std::collections::BTreeMap;

fn main() {
    let mut m: BTreeMap<(String, Vec<i32>), i32> = BTreeMap::new();
    m.insert(("hello".to_string(), vec![12]), 34);
    m.insert(("hello".to_string(), vec![4]), 56);
    m.insert(("other".to_string(), vec![4]), 44);

    // 查询所有第一个元素为"hello"的键值对
    let target = "hello".to_string();
    // 构造右边界：确保大于所有以target为第一个元素的元组
    let next_str = format!("{}z", target);
    for (k, v) in m.range((target.clone(), vec![])..(next_str, vec![])) {
        println!("key: {:?}, value: {}", k, v);
    }

    // 范围查询：第一个元素在"hello"到"other"之间（包含"hello"，不包含"other"）
    let start = "hello".to_string();
    let end = "other".to_string();
    let end_next = format!("{}z", end);
    for (k, v) in m.range((start, vec![])..(end_next, vec![])) {
        println!("range key: {:?}, value: {}", k, v);
    }
}

方案二：自定义键类型并实现Ord trait #

如果希望更直观地忽略第二个元素的比较，可以自定义一个键类型，重写Ord和PartialOrd trait，让比较逻辑仅关注第一个String元素。

示例代码：

use std::collections::BTreeMap;
use std::cmp::{Ordering, PartialOrd};

#[derive(Debug, Clone, PartialEq, Eq)]
struct CustomKey(String, Vec<i32>);

// 仅基于第一个String元素实现比较逻辑
impl PartialOrd for CustomKey {
    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
        self.0.partial_cmp(&other.0)
    }
}

impl Ord for CustomKey {
    fn cmp(&self, other: &Self) -> Ordering {
        self.0.cmp(&other.0)
    }
}

fn main() {
    let mut m: BTreeMap<CustomKey, i32> = BTreeMap::new();
    m.insert(CustomKey("hello".to_string(), vec![12]), 34);
    m.insert(CustomKey("hello".to_string(), vec![4]), 56);
    m.insert(CustomKey("other".to_string(), vec![4]), 44);

    // 查询所有第一个元素为"hello"的键值对
    let query_key = CustomKey("hello".to_string(), vec![]);
    for (k, v) in m.range(query_key.clone()..=query_key) {
        println!("key: {:?}, value: {}", k, v);
    }
}

方案三：使用filter遍历（适合小数据量） #

如果数据量不大，也可以直接遍历所有元素并过滤出第一个元素匹配的项，这种写法更直观但效率不如range查询：

示例代码：

use std::collections::BTreeMap;

fn main() {
    let mut m: BTreeMap<(String, Vec<i32>), i32> = BTreeMap::new();
    m.insert(("hello".to_string(), vec![12]), 34);
    m.insert(("hello".to_string(), vec![4]), 56);
    m.insert(("other".to_string(), vec![4]), 44);

    // 过滤第一个元素为"hello"的项
    for (k, v) in m.iter().filter(|(key, _)| key.0 == "hello") {
        println!("key: {:?}, value: {}", k, v);
    }
}

注意：方案一的右边界构造需要确保"下一个字符串"确实大于所有以目标字符串为前缀的可能字符串，使用format!("{}z", target)是简单可行的方式，也可以通过字符的递增逻辑生成更严谨的边界（比如将目标字符串的最后一个字符加1，处理进位）。

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