19.Trie

2020-03-12  本文已影响0人  Tsukinousag
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int son[N][26],cnt[N],idx;
char str[N];
void Insert(char str[])
{
    int p=0;
    for(int i=0;str[i];i++)
    {
        int u=str[i]-'a';
        if(!son[p][u]) son[p][u]=++idx;
        p=son[p][u];
    }
    cnt[p]++;
}
int Query(char str[])
{
    int p=0;
    for(int i=0;str[i];i++)
    {
        int u=str[i]-'a';
        if(!son[p][u]) return 0;
        p=son[p][u];
    }
    return cnt[p];
}
int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        char a[2];
        scanf("%s%s",a,str);
        if(a[0]=='I') Insert(str);
        else  cout<<Query(str)<<endl;
    }
    return 0;
}

思路:

一个整数,是可以转化成为一个32位的二进制数,而也就可以变成长度为32位的二进制字符串.
既然如此话,那么我们可以这么做,每一次检索的时候,我们都走与当前a[i]这一位相反的(比如当前这一位是0,我们就看一看下一个节点的1有没有,有的话要往1走),也就是让or值最大,如果说没有路可以走的话,那么就走相同的了.

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10,M=3e6+10;
int son[M][2],idx;
int a[N];
void Insert(int x)
{
    int p=0;
    for(int i=30;~i;i--)
    {
        int temp=x>>i&1;//返回x的2进制展开里的第i位是0还是1
        if(!son[p][temp])
            son[p][temp]=++idx;
        p=son[p][temp];
    }
}
int Query(int x)
{
    int p=0,res=0;
    for(int i=30;~i;i--)
    {
        int temp=x>>i&1;
        if(son[p][!temp])
        {
            res+=1<<i;//可以贡献一个i位的(1000...)
            p=son[p][!temp];
        }
        else p=son[p][temp];
    }
    return res;
}
int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        Insert(a[i]);
    }
    int res=0;
    for(int i=0;i<n;i++)
    {
        res=max(res,Query(a[i]));
    }
    cout<<res<<endl;
    return 0;
}
上一篇下一篇

猜你喜欢

热点阅读