在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);
}
相关推荐
许野平13 小时前
Rust 同步方式访问 REST API 的完整指南
java·网络·rust·restful
许野平13 小时前
Rust:如何访问 *.ini 配置文件?
开发语言·数据库·rust·ini·configparser
许野平13 小时前
Rust:开发 DLL 动态链接库时如何处理 C 字符串
c语言·开发语言·rust·字符串·动态库·dll
许野平13 小时前
Rust: 获取 MAC 地址方法大全
开发语言·macos·rust·mac
维维酱1 天前
三、目标规范与交叉编译
rust
维维酱1 天前
二、相关编程基础
rust
a cool fish(无名)1 天前
10.1通用数据类型
rust
寻月隐君1 天前
Rust Scoped Threads 实战:更安全、更简洁的并发编程
后端·rust·github
a cool fish(无名)1 天前
9.1无法恢复的错误与 panic!
rust
勇敢牛牛_2 天前
【OneAPI】网页搜索API和网页正文提取API
rust·oneapi