[FZSZOJ 1223] 上海红茶馆

1223: 上海红茶馆　～ Chinese Tea

时间限制: 1 Sec内存限制: 128 MB

题目描述

你现在正在经营一家红茶馆， 而且这里有各种各样的红茶， 你现在把这些红茶分成了N个等级， 每个等级的茶有一个品质Q。

现在每一个来的客人都会要求一个品质为S的茶， 你需要迅速的回答他是否有。

输入

第一行两个数N,M。
下面一行N个数， 分别表示每个等级的茶的品质Q。
下面一行M个数， 分别表示询问的品质S。

输出

输出一行M个字符， 表示回答是否。 Y表示有， N表示没有。

样例输入

5 5
1 3 4 6 8
1 2 3 4 5

样例输出

YNYYN

提示

30%：N,M<5000

100%：N,M<200000

【题解】

本题有多种解法：

首先我想到自己开一个bool数组来哈希一下，发现哈希的不够优秀，过了85%。

然后我就想到了排序+二分，这样就可以过了，1800ms

#include<stdio.h>
#include<algorithm>
using namespace std;
int n,m,a[200001];
inline int getint() {
    int x=0,f=1; char c;
    c=getchar();
    while(c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while(c>='0'&&c<='9') {x=(x<<3)+(x<<1)+c-'0';c=getchar();}
    return x*f;
}
int main() {
    n=getint();m=getint();
    for (int i=1;i<=n;++i)a[i]=getint();
    sort(a+1,a+n+1);
    bool f=0;
    for (int i=1;i<=m;++i,f=0) {
        int x,l=1,r=n; x=getint();
        while(l<=r) { //cout<<l<<' '<<r<<endl;
            if(a[l]==x||a[r]==x) {f=1; break;}
            int mid=(l+r)>>1;
            if(a[mid]==x) {f=1;break;}
            if(x<a[mid]) r=mid-1;
            else if(x>a[mid]) l=mid+1;
        }
        if(f) printf("Y");
        else printf("N");
    }
    printf("\n");
    return 0;
}

然后我就发现了，可以用STL的map！就又写了一个简单多的程序，AC了，4000ms

#include<stdio.h>
#include<map> 
#include<algorithm>
using namespace std;
int n,m;
map<int,bool> a; 
inline int getint() {
    int x=0,f=1; char c;
    c=getchar();
    while(c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while(c>='0'&&c<='9') {x=(x<<3)+(x<<1)+c-'0';c=getchar();}
    return x*f;
}
int main() {
    n=getint();m=getint();
    for (int i=1;i<=n;++i) {
        int x;x=getint();
        a[x]=1;
    }
    
    bool f=0;
    for (int i=1;i<=m;++i,f=0) {
        int x,l=1,r=n; x=getint();
        if(a[x]==1) printf("Y");
        else printf("N");
    }
    printf("\n");
    return 0;
}

还是二分快啊TAT我的哈希还是炸啊QAQQQQQ