算法刷题:最大异或对(Trie树扩展)、食物链(并查集扩展)

2023-12-29 19:32:01

引言

这两个扩展题能够让我们更加的熟悉Trie树和并查集的使用,这两道题可以说是比较难的了,所以说还是好好对待吧。

一、最大异或对(Trie树扩展)

1.题目描述

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

输入格式
第一行输入一个整数 N。
第二行输入 N 个整数 A1~AN。

输出格式
输出一个整数表示答案。

数据范围
1≤N≤105,0≤Ai<231
 
输入样例:
3
1 2 3

输出样例:
3

2.解题思路

  • 首先最开始当然是用暴力做一下,如下所示,很明显是O(n^2)的一个时间复杂度,然后这个数据是10 ^ 5,那就是执行10 ^ 10次,然后一般C++的时间范围在10 ^ 9内一般都是可以过的,这个当然就TTL了。
int res = 0;
for (int i = 0; i < n; ++i)
{
	for (int j = 0; j < i; ++j)
	{
		res = max(res, a[i] ^ a[j]);
	}
}
  • 然后就想怎么优化了,我们注意第二层循环是查找当然数中与a[i]异或的最大值,我们发现数据是小于 2 ^ 31 - 1的,也就是最多31位,那么是最大的,肯定先从最高位找有没有一个数跟a[i]的最高位不一样,如果有那就把他们集合到一块,没有那就挑相同的,然后找第29位有没有与a[i]第29位不同位的树,如果没有那就找相同的,然后就这样找到第0位,肯定就是与a[i]异或最大的数了,这也是一种贪心的思想,具体图解看如下的例子:
    如果右边是已经有的Trie树了,我们要找与110异或最大的,那么最高位是1,我们就要找0,然后有0,然后看下一位是1,我们就要找0,没有0,那就只能是1了,然后下一位是0,我们要找1,那就找到了,所以最终与110异或最大的数是011
    在这里插入图片描述

3.代码实现

#include <cstdio>
#include <iostream>

using namespace std;

const int N = 1e5+10, M = 31 * N;  // 最大有M个结点

int a[N];
int son[M][2], idx;  // M个结点,每个结点两个儿子0,1
int n;

void insert(int x)
{
    int p = 0;
    for(int i = 30; i >= 0; --i)
    {
        int u = x >> i & 1;
        if(!son[p][u]) son[p][u] = ++idx;
        p = son[p][u];
    }
}

int query(int x)
{
    int p = 0, res = 0;
    for(int i = 30; i >= 0; --i)
    {
        int u = x >> i & 1;
        if(son[p][!u])  // 找不同的
        {
            res = res * 2 + !u;
            p = son[p][!u];
        }
        else  // 如果不存在,那只能退而求其次
        {
            res = res * 2 + u;
            p = son[p][u];
        }
    }
    
    return res;
}

int main()
{
    scanf("%d", &n);
    for(int i = 0; i < n; ++i) scanf("%d", &a[i]);
    
    int res = 0;
    for(int i = 0; i < n; ++i)
    {
        insert(a[i]);
        
        int t = query(a[i]);  //查询与a[i]异或最大的数
        res = max(res, a[i] ^ t);
    }
    
    printf("%d\n", res);
    
    return 0;
}

4.测试

在这里插入图片描述

在这里插入图片描述

二、食物链(并查集扩展)

1.题目描述

动物王国中有三类动物 A,B,C,这三类动物的食物链构成了有趣的环形。

A 吃 B,B 吃 C,C 吃 A。

现有 N 个动物,以 1~N 编号。

每个动物都是 A,B,C 中的一种,但是我们并不知道它到底是哪一种。

有人用两种说法对这 N 个动物所构成的食物链关系进行描述:
第一种说法是 1 X Y,表示 X 和 Y 是同类。
第二种说法是 2 X Y,表示 X 吃 Y。

此人对 N 个动物,用上述两种说法,一句接一句地说出 K 句话,这 K 句话有的是真的,有的是假的。

当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
当前的话与前面的某些真的话冲突,就是假话;
当前的话中 X 或 Y 比 N 大,就是假话;
当前的话表示 X 吃 X,就是假话。

你的任务是根据给定的 N 和 K 句话,输出假话的总数。

输入格式
第一行是两个整数 N 和 K,以一个空格分隔。
以下 K 行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中 D 表示说法的种类。
若 D=1,则表示 X 和 Y 是同类。
若 D=2,则表示 X 吃 Y。

输出格式
只有一个整数,表示假话的数目。

数据范围
1≤N≤50000 ,0≤K≤100000

输入样例:
100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

输出样例:
3

2.解题思路

这道题是要输出谎话的次数,然后x,y > n,就不说了很好判断,然后其实就是判断是不是同类,是不是能吃的关系
思路就是如下图的关系,是不是同类取决于到根结点的距离是否相同,吃与被吃的关系却决于到根结点的距离,当然这些距离都要 mod 3

3.代码实现

关于最后 d[px] 的推导我画了张图,虽然很丑,但是关系描述的还是很清晰的,然后注释写的也比较详细,不懂得可以多看看
在这里插入图片描述

在这里插入图片描述

#include <cstdio>
#include <iostream>

using namespace std;

const int N = 5e5+10;

int p[N], d[N];  // d[i]是i号结点到根结点的距离
int n, m;

int find(int x)
{
    if(x != p[x]) 
    {
        int t = find(p[x]);
        d[x] += d[p[x]];  // 必须先执行find操作,将d[p[x]是真正的到根结点的距离,而不是到上一个结点的距离
        p[x] = t;  // 路径压缩
    }
    return p[x];
}

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++i) p[i] = i;
    
    int res = 0;
    while(m--)
    {
        int t, x, y;
        scanf("%d%d%d", &t, &x, &y);
        
        int px = find(x), py = find(y);
        if(x > n || y > n) res++;
        else
        {
            if(t == 1)  // 同类
            {
                if(px == py && (d[x] - d[y]) % 3) res++;  //如果是一个集合的,并且到根结点的距离不同,说明不是一个类的
                else if(px != py)  // 刚开始还没插进集合里
                {
                    p[px] = py;  // 让x的根结点的父亲为y的根结点
                    d[px] = d[y] - d[x];  // 只用将x根结点到y根结点的距离一改,之后的话,会通过上面的find函数自动修改的
                }
            }
            else  // x吃y
            {
                int link = 1;  // 这里给出2也是可以的,x比y多1或者2 都表示x吃y
                if(px == py && (d[x] - d[y] - link) % 3) res++;
                else if(px != py)
                {
                    p[px] = py;
                    d[px] = d[y] - d[x] + link;
                }
            }
        }
    }
    
    printf("%d\n", res);
    
    return 0;
}

4.测试

在这里插入图片描述

文章来源:https://blog.csdn.net/weixin_60033897/article/details/135295151
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。