Acwing算法基础课学习笔记(五)--数据结构之Tire&&并查集&&&堆

43 篇文章 2 订阅
10 篇文章 2 订阅

Trie字符串统计

#include <iostream>

using namespace std;

const int N = 100010;

int son[N][26], cnt[N], idx;//下标是0的点,既是根节点,又是空节点
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];//走到p的子结点
	}
	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;
	cin >> n;
	while (n--)
	{
		char op[2];
		cin >> op >> str;
		if (op[0] == 'I')	insert(str);
		else cout << query(str) << endl;
	}
	return 0;
}

最大异或对

这道题目很难想到是字典树,如果不是放在字典树单元的话.
其实来说,一个整数,是可以转化成为一个32位的二进制数,而也就可以变成长度为32位的二进制字符串.
既然如此话,那么我们可以这么做,每一次检索的时候,我们都走与当前AiAi这一位相反的位置走,也就是让Xor值最大,如果说没有路可以走的话,那么就走相同的路.
这道题的启示是:字典树不单单可以高效存储和查找字符串集合,还可以存储二进制数字,而计算机所有的存储方式都是二进制的,所以这给了我们很大的操作空间。

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010, M = 31 * N;
int son[M][2], a[N], n, idx;//M代表一个数字串二进制可以到多长

void insert(int x)
{
	int p = 0;
	for (int i = 30; i >= 0; i--)
	{
		int u = x >> i & 1;//取X的第i位的二进制数是什么  x>>k&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])//如果当前层有对应的不相同的数
		{
			p = son[p][!u];
			res = res * 2 + 1; //左移一位然后加上对应的不同的位
		}
		else
		{
			p = son[p][u];
			res = res * 2 + 0;
		}
	}
	return res;
}

int main()
{
    
	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\n", res);
	return 0;
}

合并集合

关于并查集的解释
这篇博客也讲的挺好

并查集的核心代码:一个非常优美的递归,只有两行:

int find(int x)//返回x的祖宗节点 + 路径压缩
{
	if (p[x] != x)	p[x] = find(p[x]);// 根据通项公式,假设p[a]的祖先节点已知,主要思想是递归
	return p[x];
}
#include <iostream>

using namespace std;

const int N = 100010;
int n, m, p[N];
// 该函数的含义:查找a所在集合的祖先节点下标,从1开始, 并内部更新p[a]为a节点的祖先节点。
int find(int x)//返回x的祖宗节点 + 路径压缩
{
	if (p[x] != x)	p[x] = find(p[x]);// 根据通项公式,假设p[a]的祖先节点已知。
	return p[x];
}

int main()
{
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= n; i++)	p[i] = i;// 初始化每个集合

	while (m--)
	{
		char op[2];
		int a, b;
		scanf("%s%d%d", op, &a, &b);
		if (op[0] == 'M')	p[find(a)] = find(b);
		else
		{
			if (find(a) == find(b)) puts("Yes");
			else puts("No");
		}
	}

	return 0;
}

连通块中点的数量

#include <iostream>

using namespace std;

const int N = 100010;
int n, m, p[N];

// 该函数的含义:查找a所在集合的祖先节点下标,从1开始, 并内部更新p[a]为a节点的祖先节点。
int find(int x)//返回x的祖宗节点 + 路径压缩
{
	if (p[x] != x)	p[x] = find(p[x]);// 根据通项公式,假设p[a]的祖先节点已知,主要思想是递归
	return p[x];
}

int main()
{
	scanf("%d%d", &n, &m);

	int size[N];
	for (int i = 1; i <= n; i++)
	{
		p[i] = i;// 初始化每个集合
		size[i] = 1;
	}
	while (m--)
	{
		string op;
		int a, b;
		cin >> op;
		if (op == "C")
		{
			scanf("%d%d", &a, &b);
			if (find(a) == find(b))	continue;
			size[find(b)] += size[find(a)];
			p[find(a)] = find(b);
		}
		else if (op == "Q1")
		{
			scanf("%d%d", &a, &b);
			if (find(a) == find(b)) puts("Yes");
			else puts("No");
		}
		else
		{
			scanf("%d", &a);
			printf("%d\n", size[find(a)]);
		}
	}

	return 0;
}

食物链

堆排序

在这里插入图片描述

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 100010;
int n, m, cnt;
int h[N];

void down(int u)
{
	int t = u;
	if (u * 2 <= cnt && h[u * 2] < h[t])	t = u * 2;
	if (u * 2 + 1 <= cnt && h[u * 2 + 1] < h[t]) t = u * 2 + 1;
	if (u != t)
	{
		swap(h[u], h[t]);
		down(t);
	}
}
int main()
{
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= n; i++)	scanf("%d", &h[i]);
	cnt = n;

	for (int i = n / 2; i; i--)	down(i);

	while (m--)
	{
	    printf("%d ", h[1]);
		h[1] = h[cnt];
		cnt--;
		down(1);
	}

	return 0;
}

up操作:

void up(int u)
{
	while (u / 2 && h[u / 2] > h[u])
	{
		swap(h[u / 2], h[u]);
		u /= 2;
	}
}

模拟堆

难点在于交换时也要处理好映射关系,需要维护交换关系。

#include<iostream>
#include<algorithm>
using namespace std;

const int N = 1e5 + 10;
int h[N];   //堆
int ph[N];  //存放第k个插入点的下标
int hp[N];  //存放堆中点的插入次序
int cnt;   //size 记录的是堆当前的数据多少

//这个交换过程其实有那么些绕 但关键是理解 如果hp[u]=k 则ph[k]=u 的映射关系
//之所以要进行这样的操作是因为 经过一系列操作 堆中的元素并不会保持原有的插入顺序
//从而我们需要对应到原先第K个堆中元素
//如果理解这个原理 那么就能明白其实三步交换的顺序是可以互换 
//h,hp,ph之间两两存在映射关系 所以交换顺序的不同对结果并不会产生影响
void heap_swap(int u, int v)
{
    swap(h[u], h[v]);
    swap(hp[u], hp[v]);
    swap(ph[hp[u]], ph[hp[v]]);

}

void down(int u)
{
    int t = u;
    if (u * 2 <= cnt && h[t] > h[u * 2]) t = u * 2;
    if (u * 2 + 1 <= cnt && h[t] > h[u * 2 + 1])  t = u * 2 + 1;
    if (u != t)
    {
        heap_swap(u, t);
        down(t);
    }
}
void up(int u)
{
    if (u / 2 > 0 && h[u] < h[u / 2])
    {
        heap_swap(u, u / 2);
        up(u >> 1);
    }
}

int main()
{
    int n;
    cin >> n;
    int m = 0;      //m用来记录插入的数的个数
                //注意m的意义与cnt是不同的 cnt是记录堆中当前数据的多少
                //对应上文 m即是hp中应该存的值
    while (n--)
    {
        string op;
        int k, x;
        cin >> op;
        if (op == "I")
        {
            cin >> x;
            m++;
            h[++cnt] = x;
            ph[m] = cnt;
            hp[cnt] = m;
            up(cnt);
        }
        else if (op == "PM")    cout << h[1] << endl;
        else if (op == "DM")
        {
            heap_swap(1, cnt);
            cnt--;
            down(1);
        }
        else if (op == "D")
        {
            cin >> k;
            int u = ph[k];                //这里一定要用u=ph[k]保存第k个插入点的下标
            heap_swap(u, cnt);          //因为在此处heap_swap操作后ph[k]的值已经发生 
            cnt--;                    //如果在up,down操作中仍然使用ph[k]作为参数就会发生错误
            up(u);
            down(u);
        }
        else if (op == "C")
        {
            cin >> k >> x;
            h[ph[k]] = x;                 //此处由于未涉及heap_swap操作且下面的up、down操作只会发生一个所以
            down(ph[k]);                //所以可直接传入ph[k]作为参数
            up(ph[k]);
        }

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值