题解 | 生词篇章查询

#include <iostream>
#include <unordered_map>
#include <set>
#include <string>
using namespace std;

int main()
{
    int n;cin>>n;

    unordered_map<string,set<int>> wordToDocs;

    for(int i=1;i<=n;i++)
    {
        int l;cin>>l;
        set<string> uniqueWords;
        for(int j=0;j<l;j++)
        {
            string word;cin>>word;
            uniqueWords.insert(word);
        }

        for(const string& w:uniqueWords)
        {
            wordToDocs[w].insert(i);
        }
    }
    int m;cin>>m;
    for(int i=0;i<m;i++)
    {
        string word;cin>>word;
        if(wordToDocs.find(word)!=wordToDocs.end())
        {
            const set<int>& docs=wordToDocs[word];
            bool first=true;
            for(auto docId:docs)
            {
                if(!first)
                {
                    cout<<" ";
                }
                cout<<docId;
                first=false;
            }
        }
        cout<<endl;
    }
    return 0;
}