LeetCode 87. 扰乱字符串--动态规划--区间DP

87. 扰乱字符串
    给定一个字符串 s1，我们可以把它递归地分割成两个非空子字符串，从而将其表示为二叉树。

下图是字符串 s1 = “great” 的一种可能的表示形式。

great

/
gr eat
/ \ /
g r e at
/
a t
在扰乱这个字符串的过程中，我们可以挑选任何一个非叶节点，然后交换它的两个子节点。

例如，如果我们挑选非叶节点 “gr” ，交换它的两个子节点，将会产生扰乱字符串 “rgeat” 。

rgeat

/
rg eat
/ \ /
r g e at
/
a t
我们将 "rgeat” 称作 “great” 的一个扰乱字符串。

同样地，如果我们继续交换节点 “eat” 和 “at” 的子节点，将会产生另一个新的扰乱字符串 “rgtae” 。

rgtae

/
rg tae
/ \ /
r g ta e
/
t a
我们将 "rgtae” 称作 “great” 的一个扰乱字符串。

给出两个长度相等的字符串 s1 和 s2，判断 s2 是否是 s1 的扰乱字符串。

示例 1:

输入: s1 = “great”, s2 = “rgeat”
输出: true
示例 2:

输入: s1 = “abcde”, s2 = “caebd”
输出: false

题解

比较明确的区间DP，因为数据量比较小，可以考虑的方向也比较多，这样考虑，假设s1的区间为[l1,r1]，s2的区间为[l1,r2]，k1取值在[l1,r1]之间，k2取值在[l2,r2]之间，定义区间dp[N][N][N][N]；dp[l1][r1][l2][r2]表示字符串s1[l1,r1]和s2[l2,r2]能否按题目进行匹配的关系，那么最后的答案就是返回dp[0][n-1][0][n-1]，这样不难理解吧，现在考虑状态转移方程，可以知道如下关系：

if dp[l1][k1][l2][k2]==ture and  dp[k1+1][r1][k2+1][r2]==true:
	dp[l1][r1][l2][r2]=true
if dp[l1][k1][k2+1][r2]==true and dp[k1+1][r1][l2][k2]==true:
	dp[l1][r1][l2][r2]=true

上面的关系不难理解吧，如此反复操作下去，那么答案就出来了，具体看代码吧。

class Solution {
public:
    bool dp[31][31][31][31];
    bool isScramble(string s1, string s2) {
        if(s1.length()!=s2.length())return false;//特判
        memset(dp,0,sizeof(dp));
        int n=s1.length();
        for(int i=0;i<n;i++)//初始化
        {
            for(int j=0;j<n;j++)
            {
                if(s1[i]==s2[j])
                dp[i][i][j][j]=true;
            }
        }
        for(int d=1;d<n;d++)//区间的宽度
        {
            for(int l1=0;l1+d<n;l1++)
            {
                for(int l2=0;l2+d<n;l2++)
                {
                    int r1=l1+d;
                    int r2=l2+d;
                    for(int k1=l1;k1<r1;k1++)
                    {
                        for(int k2=l2;k2<r2;k2++)
                        {
                            if(dp[l1][k1][l2][k2]&&dp[k1+1][r1][k2+1][r2])
                            dp[l1][r1][l2][r2]=true;
                            if(dp[l1][k1][k2+1][r2]&&dp[k1+1][r1][l2][k2])
                            dp[l1][r1][l2][r2]=true;
                        }
                    }
                }
            }
        }
        return dp[0][s1.length()-1][0][s1.length()-1];
    }
};