[HDU 5183][BestCoder #32]Negative and Positive (NP) (STL模板)

题目链接

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

题目大意

给定一个数组 (a0,a1,a2,⋯an−1) 和一个整数 K ， 请来判断一下是否存在二元组 (i,j)(0≤i≤j<n) 使得 NP−sum(i,j) 刚好为 K 。这里 NP−sum(i,j)=ai−ai+1+ai+2+⋯+(−1)j−iaj 。

思路

sum[i]=a1−a2+a3... 。用一个set当成哈希表来保存所有出现过的 sum 值，预处理完 sum 数组后，从后往前倒着遍历 sum 数组，先加入 sum[i] ,然后根据 i 的奇偶性分类判断在哈系表中是否存在对应的 j(i<=j) 满足题意
看代码就比较清楚了。
我比赛时用的是map然后TLE了，伤心

代码

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
#include <map>
#include <set>

#define MAXN 2000100

using namespace std;

typedef long long int LL;

//map<LL,bool>mp;
set<LL>mp;

LL a[MAXN],sum[MAXN];

inline LL read()
{
    LL ans=0,sign=1;
    char ch=getchar();
    while(ch==' '||ch=='\n') ch=getchar();
    if(ch=='-') sign=-1,ch=getchar();
    while(ch>='0'&&ch<='9') ans=ans*10+ch-'0',ch=getchar();
    return sign*ans;
}

int main()
{
    int T;
    scanf("%d",&T);
    for(int Case=1;Case<=T;Case++)
    {
        bool flag=false;
        mp.clear();
        LL n,K;
        n=read(),K=read();
        for(int i=1;i<=(int)n;i++)
            a[i]=read();
        for(int i=1;i<=(int)n;i++)
        {
            if(i&1)
                sum[i]=sum[i-1]+a[i];
            else
                sum[i]=sum[i-1]-a[i];
        }
        for(int i=(int)n;i>=1;i--)
        {
            mp.insert(sum[i]);
            if(flag) continue;
            if(i&1)
            {
                if(mp.find(sum[i-1]+K)!=mp.end())
                {
                    flag=true;
                    break;
                }
            }
            else
            {
                if(mp.find(sum[i-1]-K)!=mp.end())
                {
                    flag=true;
                    break;
                }
            }
        }
        if(!flag) printf("Case #%d: No.\n",Case);
        else printf("Case #%d: Yes.\n",Case);
    }
    return 0;
}