欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

并查集部分

程序员文章站 2022-03-16 08:13:12
功能①:合并两个集合void merge(int a, int b) {f[Find(a)] = Find(b);}②:查询某个元素的祖宗节点int Find(int x) {return x == f[x] ? x : f[x] = Find(f[x]);}扩展① 记录每个集合的大小(绑定到根节点)int pa = Find(a);int pb = Find(b);if(){ s[pb] += s[pa]}②每个点到根节点的距离(绑定到每一个元素身上)in...

功能

①:合并两个集合

void merge(int a, int b) {
	f[Find(a)] = Find(b);
}

②:查询某个元素的祖宗节点

int Find(int x) {
	return x == f[x] ? x : f[x] = Find(f[x]);
}

扩展

① 记录每个集合的大小(绑定到根节点)

int pa = Find(a);
int pb = Find(b);
if(){
    s[pb] += s[pa]
}

②每个点到根节点的距离(绑定到每一个元素身上)

int n, p[N], s[N], d[N];
//p[x] x的父亲节点  d[x] x到父节点的距离   s[x] 集合的大小
int Find(int x) { //距离查找
	if (p[x] != x) {
		int root = Find(p[x]);
		d[x] += d[p[x]];
		p[x] = root;
	}
	return p[x];
}

int pa = Find(a);
int pb = Find(b);
if () { //距离更新
	p[pa] = pb;
	d[pa] = s[pb];
	s[pb] += s[pa];
}

AcWing1250. 格子游戏

Alice和Bob玩了一个古老的游戏:首先画一个 n×n 的点阵(下图 n=3)。

接着,他们两个轮流在相邻的点之间画上红边和蓝边:

并查集部分

直到围成一个封闭的圈(面积不必为 1)为止,“封圈”的那个人就是赢家。因为棋盘实在是太大了,他们的游戏实在是太长了!

他们甚至在游戏中都不知道谁赢得了游戏。

于是请你写一个程序,帮助他们计算他们是否结束了游戏?

输入格式

输入数据第一行为两个整数 n 和 m。n表示点阵的大小,m 表示一共画了 m 条线。

以后 m 行,每行首先有两个数字 (x,y),代表了画线的起点坐标,接着用空格隔开一个字符,假如字符是 D,则是向下连一条边,如果是 R 就是向右连一条边。

输入数据不会有重复的边且保证正确。

输出格式

输出一行:在第几步的时候结束。

假如 m 步之后也没有结束,则输出一行“draw”。

数据范围

1 ≤ n ≤ 200 1≤n≤200 1n200
1 ≤ m ≤ 24000 1≤m≤24000 1m24000

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<string>
#include<cstring>
#include<map>
#include<vector>
#include<set>
#include<stack>
#include<algorithm>
#include<vector>
#include<utility>
#include<deque>
#define INF 0x3f3f3f3f
#define mod 1000000007
#define endl '\n'
#define eps 1e-6
inline int gcd(int a, int b) { return b ? gcd(b, a % b) : a; }
inline int lowbit(int x) { return x & -x; }


using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 50000;
int f[N];
int n,m;
int Find(int x) {
	return x == f[x] ? x : f[x] = Find(f[x]);
}

int get(int x, int y) {
	return x * n + y;
}

void merge(int a, int b) {
	f[Find(a)] = Find(b);
}

int main() {
	cin >> n >> m;

	for (int i = 0; i < N;++i)f[i] = i;
	int res = 0;
	for (int i = 1; i <= m;++i) {
		int x, y;
		char d;
		cin >> x >> y >> d;
		int a = get(x, y);
		int b = 0;
		if (d == 'D')b = get(x + 1, y);
		else b = get(x, y + 1);

		if (Find(a) != Find(b))merge(a, b);
		else {
			res = i;
			break;
		}
	}

	if (!res)puts("draw");
	else cout << res << endl;
	return 0;
}

AcWing1252. 搭配购买

思路

先分组 再做01背包


Joe觉得云朵很美,决定去山上的商店买一些云朵。

商店里有 n 朵云,云朵被编号为 1,2,…,n,并且每朵云都有一个价值。

但是商店老板跟他说,一些云朵要搭配来买才好,所以买一朵云则与这朵云有搭配的云都要买。

但是Joe的钱有限,所以他希望买的价值越多越好。

输入格式

第 1 行包含三个整数 n,m,w表示有 n 朵云,m 个搭配,Joe有 w 的钱。

第 2∼n+1行,每行两个整数 c i c_{i} ci d i d_{i} di 表示 i 朵云的价钱和价值。

第 n+2∼n+1+m行,每行两个整数 u i u_{i} ui v i v_{i} vi表示买 u i u_{i} ui 就必须买 v i v_{i} vi,同理,如果买 v i v_{i} vi 就必须买 u i u_{i} ui

输出格式

一行,表示可以获得的最大价值。

数据范围

1 ≤ n ≤ 10000 1≤n≤10000 1n10000
0 ≤ m ≤ 5000 0≤m≤5000 0m5000
1 ≤ w ≤ 10000 1≤w≤10000 1w10000
1 ≤ c i ≤ 5000 1≤ci≤5000 1ci5000
1 ≤ d i ≤ 100 1≤di≤100 1di100
1 ≤ u i , v i ≤ n 1≤ui,vi≤n 1ui,vin

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<string>
#include<cstring>
#include<map>
#include<vector>
#include<set>
#include<stack>
#include<algorithm>
#include<vector>
#include<utility>
#include<deque>
#define INF 0x3f3f3f3f
#define mod 1000000007
#define endl '\n'
#define eps 1e-6
inline int gcd(int a, int b) { return b ? gcd(b, a % b) : a; }
inline int lowbit(int x) { return x & -x; }


using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 10010;
int f[N], p[N], v[N], w[N];
int n, m, val;

int Find(int x) {
	return x == p[x] ? x : p[x] = Find(p[x]);
}


int main() {
	cin >> n >> m >> val;
	for (int i = 1;i <= n;++i)p[i] = i;
	for (int i = 1; i <= n;++i) {
		cin >> v[i] >> w[i];
	}

	while (m--) {
		int a, b;
		cin >> a >> b;
		int pa = Find(a);
		int pb = Find(b);

		if (pa != pb) {
			v[pb] += v[pa];
			w[pb] += w[pa];
			p[pa] = p[pb];
		}
	}
	for (int i = 1;i <= n;++i) {
		if (p[i] == i) {
			for (int j = val;j >= v[i];--j)
				f[j] = max(f[j], f[j - v[i]] + w[i]);
		}
	}
	cout << f[val] << endl;

	return 0;
}

AcWing237. 程序自动分析

思路

先把相等的合并 再判断不相等是否与已知条件矛盾


在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。

考虑一个约束满足问题的简化版本:假设 x 1 , x 2 , x 3 x_{1},x_{2},x_{3} x1,x2,x3,…代表程序中出现的变量,给定n个形如 x i = x j x_{i}=x_{j} xi=xj x i ≠ x j x_{i}≠x_{j} xi=xj的变量相等/不等的约束条件,请判定是否可以分别为每一个变量赋予恰当的值,使得上述所有约束条件同时被满足。

例如,一个问题中的约束条件为: x 1 = x 2 , x 2 = x 3 , x 3 = x 4 , x 1 ≠ x 4 x_{1}=x_{2},x_{2}=x_{3},x_{3}=x_{4},x_{1}≠x_{4} x1=x2x2=x3x3=x4x1=x4,这些约束条件显然是不可能同时被满足的,因此这个问题应判定为不可被满足。

现在给出一些约束满足问题,请分别对它们进行判定。

输入格式

输入文件的第1行包含1个正整数t,表示需要判定的问题个数,注意这些问题之间是相互独立的。

对于每个问题,包含若干行:

第1行包含1个正整数n,表示该问题中需要被满足的约束条件个数。

接下来n行,每行包括3个整数i,j,e,描述1个相等/不等的约束条件,相邻整数之间用单个空格隔开。若e=1,则该约束条件为 x i = x j x_{i}=x_{j} xi=xj;若e=0,则该约束条件为 x i ≠ x j x_{i}≠x_{j} xi=xj

输出格式

输出文件包括t行。

输出文件的第k行输出一个字符串“YES”或者“NO”(不包含引号,字母全部大写),“YES”表示输入中的第k个问题判定为可以被满足,“NO”表示不可被满足。

数据范围

1 ≤ n ≤ 1000000 1≤n≤1000000 1n1000000
1 ≤ i , j ≤ 1000000000 1≤i,j≤1000000000 1i,j1000000000

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<string>
#include<cstring>
#include<map>
#include<vector>
#include<set>
#include<stack>
#include<algorithm>
#include<vector>
#include<utility>
#include<deque>
#include<unordered_map>
#define INF 0x3f3f3f3f
#define mod 1000000007
#define endl '\n'
#define eps 1e-6
inline int gcd(int a, int b) { return b ? gcd(b, a % b) : a; }
inline int lowbit(int x) { return x & -x; }


using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 2000010;
int f[N], n, m;
unordered_map<int, int>S;


struct Query {
	int x, y, e;
}query[N];
int Find(int x) {
	return x == f[x] ? x : f[x] = Find(f[x]);
}

int get(int x) {
	if (S.count(x) == 0)S[x] = ++n;
	return S[x];
}
int main() {
	int t;cin >> t;
	while (t--) {
		S.clear();
		n = 0;
		
		scanf("%d", &m);
		for (int i = 0; i < m;++i) {
			int x, y, e;
			scanf("%d%d%d", &x, &y, &e);
			query[i] = { get(x),get(y),e };
		}
		
		for(int i = 1; i <= n;++i)f[i] = i;
		bool has = false;
		for (int i = 0; i < m;++i) {
			if (query[i].e == 1)f[Find(query[i].x)] = Find(query[i].y);
		}

		for (int i = 0;i < m;++i) {
			if (query[i].e == 0 && Find(query[i].x) == Find(query[i].y)) {
				has = true;
				break;
			}
		}

		if (has)puts("NO");
		else puts("YES");
	}

	return 0;
}

AcWing238. 银河英雄传说

有一个划分为N列的星际战场,各列依次编号为1,2,…,N。

有N艘战舰,也依次编号为1,2,…,N,其中第i号战舰处于第i列。

有T条指令,每条指令格式为以下两种之一:

1、M i j,表示让第i号战舰所在列的全部战舰保持原有顺序,接在第j号战舰所在列的尾部。

2、C i j,表示询问第i号战舰与第j号战舰当前是否处于同一列中,如果在同一列中,它们之间间隔了多少艘战舰。

现在需要你编写一个程序,处理一系列的指令。

输入格式

第一行包含整数T,表示共有T条指令。

接下来T行,每行一个指令,指令有两种形式:M i j或C i j。

其中M和C为大写字母表示指令类型,i和j为整数,表示指令涉及的战舰编号。

输出格式

你的程序应当依次对输入的每一条指令进行分析和处理:

如果是M i j形式,则表示舰队排列发生了变化,你的程序要注意到这一点,但是不要输出任何信息;

如果是C i j形式,你的程序要输出一行,仅包含一个整数,表示在同一列上,第i号战舰与第j号战舰之间布置的战舰数目,如果第i号战舰与第j号战舰当前不在同一列上,则输出-1。

数据范围

N ≤ 30000 , T ≤ 500000 N≤30000,T≤500000 N30000,T500000

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<string>
#include<cstring>
#include<map>
#include<vector>
#include<set>
#include<stack>
#include<algorithm>
#include<vector>
#include<utility>
#include<deque>
#include<unordered_map>
#define INF 0x3f3f3f3f
#define mod 1000000007
#define endl '\n'
#define eps 1e-6
inline int gcd(int a, int b) { return b ? gcd(b, a % b) : a; }
inline int lowbit(int x) { return x & -x; }


using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 30030;
int n, p[N], s[N], d[N];

int Find(int x) {
	if (p[x] != x) {
		int root = Find(p[x]);
		d[x] += d[p[x]];
		p[x] = root;
	}
	return p[x];
}

int main() {
		for (int i = 1;i < N;++i) {
			p[i] = i;
			s[i] = 1;
		}
		
		int n;scanf("%d", &n);
		while (n--) {
			char op[2];int a, b;
			scanf("%s%d%d", op, &a, &b);
			int pa = Find(a);
			int pb = Find(b);
			if (op[0] == 'M') {
				p[pa] = pb;
				d[pa] = s[pb];
				s[pb] += s[pa];
			}

			else {
				if (pa == pb)printf("%d\n", max(0, abs(d[a] - d[b]) - 1));
				else printf("-1\n");
			}
		}

	return 0;
}

AcWing239. 奇偶游戏

思路

s i = a 1 + a 2 + ⋯ + a i s_{i} = a_{1} + a_{2} +\dots +a_{i} si=a1+a2++ai 前i个数中1的个数

s [ l − r ] s[l-r] s[lr]中有奇数个1

   ⟺    \iff s r − s l − 1 s_{r} - s_{l-1} srsl1为奇数

   ⟺    \iff s r s_{r} sr s l − 1 s_{l-1} sl1奇偶性不同

偶数个1    ⟺    \iff s r s_{r} sr s l − 1 s_{l - 1} sl1奇偶性相同


小A和小B在玩一个游戏。

首先,小A写了一个由0和1组成的序列S,长度为N。

然后,小B向小A提出了M个问题。

在每个问题中,小B指定两个数 l 和 r,小A回答 S[l~r] 中有奇数个1还是偶数个1。

机智的小B发现小A有可能在撒谎。

例如,小A曾经回答过 S[1~3] 中有奇数个1, S[4~6] 中有偶数个1,现在又回答 S[1~6] 中有偶数个1,显然这是自相矛盾的。

请你帮助小B检查这M个答案,并指出在至少多少个回答之后可以确定小A一定在撒谎。

即求出一个最小的k,使得01序列S满足第1 ~ k个回答,但不满足第1~k+1个回答。

输入格式

第一行包含一个整数N,表示01序列长度。

第二行包含一个整数M,表示问题数量。

接下来M行,每行包含一组问答:两个整数l和r,以及回答“even”或“odd”,用以描述S[l~r] 中有偶数个1还是奇数个1。

输出格式

输出一个整数k,表示01序列满足第1k个回答,但不满足第1k+1个回答,如果01序列满足所有回答,则输出问题总数量。

数据范围

N ≤ 109 , M ≤ 10000 N≤109,M≤10000 N109,M10000

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<string>
#include<cstring>
#include<map>
#include<vector>
#include<set>
#include<stack>
#include<algorithm>
#include<vector>
#include<utility>
#include<deque>
#include<unordered_map>
#define INF 0x3f3f3f3f
#define mod 1000000007
#define endl '\n'
#define eps 1e-6
inline int gcd(int a, int b) { return b ? gcd(b, a % b) : a; }
inline int lowbit(int x) { return x & -x; }


using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 20050;

unordered_map<int, int>s;
int n, m;
int p[N], d[N];

int get(int x) {
	if (s.count(x) == 0)s[x] = ++n;
	return s[x];
}

int Find(int x) {
	if (x != p[x]) {
		int root = Find(p[x]);
		d[x] ^= d[p[x]];
		p[x] = root;
	}
	return p[x];
}
int main() {
	cin >> n >> m;
	n = 0;
	for (int i = 1;i < N;++i)p[i] = i;
	int res = m;

	for (int i = 1;i <= m;++i) {
		int a, b;
		string op;
		cin >> a >> b >> op;
		a = get(a - 1), b = get(b);

		int t = 0; //偶数
		if (op == "odd")t = 1; //奇数
		int pa = Find(a), pb = Find(b);

		if (pa == pb) {
			if ((d[a] ^ d[b]) != t) {
				res = i - 1;
				break;
			}
		}
		else {
			p[pa] = pb;
			d[pa] = d[a] ^ d[b] ^ t;
		}
	}
	cout << res << endl;
}

本文地址:https://blog.csdn.net/zzq0523/article/details/109251782