【蓝桥杯集训10】Tire树 字典树 最大异或对专题(3 / 3)

目录

字典树模板

1、插入操作

2、查询操作

143. 最大异或对 - trie + 二进制

3485. 最大异或和 - 前缀和+Trie+滑动窗口


字典树模板

活动 - AcWing

字典树:高效存储和查找字符串集合的数据结构

  • son[节点1地址][值]=节点2地址 —— 节点1的子节点为节点2
  • cnt[节点地址] —— 记录以该节点地址为结尾的字符串个数

1、插入操作

static void insert(String s)
{
    int p=0;
    for(char x:s.toCharArray())
    {
        int u=x-'a';
        if(son[p][u]==0) son[p][u]=++idx; //如果节点不存在 创建节点
        p=son[p][u]; //走到p的子节点
    }
    cnt[p]++; //把这一串字符插入后 记录以最后此节点为标记的字符串数
}

2、查询操作

static int query(String s)
{
    int p=0;
    for(char x:s.toCharArray())
    {
        int u=x-'a';
        if(son[p][u]==0) return 0; //有一个节点不存在,说明没有这个字符串
        p=son[p][u];
    }
    return cnt[p];
}
/*
    *道阻且长,行则将至*
    author:Roye_ack
*/
import java.util.*;
import java.io.*;
import java.math.*;
 
class Main
{
    static int N=100010;
    static int[] cnt=new int[N];
    static int[][] son=new int[N][26]; //son[节点1][值]=节点2  节点1的子节点是节点2
    static int idx=0;
    
    static void insert(String s)
    {
        int p=0;
        for(char x:s.toCharArray())
        {
            int u=x-'a';
            if(son[p][u]==0) son[p][u]=++idx; //如果节点不存在 创建节点
            p=son[p][u]; //走到p的子节点
        }
        cnt[p]++; //把这一串字符插入后 记录以最后此节点为标记的字符串数
    }
    
    static int query(String s)
    {
        int p=0;
        for(char x:s.toCharArray())
        {
            int u=x-'a';
            if(son[p][u]==0) return 0; //有一个节点不存在,说明没有这个字符串
            p=son[p][u];
        }
        return cnt[p];
    }
    
    public static void main(String[] args) throws IOException
    {
        PrintWriter wt=new PrintWriter(new BufferedWriter(new OutputStreamWriter(System.out)));
        
        int n=rd.nextInt();
        while(n-->0)
        {
            String[] s=rd.nextLine().split(" ");
            if(s[0].equals("I")) insert(s[1]);
            else wt.println(query(s[1]));
        }
        
        wt.flush();
    }
    
    static class rd
    {
        static BufferedReader bf=new BufferedReader(new InputStreamReader(System.in));
        static StringTokenizer tk=new StringTokenizer("");
        
        static String nextLine() throws IOException
        {
            return bf.readLine();
        }
        
        static String next() throws IOException
        {
            while(!tk.hasMoreTokens()) tk=new StringTokenizer(bf.readLine());
            return tk.nextToken();
        }
        
        static int nextInt() throws IOException
        {
            return Integer.parseInt(next());
        }
        
        static double nextDouble() throws IOException
        {
            return Double.parseDouble(next());
        }
        
        static long nextLong() throws IOException
        {
            return Long.parseLong(next());
        }
        
        static BigInteger nextBig() throws IOException
        {
            BigInteger d=new BigInteger(rd.nextLine());
            return d;
        }
    }
}

143. 最大异或对 - trie + 二进制

活动 - AcWing

题目:

在给定的N个整数 A1,A2.......AN中选出两个进行xor(异或)运算,得到的结果最大是多少?

思路:

将每个数以二进制方式存入 Trie 树

每一次检索,我们都走与当前 ai 这一位相反的位置走,也就是让异或值最大

如果没有相反位置的路可以走的话,那么就走相同位置的路

/*
    *道阻且长,行则将至*
    author:Roye_ack
*/
import java.util.*;
import java.io.*;
import java.math.*;
 
class Main
{
    static int N=100010,M=31*N;
    static int[] a=new int[N];
    static int[][] son=new int[M][2]; //son[节点1][值]=节点2  节点1的子节点是节点2
    static int idx=0;
    
    static void insert(int x)
    {
        int p=0;
        for(int i=30;i>=0;i--) //最大31位,0~30位
        {
            int u=x>>i&1;
            if(son[p][u]==0) son[p][u]=++idx;
            p=son[p][u];
        }
    }
    
    static int find(int x)
    {
        int p=0,res=0;
        for(int i=30;i>=0;i--)
        {
            int u=x>>i&1;
            if(son[p][u^1]!=0) //如果当前层有对应不同的数,走不同的分支
            {
                p=son[p][u^1];
                res=res*2+1; //该位异或后为1,将其左移i位加到res中
            }
            else 
            {
                p=son[p][u];
                res=res*2+0;
            }
        }
        return res;
    }
    
    public static void main(String[] args) throws IOException
    {
        PrintWriter wt=new PrintWriter(new BufferedWriter(new OutputStreamWriter(System.out)));
        
        int n=rd.nextInt();
        for(int i=0;i<n;i++)
        {
            a[i]=rd.nextInt();
            insert(a[i]);
        }
        int res=0;
        for(int i=0;i<n;i++) res=Math.max(res,find(a[i]));
        wt.print(res);
        wt.flush();
    }
    
    static class rd
    {
        static BufferedReader bf=new BufferedReader(new InputStreamReader(System.in));
        static StringTokenizer tk=new StringTokenizer("");
        
        static String nextLine() throws IOException
        {
            return bf.readLine();
        }
        
        static String next() throws IOException
        {
            while(!tk.hasMoreTokens()) tk=new StringTokenizer(bf.readLine());
            return tk.nextToken();
        }
        
        static int nextInt() throws IOException
        {
            return Integer.parseInt(next());
        }
        
        static double nextDouble() throws IOException
        {
            return Double.parseDouble(next());
        }
        
        static long nextLong() throws IOException
        {
            return Long.parseLong(next());
        }
        
        static BigInteger nextBig() throws IOException
        {
            BigInteger d=new BigInteger(rd.nextLine());
            return d;
        }
    }
}

 

3485. 最大异或和 - 前缀和+Trie+滑动窗口

3485. 最大异或和 - AcWing题库

题目:

思路:

本题要求异或和的最大值

求某一段区间的异或和可以求前缀异或和,也就是S[r]^S[l-1],因为a^x^x=a

因此题目就转化为求异或前缀和S[1]~S[N]中选一个最大的异或对

维护一个长度不超过m的滑动窗口,把窗口外的数从字典树删除,删除操作只要让cnt--就可以

长度不超过m的连续子数组,我们可以枚举右端点s[i],在合法区间内求最大的另一异或数

最大异或对模板

  • 想要异或结果最大,则每一位都要尽量不同,这样异或结果每一位1的个数才会最多
  • 将每个数字的二进制数(01串),从高位到低位存到trie中
  • 对于每一位ai,都尽量往跟他相反的方向走(比如ai为0则走1分支,1则走0分支),如果实在没有相反的分支,则顺着走
/*
    *道阻且长,行则将至*
    author:Roye_ack
*/
import java.util.*;
import java.io.*;
import java.math.*;
 
class Main
{
    static int N=3100010;
    static int[] s=new int[N],cnt=new int[N]; //cnt记录数的出现次数
    static int[][] tr=new int[N][2]; //son[节点1][值]=节点2  节点1的子节点是节点2
    static int idx=0;
    
    static void insert(int x,int num)
    {
        int p=0;
        for(int i=30;i>=0;i--)
        {
            int u=x>>i&1;
            if(tr[p][u]==0) tr[p][u]=++idx;
            p=tr[p][u];
            cnt[p]+=num;
        }
    }
    
    static int find(int x)
    {
        int p=0,res=0;
        for(int i=30;i>=0;i--)
        {
            int u=x>>i&1;
            if(cnt[tr[p][u^1]]!=0)
            {
                p=tr[p][u^1];
                res=res*2+1;
            }
            else
            {
                p=tr[p][u];
                res*=2;
            }
        }
        return res;
    }
    
    public static void main(String[] args) throws IOException
    {
        PrintWriter wt=new PrintWriter(new BufferedWriter(new OutputStreamWriter(System.out)));
        
        int n=rd.nextInt(),m=rd.nextInt();
        for(int i=1;i<=n;i++)
        {
            int x=rd.nextInt();
            s[i]=s[i-1]^x;
        }
        
        int res=0;
        insert(s[0],1); //刚开始插入s0
        for(int i=1;i<=n;i++) //枚举右端点
        {
            if(i>m) insert(s[i-m-1],-1); //删掉不在区间内的左端点(合法区间是[i-m,i])
            res=Math.max(res,find(s[i]));
            insert(s[i],1);
        }
        wt.print(res);
        wt.flush();
    }
    
    static class rd
    {
        static BufferedReader bf=new BufferedReader(new InputStreamReader(System.in));
        static StringTokenizer tk=new StringTokenizer("");
        
        static String nextLine() throws IOException
        {
            return bf.readLine();
        }
        
        static String next() throws IOException
        {
            while(!tk.hasMoreTokens()) tk=new StringTokenizer(bf.readLine());
            return tk.nextToken();
        }
        
        static int nextInt() throws IOException
        {
            return Integer.parseInt(next());
        }
        
        static double nextDouble() throws IOException
        {
            return Double.parseDouble(next());
        }
        
        static long nextLong() throws IOException
        {
            return Long.parseLong(next());
        }
        
        static BigInteger nextBig() throws IOException
        {
            BigInteger d=new BigInteger(rd.nextLine());
            return d;
        }
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值