【ACWing】4398. 查询字符串

题目地址：

https://www.acwing.com/problem/content/description/4401/

给定$n$个字符串$f_1,f_2,\dots ,f_n$。这些字符串两两不同。下面给定$q$个询问。其中，第$i$次询问给定一个字符串$s_i$，你的任务是：计算$f_1\sim f_n$这$n$个字符串中，包含$s_i$作为子串的字符串的数量。从$f_1\sim f_n$这$n$个字符串中，任选一个包含$s_i$作为子串的字符串输出。

输入格式：
第一行包含整数$n$。
接下来$n$行，其中第$i$行包含字符串$f_i$。
再一行包含整数$q$。
接下来$q$行，其中第$i$行包含字符串$s_i$。
所有$f_i$和$s_i$都只包含小写字母、数字以及.。

输出格式：
共$q$行，其中第$i$行输出第$i$个询问的答案。
首先输出$f_1\sim f_n$这$n$个字符串中包含$s_i$作为子串的字符串的数量。然后从$f_1\sim f_n$这$n$个字符串中任选一个包含$s_i$作为子串的字符串输出。
如果这样的字符串不唯一，则输出任意合理字符串均可，如果这样的字符串不存在，则输出-。

数据范围：
前三个测试点满足$1\le n,q\le 20$。
所有测试点满足$1\le n\le 10000$，$1\le q\le 50000$，$1\le |f_i|,|s_i|\le 8$。

观察到每个字符串都很短，可以直接用两个哈希表，一个存每个子串出现在多少个原串里，另一个存每个子串存在于哪个原串里。代码如下：

#include <iostream>
#include <unordered_map>
#include <unordered_set>
#include <vector>
using namespace std;

unordered_map<string, int> mp, mp1;
unordered_set<string> st;
vector<string> vs;
int n, q;

int main() {
  scanf("%d", &n);
  string s;
  for (int i = 0; i < n; i++) {
    cin >> s;
    vs.push_back(s);
    st.clear();
    for (int j = 0; j < s.size(); j++)
      for (int k = j; k < s.size(); k++) {
        auto t = s.substr(j, k - j + 1);
        if (!st.count(t)) {
          st.insert(t);
          mp1[t] = i;
          mp[t]++;
        }
      }
  }
  scanf("%d", &q);
  while (q--) {
    cin >> s;
    if (!mp[s])
      puts("0 -");
    else
      cout << mp[s] << " " << vs[mp1[s]] << endl;
  }
}

预处理时间复杂度$O({\sum }_i{l}_i^3)$，每次查询时间$O(l_s)$，空间$O({\sum }_i{l}_i)$。