题目
题意: 给定一个全部由小写英文字母组成的字符串,允许你至多删掉其中 3 个字符,结果可能有多少种不同的字符串?(|S|<=1e6)
思路: 可以暴力水分,pta数据比较水,用set把所有删除的串扔进去。但是字符串很长会T,而且就算不T,set放的串多了会爆。
正解自然是dp.
f[i][j]:前i个字符删j个的方案数。
f[i][j] = f[i-1][j],不删.
f[i][j] += f[i-1][j-1],删.
但是这样必定有重复计算的字符串。所以要去重。
看了半天题解也没太看懂,简单来说是对于i位置,向前找第一个与他相同的字符位置x,然后哩,会发现删除[x,i-1]和删除[x+1,i]效果一样,显然重复计算了,把这一块删去即可。
qweAbcA. 删除Abc和bcA效果就一致,一共删除j个,那么前x-1个字符就删除j - (i-1-x+1).说人话就是Abc这三个字符是必删的,剩下j-(i-x)是前边可自由支配删除字符的数量。这样可以到第二个A这里,原本计算两次的qweA,可以删掉一次。不删最后的A计算了一次,删最后的A也计算了一次。去重后只计算一次。
!对于每个字符,只找前边第一个相同的字符,因为已经去过重了,再往前找就多减了,感觉这里有点难绕,dp还是学的太菜咧,润了。
时间复杂度: O(n*4)
代码:
#include<bits/stdc++.h>
// #pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
typedef long long ll;
const int N = 1e6+10;
int n,m,k,T;
ll f[N][4];
set<string> sa;
void solve()
{
ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
string s; cin>>s; s = " "+s;
n = s.size()-1;
f[0][0] = 1;
for(int i=1;i<=n;++i)
{
for(int j=0;j<=3;++j)
{
if(i < j) break;
f[i][j] = f[i-1][j];
if(j)f[i][j] += f[i-1][j-1];
for(int x=i-1;x>=1&&j-(i-x)>=0;--x)
{
if(s[x] == s[i])
{
f[i][j] -= f[x-1][j-(i-x)];
break;
}
}
}
}
ll ans = 0;
for(int i=0;i<=3;++i) ans += f[n][i];
cout<<ans;
}
signed main(void)
{
solve();
return 0;
}