NowCoder 多校第一场E

挺有意思的DP

一开始有个大致的思路

但是没想到怎么去重

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10,M=1e9+7;
ll A[N],ans[N],DP[N][20];
int main(){
	int n,m,k;
	while(~scanf("%d%d%d", &n, &m, &k)){
		memset(DP,0,sizeof DP);
		memset(ans,0,sizeof DP);
		ans[0]=1;
		for(int i=1;i<=n;++i) scanf("%lld",&A[i]);
		for(int i=1;i<=n;++i){
			for(int j=i;j>=max(i-j,1);--j){
				ans[j]+=ans[j-1]+M-DP[j][A[i]];
				ans[j]%=M;
				DP[j][A[i]]=ans[j-1];
			}
		}
		printf("%lld\n", (ans[n-m]%M + M) % M);
	}
}

猜你喜欢

转载自blog.csdn.net/Gipsy_Danger/article/details/82182729