题面:cf831c
简要题意:给出k个数a[i],给出n个b[i],每个b[i]值为a[1]+a[2]+…+a[不知道位置]+某一未知初始值(每种情况初始值是一样的),求可能的初始值有多少
n<=2000
,直接暴力即可
我们枚举b[1]这个值出现的位置在哪里,然后往左往右一一匹配,如果每个b[i]都出现过了说明这个初始值是合法的,用map判断去重后答案加1
时间复杂度
O(n2)
只不过预处理的时候写的姿势要好,否则要小心TLE
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <iostream>
#include <ctime>
#include <map>
#include <queue>
#include <cstdlib>
#include <string>
#include <climits>
#include <set>
#include <vector>
using namespace std;
map<int,bool>mp;
int k[8000010],q[4010],a[4010],b[4010];
int main()
{
int K,n;scanf("%d%d",&K,&n);
for(int i=1;i<=K;i++)scanf("%d",&a[i]);
for(int i=1;i<=n;i++){
scanf("%d",&b[i]);
k[b[i]+4000000]++;
}
int ans=0;
for(int i=1;i<=K;i++){
int p=0;memset(q,0,sizeof q);
int r=b[1];for(int j=i+1;j<=K;j++){
r+=a[j];
if(r>=-4000000&&r<=4000000&&k[r+4000000])p++,k[r+4000000]--,q[j]=1;
}
r=b[1];for(int j=i;j;j--){
if(r>=-4000000&&r<=4000000&&k[r+4000000])p++,k[r+4000000]--,q[j]=1;
r-=a[j];
}
if(p==n&&!mp[r])mp[r]=1,ans++;
for(int j=1;j<=K;j++){
r+=a[j];if(q[j])k[r+4000000]++;
}
}
printf("%d",ans);
return 0;
}