PTA 7-15 至多删三个字符 （30 分)

题意：给你一个字符串，至多删3个字符，问你能得到多少种不同的字符串

思路：很显然的dp，dp[i][j]表示到第i个字符，删了j个的数量，答案就是sigma(dp[n][i]), i :[0. 3]。

考虑转移：

删第i个字符：dp[i][j] += dp[i-1][j-1]（j != 0）

不删第i个字符：dp[i][j] += dp[i-1][j];

这样显然是有重复的，比如

qweAbcAplm

显然删去Abc, bcA这两个字符串后得到的字符串是一样的，所以对于相同的字符位置，我们要考虑去重操作。

那么在走到第二个A的时候，如何知道重复地计算了多少次呢？

记第一个A的位置为k，那么在计算第二个A的时候，重复计算的次数就是dp[k-1][j - (i - x)]

其实也挺好理解，dp[i][j]删j次嘛，能有重复肯定是要把[x, i-1]这一段删了，然后剩余j - (i - x) 个删的位置，就在前面得到喽。

最后就是答案最大肯定是超过int的，题目又没说取模，那么用long long 存即可。

AC Code:

//My Conquest Is the Sea of Stars.
#pragma GCC diagnostic error "-std=c++11"
#include<set>
#include<map>
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<cstdio>
#include<string>
#include<vector>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<iostream>
#include<algorithm>
#define endl "\n"
#define fi first
#define se second
#define gcd __gcd
#define pb push_back
#define mp make_pair
#define lowbit(x) x & (-x)
#define PII  pair<int, int> 
#define all(x) x.begin(), x.end()
#define rep(i, a, b) for(int i = a; i <= (b); i++)
#define Rep(i, a, b) for(int i = a; i >= (b); i--)
#define FAST ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)
typedef long long ll;
typedef unsigned long long ull;
const int inf = 0x3f3f3f3f;
const int mod = (int)1e9 + 7;
const int maxn = (int)1e6 + 5;
const ll INF = 0x3f3f3f3f3f3f3f3f;
using namespace std;

char s[maxn];
ll dp[maxn][4];

int main()
{
	scanf("%s", s + 1);
	int n = strlen(s + 1);
	dp[0][0] = 1;
	rep(i, 1, n){
		rep(j, 0, 3){
			dp[i][j] += dp[i-1][j];
			if(j) dp[i][j] += dp[i-1][j-1];
			for(int k = i - 1; k >= 1 && i - k <= j; k--){
				if(s[k] == s[i]){
					dp[i][j] -= dp[k-1][j-(i-k)];
					break;
				}
			}
		}
	}
	printf("%lld\n", dp[n][0] + dp[n][1] + dp[n][2] + dp[n][3]);
	return 0;
}