题意:问有没有数对(i,j)(0<=i<=j<n),使得a[i]-a[i+1]+...+(-1)^(j-i)a[j]为K.
解法:两种方法,枚举起点或者枚举终点。
先保存前缀和:a1-a2+a3....+/- an
枚举起点法: 设起点为x,实际是枚举x-1,分两种情况:
1.起点x为奇,那么就看有没有a[j]-a[x-1] = K的,即a[j] = a[x-1]+K。因为奇数位置的ai数符为正。
2.起点x为偶,那么就看有没有a[j]-(-K) = a[x-1],即a[j] = a[x-1]-K。因为偶数位置ai数符为负,即x到j这一段的数是负的 选x为起点的x到j的这一段和,所以中间实际上是-K。
每次将sum[i]标记为出现过。
只需要一个hashmap即可。
由于枚举到一个起点x,需要判断a[j](j>x)是否出现,所以要逆序枚举。
代码:
#include <iostream> #include <cstdio> #include <cstring> #include <cstdlib> #include <cmath> #include <algorithm> #define Mod 1000000007 #define lll __int64 #define ll long long using namespace std; #define N 1000007 lll sum[N]; const unsigned long long SMod=1000007; struct hashmap{ struct Edge { long long num; int next; }; Edge edge[2*N]; int countedge; int head[SMod+100]; void init() { memset(head,-1,sizeof(head)); countedge=0; } void addedge(long long num) { int start=num%SMod; edge[countedge].next=head[start]; edge[countedge].num=num; head[start]=countedge; countedge++; } int Find(long long num) { int start=num%SMod; int ind; for(ind=head[start]; ind!=-1; ind=edge[ind].next) { if(edge[ind].num==num)break; } return ind; } }ST; int main() { int n,i,j,cs = 1,t,x,K; scanf("%d",&t); while(t--) { ST.init(); scanf("%d%d",&n,&K); sum[0] = 0; for(i=1;i<=n;i++) { scanf("%d",&x); if(i%2) sum[i] = sum[i-1] + x; else sum[i] = sum[i-1] - x; } ST.addedge(sum[n]); int tag = 0; for(i=n-1;i>=0;i--) { if(i%2 == 0 && ST.Find(sum[i]+K) != -1) { tag = 1; break; } if(i%2 && ST.Find(sum[i]-K) != -1) { tag = 1; break; } ST.addedge(sum[i]); } printf("Case #%d: ",cs++); if(tag) puts("Yes."); else puts("No."); } return 0; }
枚举终点法:
建立两个hashmap,一个记录sum[1],sum[3],...sum[2*cnt+1] (2*cnt+1<=n)即奇数位置是否出现过,另一个记录偶数位置的sum值是否出现过。
枚举终点y的话,起点可能是1~y的任何一个(这里下标从题目中的0~n-1转为了1~n),当起点x=1的时候,这时NP-SUM(x,y) = sum[y], 记为XX。以n=4为例。
那么起点为2的时候整个值就等于 -XX+a1, (-(a1-a2+a3-a4) +a1 = a2-a3+a4))
起点为3的时候整个值等于 XX-sum[2] (a1-a2+a3-a4 - (a1-a2) = a3-a4 )
...以此类推,归为两类 :
1. XX-sum[0] , XX-sum[2] , ... XX-sum[偶数] 是否为K
2. -XX+sum[1], -XX+sum[3], ... -XX+sum[奇数] 是否为K
设他们为K,那么即判断 XX-K在偶数的hashmap中有没有出现, 判断XX+K在奇数的hashmap中有没有出现。
每次将sum[i]加入到对应的hashmap中。
顺序枚举。
代码:
#include <iostream> #include <cstdio> #include <cstring> #include <cstdlib> #include <cmath> #include <algorithm> #define Mod 1000000007 #define lll __int64 #define ll long long using namespace std; #define N 1000007 const unsigned long long SMod=1000007; struct hashmap{ struct Edge { long long num; int next; }; Edge edge[2*N]; int countedge; int head[SMod+100]; void init() { memset(head,-1,sizeof(head)); countedge=0; } void addedge(long long num) { int start=num%SMod; edge[countedge].next=head[start]; edge[countedge].num=num; head[start]=countedge; countedge++; } int Find(long long num) { int start=num%SMod; int ind; for(ind=head[start]; ind!=-1; ind=edge[ind].next) { if(edge[ind].num==num)break; } return ind; } }mpe,mpo; int main() { int n,i,j,cs = 1,t,x,K; scanf("%d",&t); for(cs=1;cs<=t;cs++) { mpo.init(); mpe.init(); scanf("%d%d",&n,&K); lll sum = 0; mpe.addedge(0); int tag = 0; for(i=1;i<=n;i++) { scanf("%d",&x); if(i&1) sum += x; else sum -= x; if(i&1) mpo.addedge(sum); else mpe.addedge(sum); if(mpe.Find(sum-K) != -1) { tag = 1; } if(mpo.Find(sum+K) != -1) { tag = 1; } } printf("Case #%d: ",cs); if(tag) puts("Yes."); else puts("No."); } return 0; }
注意:
如果hashmap中的SMod 用宏定义的方式就会T, 用const unsigned long long 就不会。不知道为什么。
hashmap模板借鉴了love_dn的代码。