HDU5183Negative and Positive (NP)(哈希表)

来源：互联网发布：java入门视频百度网盘编辑：程序博客网时间：2024/04/30 12:49

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=5183

题意：

给定一个长度为n的序列，判断是否存在

一段序列 sum（i,j）=a i −a i+1 +a i+2 +⋯+(−1) j−i aj = k;

分析：

我们维护一段前缀和

sum[j]表示从开始到第j个元素，即 sum(0,j);

然后我们将其加入到哈希表中

因为不同要么是正负正。。。。，要么就是负正负。。。。的结构

因此我们维护两个哈希表

表1插入 sum ,表2插入-sum;

当i为奇数的时候在表2中查询 sum-k;

当i为偶数的时候在表1中查询 sum+k

然后判断存不存在就好

代码如下：

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;typedef long long LL;const int MAXN=1000010;const int HASH=1000007;inline LL read()//输入外挂{    char ch=getchar();LL x=0,f=1;    while(ch>'9'||ch<'0'){if(ch=='-')f=-1;ch=getchar();}    while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}    return x*f;}struct hashmap//建立哈希表{    LL a[MAXN];    int head[HASH],next[MAXN],size;    void init(){//初始化        memset(head,-1,sizeof(head));        size=0;    }    bool find(LL val){//查找一个元素是否在哈希表内        int tmp = (val%HASH + HASH)%HASH;        for(int i = head[tmp];i!=-1;i=next[i])            if(val==a[i]) return true;        return false;    }    void add(LL val){//添加元素到哈希表中        int tmp =(val%HASH+HASH)%HASH;        if(find(val)) return;        a[size]=val;        next[size]=head[tmp];        head[tmp]=size++;    }}h1,h2;LL a[MAXN];int main(){    int t,n,cas=1,k;    t=read();    while(t--){        n=read();        k=read();        for(int i=0;i<n;i++)            a[i]=read();        LL sum=0;        h1.init(),h2.init();        h1.add(0),h2.add(0);        bool flag = 0;        for(int i=n-1;i>=0;i--){            if(i&1) sum-=a[i];            else sum+=a[i];            if(i%2==0){                if(h1.find(sum-k)) flag=1;            }            else{                if(h1.find(sum+k)) flag=1;            }            h1.add(sum);            h2.add(-sum);            if(flag) break;        }        printf("Case #%d: ",cas++);        if(flag)puts("Yes.");        else puts("No.");    }    return 0;}

0 0