算法基础13-数组模拟trie树

trie树相关定义直接在csdn上找别人的文章写都很生动形象，包括使用trie树的原因----节省空间等等等等。

//AcWing 835. Trie字符串统计 
#include <iostream>

using namespace std;

const int N = 1e5 + 10;

//son数组的第一维大小和字符串长度有关，因为字符串长度有多长，trie树就需要多少个节点，极端情况下每个字符都可能是一个parent，
//son数组的第二大小和一个节点能开多少叉有关，如果是最大异或对那题一个节点开叉最多0和1所以第二维是2，这里一个节点经过-‘a'处理最多开叉为26个数；
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;//将trie树里的idx想象成链表里面的东西，它的作用是给当前操作的节点一个唯一的id号，以此id号方便以后找到这个节点
         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;//如果这个节点不存在说明树中压根就没存这个字符串直接返回0即可
        p = son[p][u];
    }
    return cnt[p];
}

int main()
{
    int n;
    scanf("%d", &n);
    while (n -- )
    {
        char op[2];
        scanf("%s%s", op, str);
        if (*op == 'I') insert(str);
        else printf("%d\n", query(str));
    }
    return 0;
}

AcWing 143. 最大异或对，很好的题目，每次卡我很久

//方法1，本方法query函数返回的是一条路径，不是最终的结果值，当初卡我很久QAQ
//并且现在可以过样例，但是ac不了不知道为啥
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10, M = 31 * N;

int son [M][2], a[N], idx;

void insert(int x)
{
    int p =0;
    for (int i = 30; i >= 0; i -- )
    {
        int u = x >> i & 1;
        int &s = son[p][u];//这里轩了个技，意思是用s的地址指向的就是son[p][u],后面把s看作son[p][u]就行
        if (!s) s = ++ idx ;
        p = s;
    }
}

int query(int x)//返回的是可以找到的一条异或之后最大的路径
{
    int res = 0;//res是我们要返回的一条路径，就一个要用来异或的数
    int p = 0;
    for (int i = 30; i >= 0; i -- )
    {
        int u = x >> i & 1;
        if (son[p][!u]) 
        {
            p = son[p][!u];
            //一直没ac原因就是<< 优先级小于 + ，这里必须加括号，这里和res += 1 << i哪里不一样，+=会让res加上+=右边运算完的结果，因此不用加括号
            res = (res << 1) + !u;//可以<<代替乘2，类似十进制里面的 res * 10 + u
        }
        else 
        {
            p = son[p][u];
            res = (res << 1) + u;
        }    
    }
    return res;
}
int main()
{
    int n;
    scanf("%d", &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 ++ )
    {
        int t = query(a[i]);
        res = max(res, a[i]^t);
    }
    
    printf("%d", res);
    return 0;
}

注意方法一一直没ac原因就是res = res << 1 + u中<< 优先级小于 + ，必须加括号。这里和res += 1 << i那里不一样，+=会让res加上+=右边运算完的结果，相当于编译器自己帮我们加了一个括号，因此不用加括号

//方法2，query函数返回的就是数列里面两个树异或后的最大结果
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10, M = 31 * N;

int son[M][2], a[N], idx;

void insert(int x)
{
    int p =0;
    for (int i = 30; i >= 0; i -- )
    {
        int u = x >> i & 1;
        int &s = son[p][u];//这里轩了个技，意思是用s的地址指向的就是son[p][u],后面把s看作son[p][u]就行
        if (!s) s = ++ idx ;
        p = s;
    }
}

int query(int x)//返回的是可以找到的一条异或之后最大的路径
{
    int res = 0;//res是我们要返回的一条路径，就一个要用来异或的数
    int p = 0;
    for (int i = 30; i >= 0; i -- )
    {
        int u = x >> i & 1;//x >> i & 1在位运算里表示，获得从右往左第i位的二进制值是多少，是0 还是 1
        if (son[p][!u]) 
        {
            p = son[p][!u];
            res += 1 << i;//找到了异或后为1的路径，返回的是异或后的结果值，因此要加上向左位移i位后的1
        }
        else 
        {
            p = son[p][u];
           // res += 0 << i;没找到异或后为1的路径，出事化的时候res所有位上都是0，因此不需要加上向左位移i位后的0了，本身也没有任何意义
        }    
    }
    return res;
}
int main()
{
    int n;
    scanf("%d", &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]));
    }
    
    printf("%d", res);
    return 0;
}