题目链接:题目详情 - L3-020 至多删三个字符 (pintia.cn)
样例输入:
ababcc
样例输出:
25
分析:这种题目显然就是动态规划,设f[i][j]代表前i个字符中删除j个字符的不同字符串个数,那么我们枚举到第i个字符时会有两种情况,一种是删除第i个字符,那么就是在前i-1个字符中再删除j-1个字符,还有一种情况是保留第i个字符,那么就是在前i-1个字符中删除j个字符。那么就有
f[i][j]=f[i-1][j]+f[i-1][j-1]
但这个时候会发现这样统计是会重复统计的,比如字符串dabd,我们当前枚举到第四个字符d,那么如果要是删除3个字符,有f[4][3]=f[3][3]+f[3][2],那么f[3][3]中会有删除dab这种情况,就剩下了当前字符d,f[3][2]中会有删除ab这种情况,这样前三个字符中剩下了一个d,而当前字符也要被删除,也是只剩了一个d。这样就会计算重复。那我们来看看这样为什么会重复统计呢,那是因为当前字符到当前字符上一次出现的位置中间的字符全部都被删除了,那么就会重复计算。那怎样避免这种情况呢?换句话说能否知道重复了多少次呢?假如当前字符上一次出现的位置记为last,当前字符位置为i,那么也就是删除字符数大于等于i-pos时可能会造成重复统计,重复统计了多少呢?就是f[last-1][j-(i-pos)],因为我们默认是把这两个字符中的一个和其之间的字符全部删除,那么就是删除i-pos个,由于我们当前枚举的是删除j个字符,那么我们就需要在前last-1个字符中删除j-(i-pos)个字符,这也就是f[last-1][j-(i-pos)]的由来了。
细节见代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
#include<queue>
#include<vector>
#include<cmath>
using namespace std;
const int N=1e6+10;
long long f[N][4];//f[i][j]代表前i个字符中删除j个字符的不同字符串个数
char s[N];
int last[N];
int main()
{
scanf("%s",s+1);
int n=strlen(s+1);
for(int i=0;i<26;i++)
last[i]=-1;
f[0][0]=1;
for(int i=1;i<=n;i++)
{
f[i][0]=1;
for(int j=1;j<=3;j++)
{
f[i][j]=f[i-1][j]+f[i-1][j-1];
if(last[s[i]-'a']!=-1&&i-last[s[i]-'a']<=j) f[i][j]-=f[last[s[i]-'a']-1][j-(i-last[s[i]-'a'])];
}
last[s[i]-'a']=i;
}
long long ans=0;
for(int i=0;i<=3;i++)
ans+=f[n][i];
printf("%lld",ans);
return 0;
}