洛谷——P3879 [TJOI2010] 阅读理解（STL:hash+set,c++）

---

一、题目

[TJOI2010] 阅读理解

题目描述

英语老师留了 $N$ 篇阅读理解作业，但是每篇英文短文都有很多生词需要查字典，为了节约时间，现在要做个统计，算一算某些生词都在哪几篇短文中出现过。

输入格式

第一行为整数 $N$ ，表示短文篇数，其中每篇短文只含空格和小写字母。

按下来的 $N$ 行，每行描述一篇短文。每行的开头是一个整数 $L$ ，表示这篇短文由 $L$ 个单词组成。接下来是 $L$ 个单词，单词之间用一个空格分隔。

然后为一个整数 $M$ ，表示要做几次询问。后面有 $M$ 行，每行表示一个要统计的生词。

输出格式

对于每个生词输出一行，统计其在哪几篇短文中出现过，并按从小到大输出短文的序号，序号不应有重复，序号之间用一个空格隔开（注意第一个序号的前面和最后一个序号的后面不应有空格）。如果该单词一直没出现过，则输出一个空行。

样例 #1

样例输入 #1

3
9 you are a good boy ha ha o yeah
13 o my god you like bleach naruto one piece and so do i
11 but i do not think you will get all the points
5
you
i
o
all
naruto

样例输出 #1

1 2 3
2 3
1 2
3
2

提示

对于 $30\%$ 的数据， $1\le M\le 10^3$ 。

对于 $100\%$ 的数据，$1\le M\le 10^4$，$1\le N\le 10^3$ 。

每篇短文长度（含相邻单词之间的空格）$\le 5\times 10^3$ 字符，每个单词长度 $\le 20$ 字符。

每个测试点时限 $2$ 秒。

感谢@钟梓俊添加的一组数据。

二、题解

基本思路：

• 这道题要统计单词在哪几篇短文中出现过，序号不能重复且按从小到大输出短文的序号。
• (1).有没有出现过可以用哈希来判断。
• (2).序号不能重复且按从小到大输出短文的序号，很显然这里可以用STL中的set（集合）来存放序号。
• (3).那么该怎么把哈希和set结合起来呢？这里我做了一番尝试，既然之前写的时候遇到了unordered_map<int,vector> ，那么一定也可以有unordered_map<int,set>,这样就结合在一起了。

代码

#include<bits/stdc++.h>
using namespace std;#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define endl "\n"
#define int long long 
#define repn(i,o,n) for(int i=o;i<=n;i++)
#define rep(i,o,n) for(int i=o;i<n;i++)void solve(){unordered_map<string,set<int>> mp;int n,m;cin>>n;repn(i,1,n){int L;cin>>L;repn(j,1,L){string str;cin>>str;//单词序号插入到该单词对应的序号集合 mp[str].insert(i);}}cin>>m;while(m--){string str;cin>>str;if(mp[str].empty()){cout<<endl;//不存在要输出空行，注意是空行哦！（bushi空格T_T） continue;}bool flag=false;for(auto i:mp[str]){//输出 if(flag) cout<<" ";cout<<i;flag=true;}cout<<endl;}}signed main(){IOS;int T=1;//cin>>T;while(T--){solve();}return 0;
}