forked from kamyu104/LeetCode-Solutions
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathlongest-string-chain.cpp
More file actions
26 lines (25 loc) · 884 Bytes
/
longest-string-chain.cpp
File metadata and controls
26 lines (25 loc) · 884 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
// Time: O(n * l^2)
// Space: O(n * l)
class Solution {
public:
int longestStrChain(vector<string>& words) {
sort(words.begin(), words.end(),
[](const string& a, const string& b) {
return less<int>()(a.length(), b.length());
});
unordered_map<string, int> dp;
for (const auto& w : words) {
for (int i = 0; i < w.length(); ++i) {
auto tmp = w.substr(0, i);
tmp += w.substr(i + 1);
dp[w] = max(dp[w], dp[tmp] + 1);
}
}
using pair_type = decltype(dp)::value_type;
return max_element(dp.cbegin(), dp.cend(),
[] (const pair_type& a,
const pair_type& b) {
return a.second < b.second;
})->second;
}
};