【Leetcode】943. Find the Shortest Superstring

题目地址：

https://leetcode.com/problems/find-the-shortest-superstring/description/

给定一个长$n$的单词数组$A$，$1\le n\le 12$，每个单词长度不超过$20$。求最短的字符串$s$，使得$A$里每个单词都是其子串。题目保证$A$中单词各不相同，并且没有一个单词是另一个单词的子串。

设$f[x][i]$是已经选择的单词的状态为$x$，并且以$A[i]$结尾的情况下所能得到的最短$s$的长度。$x$是已经选择的单词的所有下标的集合的二进制压缩（显然同一个单词不会选择超过$1$次，这样没有意义）。我们可以考虑由$f[x][i]$可以递推出什么样的状态，递推的前提是$x$的二进制第$i$位为$1$，如果转移到以$j$结尾的状态（前提是$x$不含$j$），那么长度的增加值实际上是$l_{A[i]}+l_{A[j]}$再减去$A[i]$的后缀与$A[j]$的前缀的最长重合的长度（这个长度我们要预处理一下）。最后的答案长度即为${\min }_{i}f[2^n-1][i]$。再由这个答案倒过来推出$s$是什么即可。代码如下：

class Solution {
 public:
  string shortestSuperstring(vector<string> &ws) {
    int n = ws.size();
    auto calc = [&](int i, int j) {
      string &s1 = ws[i], &s2 = ws[j];
      for (int k = min(s1.size(), s2.size()); k; k--)
        if (s1.substr(s1.size() - k) == s2.substr(0, k)) return k;
      return 0;
    };
    vector<vector<int>> p(n, vector<int>(n));
    for (int i = 0; i < n; i++)
      for (int j = 0; j < n; j++)
        if (i != j) p[i][j] = calc(i, j);

    const int INF = 2e8;
    vector<vector<int>> f(1 << n, vector<int>(n, INF));
    for (int i = 0; i < n; i++) f[1 << i][i] = ws[i].size();
    for (int i = 0; i < 1 << n; i++)
      for (int j = 0; j < n; j++)
        if (i >> j & 1)
          for (int k = 0; k < n; k++)
            if (!(i >> k & 1))
              f[i | (1 << k)][k] = min(f[i | (1 << k)][k],
                                       f[i][j] + (int)ws[k].size() - p[j][k]);

    int i = (1 << n) - 1, j = -1;
    for (int k = 0; k < n; k++)
      if (!~j || f[i][k] < f[i][j]) j = k;
    vector<int> v;
    while (i) {
      v.push_back(j);
      i -= 1 << j;
      for (int k = 0; k < n; k++)
        if ((i >> k & 1) &&
            f[i][k] + ws[j].size() - p[k][j] == f[i + (1 << j)][j]) {
          j = k;
          break;
        }
    }
    reverse(v.begin(), v.end());
    string s = ws[v[0]];
    for (i = 1; i < n; i++) s += ws[v[i]].substr(p[v[i - 1]][v[i]]);
    return s;
  }
};

时间复杂度$O(n^2(2^n+l^2))$，$l$为单词最长长度，空间$O(n{2}^n)$。