查找（二分、hash、桶）

先上一个最简单的题

1230 元素查找

给出n个正整数，然后有m个询问，每个询问一个整数，询问该整数是否在n个正整数中出现过。

输入描述 Input Description

第一行两个整数 n 和m。

第二行n个正整数（1<=n<= 100000）

第三行m个整数(1<=m<=100000)

输出描述 Output Description

一共m行，若出现则输出YES，否则输出NO

 

这个题数据很小，既可以用二分、桶、做，也可以用hash来做

用桶的思想来做的话是最简单的，定义一个bool数组，出现的正整数在数组里标记为TRUE；查找sum有没有出现过时只需要调用bool下标为sum的数组就行。

#include<iostream>
using namespace std;

bool s[100005];
int m,n;

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;++i)
    {
        int sum;
        cin>>sum;
        s[sum]=true;
    }
    for(int i=1;i<=m;++i)
    {
        int sum;
        cin>>sum;
        if(s[sum])cout<<"YES\n";
        else cout<<"NO\n";
    }
    return 0;
}

桶

如果数据进一步升级，出现过的数进一步增大，空间进一步限制，这时用bool就不行了，需要用到二分查找，先存数，然后查找即可；

#include<iostream>
#include<algorithm>
#include<string>
using namespace std;

bool boo[150000];
int s[150000];
int n,m;

bool  find(int sum)
{
    int right=n,left=1;
    while(right>=left)
    {
        int mid=left+(right-left)/2;
        if(s[mid]==sum)return true;
        if(mid<sum)left=mid+1;
        else right=mid-1;
    }
    return 0;
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;++i)
        cin>>s[i];
    sort(s+1,s+n+1);
    for(int i=1;i<=m;++i)
    {
        int sum;
        cin>>sum;
        if(find(sum))
            cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}

二分查找

最后，如果再进一步升级，咳咳，还是用hash吧；

这里是把数模一个数，然后建立以余数和数的边，查找是将待查找的数取余，查找余数所连接的边。

#include<iostream>
#define mod 4567;
using namespace std;

struct node
{
    int a,b,next;
};
node s[100006];

int head[100005],sum=1;
int m,n;

void push(int a,int b)
{
    s[sum].a=a;
    s[sum].b=b;
    s[sum].next=head[a];
    head[a]=sum++;
}

bool find (long long sum)
{
    int x=sum%mod;
    int y=sum/mod;
    for(int k=head[x];k!=-1;k=s[k].next)
        if(s[k].b==y)return true;
}

int main()
{
    for(int i=1;i<=69439;++i)
        head[i]=-1;
    int m,n;
    cin>>m>>n;
    for(int i=1;i<=m;++i)
    {
        long long sum;
        cin>>sum;
        int x=sum%mod;
        int y=sum/mod;
        push(x,y);
    }
    for(int i=1;i<=n;++i)
    {
        long long sum;
        cin>>sum;
        if(find(sum))
            cout<<"YES\n";
        else cout<<"NO\n";
    }
    return 0;
}

hash