题意:一个比赛中,有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;
}