【力扣:1707 1803】0-1字典树

思路:树上每个节点存储拥有该节点的数组元素的最小值,left节点表示0,right节点表示1,构建完成后遍历树当子节点没有比mi小的元素时直接输出-1,否则向下构造。

cpp 复制代码
struct tree{
    int m;
    tree*left=nullptr,*right=nullptr;
    tree(int val=INT_MAX):m(val){}
};
class Solution {
    tree*root=new tree;
    void add(int val){
        tree*cur=root;
        for(int i=31;i>=0;i--){
            if(1<<i&val){
                if(!cur->right) cur->right=new tree(val);
                else cur->right->m=min(val,cur->right->m);
                cur=cur->right;
            }
            else{
                if(!cur->left) cur->left=new tree(val);
                else cur->left->m=min(val,cur->left->m);
                cur=cur->left;
            }
        }
    }
    int find(int val,int tar){
        int x=0;
        tree*cur=root;
        for(int i=31;i>=0;i--){
            if(1<<i&val){
                if(cur->left&&cur->left->m<=tar) x|=1<<i,cur=cur->left;
                else if(cur->right&&cur->right->m<=tar) cur=cur->right;
                else return -1;
            }
            else {
                if(cur->right&&cur->right->m<=tar) x|=1<<i,cur=cur->right;
                else if(cur->left&&cur->left->m<=tar) cur=cur->left;
                else return -1;
            }
        }
        return x;
    }
public:
    vector<int> maximizeXor(vector<int>& nums, vector<vector<int>>& queries) {
        //sort(nums.begin(),nums.end());
        for(int i:nums) add(i);
        vector<int> res;
        for(auto& i:queries){
            res.push_back(find(i[0],i[1]));
        }
        return res;
    }
};

思路:已知nums[i],nums[j]异或值应在[low,high]之间,因而可以转化为小于high的数量减去小于low-1的数量,将问题转化为两个数的异或值小于target的数量,所以树的节点应该记录该节点下元素的数量,然后按位构造当target的此位是0的时候不能构造为1,当target的此位是1时可以构造为0或1,可以直接加上0节点下元素数量,然后向1处接着构造,这样累加之后就得到了异或值小于target的数量

cpp 复制代码
struct tree{
    int cnt=0;
    tree*children[2]={nullptr,nullptr};
};
class Solution {
    tree*root;
    void add(int val){
        tree*cur=root;
        for(int i=31;i>=0;i--){
            int index=val>>i&1;
            if(!cur->children[index]) {
                cur->children[index]=new tree;
            }
            cur=cur->children[index];
            cur->cnt++;
        }
    }
    int find(int val,int m){
        int x=0;
        tree*cur=root;
        for(int i=31;i>=0;i--){
            int index=val>>i&1;
            if(m>>i&1){
                if(cur->children[index]) x+=cur->children[index]->cnt;
                if(!cur->children[index^1]) return x;
                cur=cur->children[index^1];
            }
            else {
                if(!cur->children[index]) return x;
                cur=cur->children[index];
            }
        }
        return x+cur->cnt;
    }
    int f(vector<int>& nums,int x){
        root=new tree;
        int res=0;
        for(int i=1;i<nums.size();i++){
            add(nums[i-1]);
            res+=find(nums[i],x);
        }
        return res;
    }
public:
    int countPairs(vector<int>& nums, int low, int high) {
        return f(nums,high)-f(nums,low-1);
    }
};
相关推荐
岁忧1 小时前
(nice!!!)(LeetCode 面试经典 150 题 ) 30. 串联所有单词的子串 (哈希表+字符串+滑动窗口)
java·c++·leetcode·面试·go·散列表
SunkingYang2 小时前
MFC/C++语言怎么比较CString类型最后一个字符
c++·mfc·cstring·子串·最后一个字符·比较
界面开发小八哥2 小时前
MFC扩展库BCGControlBar Pro v36.2新版亮点:可视化设计器升级
c++·mfc·bcg·界面控件·ui开发
R-G-B2 小时前
【15】MFC入门到精通——MFC弹窗提示 MFC关闭对话框 弹窗提示 MFC按键触发 弹窗提示
c++·mfc·mfc弹窗提示·mfc关闭弹窗提示·mfc按键触发 弹窗提示
艾莉丝努力练剑2 小时前
【数据结构与算法】数据结构初阶:详解顺序表和链表(四)——单链表(下)
c语言·开发语言·数据结构·学习·算法·链表
十秒耿直拆包选手2 小时前
Qt:QCustomPlot类介绍
c++·qt·qcustomplot
珊瑚里的鱼2 小时前
第十三讲 | map和set的使用
开发语言·c++·笔记·visualstudio·visual studio
逑之2 小时前
C++笔记1:命名空间,缺省参数,引用等
开发语言·c++·笔记
berlin51512 小时前
c++判断文件或目录是否存在
c++
L_autinue_Star3 小时前
从0到1实现Shell!Linux进程程序替换详解
linux·运维·服务器·c++·chrome