在Web中搜索(Searching the Web, ACM/ICPC Beijing 2004, UVa1597)rust解法

输入n篇文章和m个请求(n<100,m≤50000),每个请求都是以下4种格式之一。

A:查找包含关键字A的文章。

A AND B:查找同时包含关键字A和B的文章。

A OR B:查找包含关键字A或B的文章。

NOT A:查找不包含关键字A的文章。

处理询问时,需要对于每篇文章输出证据。前3种询问输出所有至少包含一个关键字的行,第4种询问输出整篇文章。关键字只由小写字母组成,查找时忽略大小写。每行不超过80个字符,一共不超过1500行。

样例:

输入

复制代码
4
A manufacturer, importer, or seller of
digital media devices may not (1) sell,
or offer for sale, in interstate commerce,
or (2) cause to be transported in, or in a
manner affecting, interstate commerce,
a digital media device unless the device
includes and utilizes standard security
technologies that adhere to the security
system standards.
**********
Of course, Lisa did not necessarily
intend to read his books. She might
want the computer only to write her
midterm. But Dan knew she came from
a middle-class family and could hardly
afford the tuition, let alone her reading
fees. Books might be the only way she
could graduate
**********
Research in analysis (i.e., the evaluation
of the strengths and weaknesses of
computer system) is essential to the
development of effective security, both
for works protected by copyright law
and for information in general. Such
research can progress only through the
open publication and exchange of
complete scientific results
**********
I am very very very happy!
What about you?
**********
6
computer
books AND computer
books OR protected
NOT security
very
slick

输出

复制代码
want the computer only to write her
----------
computer system) is essential to the
==========
intend to read his books. She might
want the computer only to write her
fees. Books might be the only way she
==========
intend to read his books. She might
fees. Books might be the only way she
----------
for works protected by copyright law
==========
Of course, Lisa did not necessarily
intend to read his books. She might
want the computer only to write her
midterm. But Dan knew she came from
a middle-class family and could hardly
afford the tuition, let alone her reading
fees. Books might be the only way she
could graduate
----------
I am very very very happy!
What about you?
==========
I am very very very happy!
==========
not found
==========

解法:

rust 复制代码
use std::{
    collections::{BTreeMap, BTreeSet, HashMap},
    io,
};
#[derive(PartialEq)]
enum WordOp {
    AND,
    OR,
    None,
    NOT,
}

fn get_words(s: &String) -> Vec<String> {
    let w: String = s
        .chars()
        .map(|x| {
            if x.is_alphabetic() {
                x.to_ascii_lowercase()
            } else {
                ' '
            }
        })
        .collect();
    let wds: Vec<String> = w.split_whitespace().map(|x| x.to_string()).collect();
    wds
}
fn main() {
    let mut buf = String::new();
    io::stdin().read_line(&mut buf).unwrap();
    let n: usize = buf.trim().parse().unwrap();
    let mut articles: Vec<Vec<String>> = vec![];//每篇文章的所有行
    let mut words: Vec<HashMap<String, BTreeSet<usize>>> = vec![];//每篇文章的所有单词和单词所在的行号
    for _i in 0..n {
        let mut article: Vec<String> = vec![];
        let mut wd: HashMap<String, BTreeSet<usize>> = HashMap::new();
        loop {
            let mut buf = String::new();
            io::stdin().read_line(&mut buf).unwrap();
            if buf.trim() == "*".repeat(10) {
                break;
            }
            article.push(buf.trim().to_string());
            //count words
            let v: Vec<String> = get_words(&buf);
            let line_idx = article.len() - 1;
            for w in v.iter() {
                wd.entry(w.to_string())
                    .and_modify(|x| {
                        x.insert(line_idx);
                    })
                    .or_insert(BTreeSet::from([line_idx]));
            }
        }
        articles.push(article);
        words.push(wd);
    }
    let mut buf = String::new();
    io::stdin().read_line(&mut buf).unwrap();
    let m: usize = buf.trim().parse().unwrap();
    let mut cmds = vec![];
    for _i in 0..m {
        let mut buf = String::new();
        io::stdin().read_line(&mut buf).unwrap();
        cmds.push(buf.trim().to_string());
    }
    for cmd in cmds.iter() {
        if let Some(idx) = cmd.find("OR") {
            let word1 = cmd[0..idx - 1].to_string();
            let word2 = cmd[idx + 3..].to_string();
            find_word(&articles, &words, &word1, &word2, WordOp::OR);
        } else if let Some(idx) = cmd.find("AND") {
            let word1 = cmd[0..idx - 1].to_string();
            let word2 = cmd[idx + 4..].to_string();
            find_word(&articles, &words, &word1, &word2, WordOp::AND);
        } else if let Some(idx) = cmd.find("NOT") {
            let word1 = cmd[idx + 4..].to_string();
            find_word(&articles, &words, &word1, &"".to_string(), WordOp::NOT);
        } else {
            let word1 = cmd;
            find_word(&articles, &words, word1, &"".to_string(), WordOp::None);
        }
        println!("{}", "=".repeat(10));
    }
}

fn print_result(find_result: BTreeMap<usize, BTreeSet<usize>>, articles: &Vec<Vec<String>>) {
    if find_result.is_empty() {
        println!("not found");
    } else {
        let mut cnt = 0;
        for (k, v) in find_result.iter() {
            for i in v.iter() {
                println!("{}", articles[*k][*i]);
            }
            cnt += 1;
            if cnt != find_result.len() {
                println!("{}", "-".repeat(10));
            }
        }
    }
}

fn find_word(
    articles: &Vec<Vec<String>>,
    words: &Vec<HashMap<String, BTreeSet<usize>>>,
    word1: &String,
    word2: &String,
    op: WordOp,
) {
    let mut find_result: BTreeMap<usize, BTreeSet<usize>> = BTreeMap::new();
    for (aidx, lines) in articles.iter().enumerate() {
        let mut find_line_idx: BTreeSet<usize> = BTreeSet::new();
        let ws = words.get(aidx).unwrap();
        if op == WordOp::OR {
            if ws.contains_key(word1) || ws.contains_key(word2) {
                if let Some(idx) = ws.get(word1) {
                    find_line_idx.append(&mut idx.clone());
                }
                if let Some(idx) = ws.get(word2) {
                    find_line_idx.append(&mut idx.clone());
                }
                find_result.insert(aidx, find_line_idx);
            }
        } else if op == WordOp::AND {
            if ws.contains_key(word1) && ws.contains_key(word2) {
                let idx = ws.get(word1).unwrap();
                find_line_idx.append(&mut idx.clone());
                let idx = ws.get(word2).unwrap();
                find_line_idx.append(&mut idx.clone());
                find_result.insert(aidx, find_line_idx);
            }
        } else if op == WordOp::None {
            if ws.contains_key(word1) {
                let idx = ws.get(word1).unwrap();
                find_line_idx.append(&mut idx.clone());
                find_result.insert(aidx, find_line_idx);
            }
        } else if op == WordOp::NOT {
            if !ws.contains_key(word1) {
                find_line_idx.append(&mut (0..lines.len()).collect());
                find_result.insert(aidx, find_line_idx);
            }
        }
    }
    print_result(find_result, &articles);
}
相关推荐
咸甜适中7 小时前
rust语言 (1.88) egui (0.32.1) 学习笔记(逐行注释)(十四)垂直滚动条
笔记·学习·rust·egui
张志鹏PHP全栈10 小时前
Rust第四天,Rust中常见编程概念
后端·rust
咸甜适中14 小时前
rust语言 (1.88) egui (0.32.1) 学习笔记(逐行注释)(十五)网格布局
笔记·学习·rust·egui
susnm1 天前
最后的最后
rust·全栈
bruce541103 天前
深入理解 Rust Axum:两种依赖注入模式的实践与对比(二)
rust
该用户已不存在3 天前
这几款Rust工具,开发体验直线上升
前端·后端·rust
m0_480502645 天前
Rust 入门 生命周期-next2 (十九)
开发语言·后端·rust
寻月隐君6 天前
Rust Web 开发实战:使用 SQLx 连接 PostgreSQL 数据库
后端·rust·github
Moonbit6 天前
MoonBit Pearls Vol.05: 函数式里的依赖注入:Reader Monad
后端·rust·编程语言
Vallelonga7 天前
Rust 异步中的 Waker
经验分享·rust·异步·底层