BC#32 1002 hash

时间:2022-07-02 23:43:35

代码引用kuangbin大神的,膜拜

第一次见到hashmap和外挂,看来还有很多东西要学

维护前缀和sum[i]=a[0]-a[1]+a[2]-a[3]+…+(-1)^i*a[i]
枚举结尾i,然后在hash表中查询是否存在sum[i]-K的值。
如果当前i为奇数,则将sum[i]插入到hash表中。
上面考虑的是从i为偶数为开头的情况。
然后再考虑以奇数开头的情况,按照上述方法再做一次即可。
不同的是这次要维护的前缀和是sum[i]=-(a[0]-a[1]+a[2]-a[3]+…+(-1)^i*a[i])
I为偶数的时候将sum[i]插入到hash表。
总复杂度o(n)
 #include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
const int MAXN = ;
int a[MAXN]; const int HASH = ;
struct HASHMAP
{
int head[HASH],next[MAXN],size;
long long state[MAXN];
void init()
{
size = ;
memset(head,-,sizeof(head));
}
bool check(long long val){
int h = (val%HASH+HASH)%HASH;
for(int i = head[h];i != -;i = next[i])
if(val == state[i])
return true;
return false;
}
int insert(long long val)
{
int h = (val%HASH+HASH)%HASH;
for(int i = head[h]; i != -;i = next[i])
if(val == state[i])
{
return ;
}
state[size] = val;
next[size] = head[h];
head[h] = size++;
return ;
}
} H1,H2;
template <class T>
inline bool scan_d(T &ret) {
char c; int sgn;
if(c=getchar(),c==EOF) return ; //EOF
while(c!='-'&&(c<''||c>'')) c=getchar();
sgn=(c=='-')?-:;
ret=(c=='-')?:(c-'');
while(c=getchar(),c>=''&&c<='') ret=ret*+(c-'');
ret*=sgn;
return ;
} int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int T;
int iCase = ;
scanf("%d",&T);
while(T--){
iCase++;
int n;
long long K;
scanf("%d%I64d",&n,&K);
for(int i = ;i < n;i++)
scan_d(a[i]);
H1.init();
H2.init();
long long sum = ;
bool flag = false;
H1.insert();
H2.insert();
for(int i = n-;i >= ;i--){
if(i&)sum -= a[i];
else sum += a[i];
if(i% == ){
if(H1.check(sum-K))
flag = true;
}
else {
if(H2.check(-sum-K))
flag = true;
}
if(flag)break;
H1.insert(sum);
H2.insert(-sum);
}
if(flag)printf("Case #%d: Yes.\n",iCase);
else printf("Case #%d: No.\n",iCase);
}
return ;
}