LeetCode 1745. 回文串分割 IV--动态规划--区间DP

1745. 回文串分割 IV
      给你一个字符串 s ，如果可以将它分割成三个 非空 回文子字符串，那么返回 true ，否则返回 false 。

当一个字符串正着读和反着读是一模一样的，就称其为 回文字符串 。

示例 1：

输入：s = “abcbdd”
输出：true
解释：“abcbdd” = “a” + “bcb” + “dd”，三个子字符串都是回文的。
示例 2：

输入：s = “bcbddxy”
输出：false
解释：s 没办法被分割成 3 个回文子字符串。

提示：

3 <= s.length <= 2000
s​​​​​​ 只包含小写英文字母。

题解

我们这样考虑，字符串长度为n，于是整个访问区间为[0,n]，如果某个位置x使得
[0,x-1]可以构成一个回文串，[x,n]也可以构成一个回文串，那么[0,n]就可以构成两个回文串，这个好理解的吧，那么同样的，如果[0,x-1]通过刚才说的判断方法可以构成两个回文串，而且[x,n]也恰好可以构成一个回文串，那么[0,n]就可以构成3个回文串，答案就出来了。我们怎么快速的知道那些区间可以构成一个回文串呢？这个需要初始化，看逻辑代码

bool vis[2001][2001];
void init(int pos)
{
    for(int i=pos,j=pos;i>=0&&j<n;i--,j++)
    {
        if(s[i]!=s[j])break;
        vis[i][j]=true;
    }
    for(int i=pos-1,j=pos;i>=0&&j<n;i--,j++)
    {
        if(s[i]!=s[j])break;
        vis[i][j]=true;
    }
}
for(int i=0;i<n;i++)
init(i);

这个是比较快速的，能通过n方的时间复杂度找出字符串s中所有的回文子串。

于是我们进行动态规划处理。

我们定义dp[N][N][3];dp[i][j][x]表示在区间[i,j]能构成x个回文串。

dp[i][j][1]=vis[i][j]

于是从小到大：

for(int k=2;k<=3;k++)
{
	for(int i=0;i<n;i++)
	{
		for(int j=i;j>0;j--)
		{
			if(dp[0][j-1][k-1]&&dp[j][i][1])
			dp[0][i][k]=true;
		}
	}
}

最后返回dp[0][n][3]；便是答案，思路还是比较简单明了的。

AC代码

class Solution {
public:
    bool dp[2001][2001][4];
    void init(string s,int pos)
    {
        for(int i=pos,j=pos;i>=0&&j<s.length();i--,j++)
        {
            if(s[i]!=s[j])break;
            dp[i][j][1]=true;
        }
        for(int i=pos,j=pos+1;i>=0&&j<s.length();i--,j++)
        {
            if(s[i]!=s[j])break;
            dp[i][j][1]=true;
        }
    }
    bool checkPartitioning(string s) {
        memset(dp,0,sizeof(dp));
        for(int i=0;i<s.length();i++)
        init(s,i);
        for(int k=2;k<=3;k++)
        {
            for(int i=0;i<s.length();i++)
            {
                for(int j=i;j>=1;j--)
                {
                    if(dp[0][j-1][k-1]&&dp[j][i][1])
                    {
                        dp[0][i][k]=true;
                    }
                }
            }
        }
        return dp[0][s.length()-1][3];
    }
};