Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
find-the-shortest-superstring.cpp 2.29 KB
一键复制 编辑 原始数据 按行查看 历史
kamyu 提交于 2018-11-18 17:15 +08:00 . Update find-the-shortest-superstring.cpp
// Time: O(n^2 * (l^2 + 2^n))
// Space: O(n^2)
class Solution {
public:
string shortestSuperstring(vector<string>& A) {
const int n = A.size();
vector<vector<int>> overlaps(n, vector<int>(n));
for (int i = 0; i < n; ++i) {
for (int j = 0; j < n; ++j) {
for (int l = min(A[i].length(), A[j].length());
l >= 0; --l) {
if (A[i].substr(A[i].length() - l) == A[j].substr(0, l)) {
overlaps[i][j] = l;
break;
}
}
}
}
vector<vector<int>> dp(1 << n, vector<int>(n));
vector<vector<int>> prev(1 << n, vector<int>(n, -1));
for (int mask = 1; mask < 1 << n; ++mask) {
for (int bit = 0; bit < n; ++bit) {
if (((mask >> bit) & 1) == 0) {
continue;
}
auto prev_mask = mask ^ (1 << bit);
for (int i = 0; i < n; ++i) {
if (((prev_mask >> i) & 1) == 0) {
continue;
}
auto value = dp[prev_mask][i] + overlaps[i][bit];
if (value > dp[mask][bit]) {
dp[mask][bit] = value;
prev[mask][bit] = i;
}
}
}
}
int bit = -1;
for (int i = 0; i < n; ++i) {
if (bit == -1 ||
dp.back()[i] > dp.back()[bit]) {
bit = i;
}
}
vector<int> words;
for (int mask = (1 << n) - 1; bit != -1;) {
words.emplace_back(bit);
tie(mask, bit) = make_pair(mask ^ (1 << bit),
prev[mask][bit]);
}
reverse(words.begin(), words.end());
unordered_set<int> lookup(words.begin(), words.end());
for (int i = 0; i < n; ++i) {
if (!lookup.count(i)) {
words.emplace_back(i);
}
}
auto result = A[words[0]];
for (int i = 1; i < words.size(); ++i) {
auto overlap = overlaps[words[i - 1]][words[i]];
result += A[words[i]].substr(overlap);
}
return result;
}
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助