暴力+map——Codeforces831C Jury Marks

来源:互联网 发布:淘宝无忧退货流程 编辑:程序博客网 时间:2024/06/06 19:15

题面: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;}
原创粉丝点击