424C
来源:互联网 发布:外国人学中文的软件 编辑:程序博客网 时间:2024/06/03 18:02
题意:一个比赛中,有n次打分,有一个初始分数,这个初始分数加上这n次的分可得出n个结果,而作者只记得其中k次的结果,且不确定是哪几次的。问初始分数有多少中可能?
解答:首先要初始化这n次的打分的前缀和。注意前缀和地去重处理!v将这k个结果分别枚举为第一次到第n次的结果,分别计算出初始值。如果这个得出某个初始值得次数恰好为k,那么这个初始值即为可能的结果。
#include<iostream>#include<cstdio>#include<cstring>#include<vector>#include<algorithm>#include<map>const int MAXN = 2200;int a[MAXN];int b[MAXN];int sum[MAXN];using namespace std;vector<int> s;int main(){ int n,k; while(~scanf("%d%d",&n,&k)) { s.clear(); map<int,int> w; memset(sum,0,sizeof(sum)); for(int i = 1;i <= n;i++){ scanf("%d",&a[i]); sum[i] = sum[i-1] + a[i]; } sort(sum+1,sum+1+n); int len = unique(sum+1,sum+n+1) - (sum+1); for(int i = 1;i <= k;i++) scanf("%d",&b[i]); int ans = 0; for(int i = 1;i <= k;i++) { for(int j = 1;j <= len;j++){ int tmp = b[i] - sum[j]; s.push_back(tmp); } } sort(s.begin(),s.end()); for(int i = 0;i < s.size();i++) { w[s[i]]++; if(w[s[i]] == k) ans++; } printf("%d\n",ans); } return 0;}