动态规划的好题
Given a string, your task is to count how many palindromic substrings in this string.
The substrings with different start indexes or end indexes are counted as different substrings even they consist of same characters.
Example 1:
Input: "abc" Output: 3 Explanation: Three palindromic strings: "a", "b", "c".
Example 2:
Input: "aaa" Output: 6 Explanation: Six palindromic strings: "a", "a", "a", "aa", "aa", "aaa".
Note:
- The input string length won't exceed 1000.
属于线性结构的动态规划,思维不要局限在一维数组上。。。
题目要求一个字符串中回文子串的个数,定义d[i][j]表示位置i到j的字符串中回文串的个数,考虑头和尾,如果头和尾的两个字符不一样,那么回文字符串个数=去掉头的字符串中回文的个数+去掉尾的字符串中回文的个数-去掉头尾的字符串中回文的个数(容斥原理),如果头尾两个字符串一样,那么回文字符串的个数=1+去掉头尾的字符串中回文的个数(1表示在子串已经回文的前提下,加上头尾以后多形成的那个字符串)
写成状态转移方程:
d[i][j]=d[i+1][j]+d[i][j-1]-d[i+1][j-1], str[i]!=str[j]
d[i][j]=d[i+1][j-1]+1, str[i]=str[j]
但是这样写出来的代码是有问题的:
int solve(string str){
int len=str.size();
vector<vector<int>> dp(len,vector<int>(len));
for(int j=0;j<len;j++){
dp[j][j]=1;
for(int i=j-1;i>=0;i--){
dp[i][j]=dp[i+1][j]+dp[i][j-1]-dp[i+1][j-1];
if(str[i]==str[j])
dp[i][j]+=1+dp[i+1][j-1];
}
}
return dp[0][len-1];
}
另外一种DP的方式是:定义d[i][j]:若从i到j的字符串为回文,则为真(1),否则为假(0),那么d[i][j]为真的前提是:头尾两个字符串相同并且去掉头尾以后的字串也是回文(即d[i+1][j-1]为真),这里面要注意特殊情况,即:去掉头尾以后为空串,所以如果j-i<3,并且头尾相等,也是回文的。
这样就得到了下面的关键代码:
dp[i][j]=((s[i]==s[j])&&(j-i<3||dp[i+1][j-1]));
AC代码:
class Solution {
public:
int countSubstrings(string s) {
int len=s.size(),res=0;
vector<vector<int>> dp(len,vector<int>(len,0));
for(int i=len-1;i>=0;i--){
for(int j=i;j<len;j++){
dp[i][j]=((s[i]==s[j])&&(j-i<3||dp[i+1][j-1]));
if(dp[i][j])
res++;
}
}
return res;
}
};