标签:style blog http color io ar for 2014 div
晚上做了一下leetcode的Interleaving String,觉得特别适合比较树的遍历和动态规划的效率比较。
题目如下:
Given s1, s2, s3, find whether s3 is formed by the interleaving of s1 and s2.
For example,
Given:
s1 = "aabcc"
,
s2 = "dbbca"
,
When s3 = "aadbbcbcac"
, return true.
When s3 = "aadbbbaccc"
, return false.
第一反应很适合用递归:
bool isInterleave_res(string s1, string s2, string s3) { if ((s1.empty() && s2.empty() && s3.empty()) || (!s1.empty() && !s3.empty() && s3[0] == s1[0] && isInterleave_res(s1.substr(1, s1.length()-1), s2, s3.substr(1, s3.length()-1))) || (!s2.empty() && !s3.empty() && s3[0] == s2[0] && isInterleave_res(s1, s2.substr(1, s2.length()-1), s3.substr(1, s3.length()-1))) ) { return true; } return false; }
这样本质上就是树的深度优先遍历,但是效率就比较低,很多节点会多次计算到,果然TLE,用例是:
s1 = "bbbbbabbbbabaababaaaabbababbaaabbabbaaabaaaaababbbababbbbbabbbbababbabaabababbbaabababababbbaaababaa";
s2 = "babaaaabbababbbabbbbaabaabbaabbbbaabaaabaababaaaabaaabbaaabaaaabaabaabbbbbbbbbbbabaaabbababbabbabaab";
s3 = "babbbabbbaaabbababbbbababaabbabaabaaabbbbabbbaaabbbaaaaabbbbaabbaaabababbaaaaaabababbababaababbababbbababbbbaaaabaabbabbaaaaabbabbaaaabbbaabaaabaababaababbaaabbbbbabbbbaabbabaabbbbabaaabbababbabbabbab";
这题很容易也想到用动态规划,设置表dp,dp[i][j]代表s1的前i个和s2的前j个能和s3的前i+j个匹配上,则根据dp[i][j]就可以推出dp[i+1][j]和dp[i][j+1]。返回dp最右下角的值。
例如题目中的例子1:s3 = "aadbbcbcac"
例子2:s3 = "aadbbcbccc"
class Solution { public: bool isInterleave(string s1, string s2, string s3) { int len1 = s1.length(), len2 = s2.length(), len3=s3.length(); if (len1+len2!=len3) { return false; } vector<vector<bool>> dp(len1+1, vector<bool>(len2+1, false)); dp[0][0] = true; for (int i = -1; i < len1; ++i) { for (int j = -1; j < len2; ++j) { if (dp[i+1][j+1]==true) { if (i+1 < len1 && i+j+2 < len3 && s1[i+1]==s3[i+j+2]) { dp[i+2][j+1] = true; } if (j+1 < len2 && i+j+2 < len3 && s2[j+1]==s3[i+j+2]) { dp[i+1][j+2] = true; } } } } return dp[len1][len2]; } };
这样代码就通过了。
仔细观察,这题用遍历就是一棵二叉树的深度优先遍历,而这棵树对应dp的表格就是(i, j)的左节点就是(i+1, j)右节点是(i, j+1),所以(i+1, j)的右节点和(i, j+1)的左节点是同样的(i+1, j+1),进一步加剧了遍历树的重复。
但是观察dp表,其实也浪费了很多,如果s1长度为m而s2长度为n,DP的时间复杂度和空间复杂度都为m*n(空间复杂度可以优化为2×m或者2×n,因为只和上一行相关),但是其实我们关心的只是其中为T的点,最好情况下其实只需要m+n。
我们可以将树的遍历(只走了为T的点)和DP表结合起来,可以理解为将DP表用稀释矩阵的形式保存,只保存T的点,而走过的T点就不再遍历,这样在m和n很大的情况下应该有较大提升,代码如下:
class Solution { public: bool isInterleave(string s1, string s2, string s3) { int len1 = s1.length(), len2 = s2.length(), len3=s3.length(); if (len1+len2!=len3) { return false; } stack<pair<int, int>> node; set<pair<int, int>> passed; node.push(pair<int, int>(-1, -1)); pair<int, int> last_node; int count = 0; while (!node.empty()) { count++; int i = node.top().first; int j = node.top().second; passed.insert(pair<int, int>(i, j)); if (i == len1-1 && j == len2-1) { return true; } node.pop(); if (i+1 < len1 && i+j+2 < len3 && s1[i+1]==s3[i+j+2] && passed.find(pair<int, int>(i+1, j)) == passed.end()) { node.push(pair<int, int>(i+1, j)); } if (j+1 < len2 && i+j+2 < len3 && s2[j+1]==s3[i+j+2] && passed.find(pair<int, int>(i, j+1)) == passed.end()) { node.push(pair<int, int>(i, j+1)); } } return false; } };
三种方法,第一种,Time Limit Exceeded;第二种16ms通过所有测试用例,第三种8ms
[LeetCode]Interleaving String关于遍历和动态规划
标签:style blog http color io ar for 2014 div
原文地址:http://www.cnblogs.com/yezhangxiang/p/3952659.html