P3879 [TJOI2010] 阅读理解
题面
英语老师留了 \(N\) 篇阅读理解作业,但是每篇英文短文都有很多生词需要查字典,为了节约时间,现在要做个统计,算一算某些生词都在哪几篇短文中出现过。
输入格式
第一行为整数 \(N\) ,表示短文篇数,其中每篇短文只含空格和小写字母。
按下来的 \(N\) 行,每行描述一篇短文。每行的开头是一个整数 \(L\) ,表示这篇短文由 \(L\) 个单词组成。接下来是 \(L\) 个单词,单词之间用一个空格分隔。
然后为一个整数 \(M\) ,表示要做几次询问。后面有 \(M\) 行,每行表示一个要统计的生词。
输出格式
对于每个生词输出一行,统计其在哪几篇短文中出现过,并按从小到大输出短文的序号,序号不应有重复,序号之间用一个空格隔开(注意第一个序号的前面和最后一个序号的后面不应有空格)。如果该单词一直没出现过,则输出一个空行。
提示说明
对于 \(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\) 秒。
思路
这道题其实是可以用STL的。
首先,统计可以用 std::map
,对于一段有几个单词的问题,可以使用 std::set
。
其实就是 std::map<std::string,std::set<int> >
。
时间复杂度 \(O(N(\log \max\{L\}) + MN(\log^2 N))\)。好复杂
代码
#include <bits/stdc++.h>
using namespace std;
int n,m;
map<string,set<int> > dictionary;
int main(){
cin>>n;
for(int i=1,slen;i<=n;i++){
string s;
cin>>slen;
for(int j=1;j<=slen;j++){
cin>>s;
dictionary[s].insert(i);
}
}
cin>>m;
for(int i=1;i<=m;i++){
string word;
cin>>word;
if(dictionary.count(word)){
for(auto j:dictionary[word]){
cout<<j<<' ';
}
}
cout<<'\n';
}
return 0;
}