题目描述
给定一个字符串str,如果可以在str的任意位置添加字符,请返回在添加字符最少的情况下,让str整体都是回文字符串的一种结果。
输入描述:
输入包含一行字符串,代表str(1 \leq length_{str} \leq 5000)(1≤length
str
≤5000)。
输出描述:
输出一行,代表返回的字符串。
示例1
输入
复制
ABA
输出
复制
ABA
示例2
输入
复制
AB
输出
复制
ABA
备注:
时间复杂度O(n^2)O(n
2
),空间复杂度O(n^2)O(n
2
)
#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
int dp[5007][5007];//i到j转变为回文串需要添加的最少字符个数
//状态转移方程: dp[i][j]=min(dp[i-1][j],dp[i][j-1])
int main()
{
string s;
cin>>s;
if(s.length()==1)
{
cout<<s<<endl;
return 0;
}
for(int i=s.length()-1;i>=0;i--)
{
for(int j=i;j<s.length();j++)
{
if( s[i]==s[j] && (j-i<=2) )
dp[i][j]=0;
else if(s[i]==s[j])
dp[i][j]=dp[i+1][j-1];
else
dp[i][j]=min(dp[i+1][j],dp[i][j-1])+1;
}
}
string t(dp[0][s.length()-1]+s.size(),'\0');//生成一个字符串,包含 dp[0][s.length()-1]+s.size() 个0字符
int l=0,r=t.length()-1;
int i=0,j=s.length()-1;
while(l<=r)
{
if(s[i]==s[j])
{
t[r--]=t[l++]=s[i];
i++,j--;
}
else if( dp[i+1][j]<dp[i][j-1])
{
t[r--]=t[l++]=s[i];
i++;
}
else
{
t[r--]=t[l++]=s[j];
j--;
}
}
cout<<t<<endl;
return 0;
}