题目
https://vjudge.net/problem/POJ-1159(POJ崩了,题目参照VJudge里的)
题意:给定一个字符串,求最少需要插入多少字符,使其成为回文串。如Ab3bD最少需要插入2个字符。
解题思路
要使插入的字符最少,那就要尽可能利用原字符串中的最长回文子序列。求出了最长回文子序列的长度,用原串长度减去它即答案。
- 将输入串s逆序,得到s2
- 求s和s2的最长公共子序列LCS(即最长回文子序列)
- 答案 = 输入串长度 - 最长回文子序列长度
由于输入串长度最大可能为5000,dp数组太大导致MLE,因此采用滚动数组(dp[i][j]只取决于上一行和当前行的左侧)。
用二维数组的前两行作为滚动数组,dp[e][]表示当前行,dp[1-e][]表示上一行,利用e=1-e切换。
注:在LCS中,用下标1…n而不是0..n-1来存放结果,是为了防止0-1=-1越界。
AC代码
#include <iostream>
#include <cstdio>
#include <string>
#include <algorithm>
using namespace std;
char s[5005], s2[5005]; //s[1..n],s2[1..n]
int dp[2][5005]; //采用滚动数组,节省空间
int main()
{
int n;
scanf("%d", &n);
getchar();
s[0] = s2[0] = '!';
for (int i = 1; i <= n; ++i)
{
scanf("%c", &s[i]);
s2[n+1-i] = s[i]; //逆序
}
s[n+1] = s2[n+1] = '\0';
//cout << s << ' ' << s2 << endl;
fill(dp[0], dp[0]+2*5005, 0);
int e = 0;
for (int i = 1; i <= n; ++i)
{
e = 1-e; //滚动数组,用e代表当前行,1-e表示上一行
for (int j = 1; j <= n; ++j)
{
if (s[i] == s2[j]) //字符相等
dp[e][j] = dp[1-e][j-1] + 1; //(1-e,j-1)即(i-1,j-1)
else
dp[e][j] = max(dp[1-e][j], dp[e][j-1]);//(1-e,j)即(i-1,j),(e,j-1)即(i,j-1)
}
}
int len = dp[e][n]; //原串中最长回文子序列长度
cout << n-len << endl;
return 0;
}