【Leetcode】30. Substring with Concatenation of All Words

题目地址：

https://leetcode.com/problems/substring-with-concatenation-of-all-words/

给定一个长$n$的字符串$s$和一个长$m$的字符串数组$A$，$A$里每个单词的长度都相等，都是$l$，要求寻找所有的位置$i$使得$s[i:i+ml-1]$恰好是$A$中所有字符串按某个顺序的拼接。

先将$A$的所有字符串加入一个哈希表，接着开始遍历$s$。我们将对于$s$的遍历分为$l$次，每次从$[0,l)$其中一个下标出发，设$0\le t<l$，那么可以得出$m$个字符串$s[t,t+l),s[t+l,t+2l),...,s[t+(m-1)l,t+ml)$，只需要看一下这$m$个字符串是不是恰好是$A$中的那些字符串即可。但是这里，直接比较两个哈希表是否相等效率太低，我们可以另外开一个变量，这个变量存这$n$个字符串里包含于多重集$A$的那些字符串有多少个（$A$中可能有重复，所以要视$A$为一个多重集），如果恰好是$m$个，就说明两个哈希表相等了。为了加速，可以用字符串哈希，将对字符串的处理变为对整数的处理。代码如下：

class Solution {
 public:
  using ull = unsigned long long;
  vector<int> findSubstring(string s, vector<string>& ws) {
    int n = s.size(), m = ws.size(), len = ws[0].size();
    vector<ull> ha(n + 1), pow(n + 1);
    pow[0] = 1;
    constexpr ull P = 131;
    for (int i = 1; i <= n; i++)
      ha[i] = ha[i - 1] * P + s[i - 1], pow[i] = pow[i - 1] * P;
    auto hash_s = [&](int l, int r) {
      return ha[r + 1] - ha[l] * pow[r - l + 1];
    };
    // 将A的所有字符串哈希值存入一个哈希表
    unordered_map<ull, int> mp;
    for (int k = 0; k < ws.size(); k++) {
      auto& w = ws[k];
      ull ha = 0;
      for (int i = 0; i < w.size(); i++) ha = ha * P + w[i];
      ++mp[ha];
    }
    vector<int> res;
    for (int st = 0; st < len; st++) {
      unordered_map<ull, int> cnt_mp;
      int cnt = 0;
      for (int i = st; i + len - 1 < n; i += len) {
        // 有一个字符串出窗口了，将其删去
        if (i - m * len >= 0) {
          auto h = hash_s(i - m * len, i - (m - 1) * len - 1);
          if (cnt_mp[h]-- <= mp[h]) cnt--;
        }
        auto h = hash_s(i, i + len - 1);
        if (++cnt_mp[h] <= mp[h]) cnt++;
        if (cnt == m) res.push_back(i - (m - 1) * len);
      }
    }
    return res;
  }
};

时间复杂度$O(lm+n)$，空间$O(n+m)$。