题目
题解
动态规划。
状态表示:f[i][j]
表示前i
个字符删除j
个后不同的字符串数。
首先简单考虑一下,对于第i
个字符,我们可以选择删除或者不删除,即f[i][j] = f[i-1][j-1] + f[i-1][j]
看个例子cdabnaxy
,删除abn
和删除bna
后得到的字符串都是cdaxy
,说明存在多统计的情况,那么我们要减去这些重复的情况。
根据上面那个例子可以发现对于一个字符s[i]
,如果在i之前存在一个x
使得s[x]=s[i]
,那么删除[x,i-1]
间的字符和删除[x+1,i]
间的字符其实是重复的,等价的,那么f[i][j]
就要减去f[x-1][j-(i-x)]
,减去删除这段造成的重复串。
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e6+10;
string s;
int n, last[N];
LL ans, f[N][5];
int main()
{
cin >> s;
s = '.' + s;
for (int i = 0;i < s.size();i ++)
f[i][0] = 1;
for (int i = 1;i < s.size();i ++) {
int x = s[i] - 'a';
for (int j = 1;j <= 3;j ++) {
f[i][j] = f[i-1][j-1] + f[i-1][j];
if (last[x] && j - (i - last[x]) >= 0)
f[i][j] -= f[last[x] - 1][j - (i - last[x])];
}
last[x] = i;
}
for (int i = 0;i <= 3;i ++) ans += f[s.size()-1][i];
cout << ans;
return 0;
}