数据结构篇——并查集
基本概念
并查集是一种维护集合的数据结构，“并”，“查”，“集” 三个字分别取自 Union（合并），Find（查找），Set（集合）。并查集是若干个不相交集合，能够在 \(O(1)\) 实现两个集合的合并，判断两个元素是否属于同一集合应用，如其求无向图的连通分量个数、实现kruskal算法求最小生成树。
并查集的实现方法就是一个数组：
int pre[N];
其中 pre[i] 表示元素 i 结点的父节点，pre[i] 和 i 结点属于同一集合。例如：pre[1] = 2 就表明元素1的父节点是元素2，元素1和元素2属于同一集合。如果 pre[i]==i 则说明元素 i 是该集合的根结点，对于同一个集合来说，只存在一个根结点。
实现
准备工作
初始化一个pre数组。用于记录了每个节点属于哪个集合；初始时数组内的值与数组的下角标一致。即每个数字都自己一个集合。
void initialize(int n) {
for (int i = 1; i <= n; i++) {
pre[i] = i;
}
}
查找操作
查找的目的是找到集合的 根结点 Big Boss，而不是前驱，因为前驱不一定是根结点，比如图中的5，我们要找到的是3，而不是4。
看这幅图，我们可以知道，当出现 pre[x]==x 时，证明找到了 Big Boss。
这里出现了一个问题，5和4的“顶头上司”都是3，但是5却要多走了一步，如果这个层级更多，那势必会降低算法的效率。
于是就有了在查找过程中的路径压缩
//查找
int Find(int x) {
if (pre[x] == x)return x;
return pre[x] = Find(pre[x]);
////你爸爸的爸爸就是你的爸爸
}
合并操作
初始三个节点，各自为一个集合
如下图，合并3，4操作，此时3和4合并为一个集合。3挂在4上，还是4挂在3上都是一样的，它们都表示3和4是同一个集合，只是集合的Big Boss不一样。
我们继续进行合并4,5操作，通过路径压缩我们获得了右边的形式，此时三个数已经为同一个集合了。
//合并
void Union(int x, int y) {
int fx = Find(x), fy = Find(y);
//如果x,y已经是同一集合了，返回
if (fx == fy) return;
//这里通过深度来确定fx挂fy上，还是fy挂在fx上，实际意义不大，就简单点写了。
pre[fy] = fx;
}
完整代码，这鬼东西好用就好用在很短，真的太好写了
int pre[5001];
void initialize(int n) {
for (int i = 1; i <= n; i++) {
pre[i] = i;
}
}
int Find(int x) {
if (x == pre[x])return pre[x];
else return pre[x] = Find(pre[x]);
}
void Union(int x, int y) {
int fx = Find(x), fy = Find(y);
if (fx == fy) return;
pre[fy] = fx;
}
例题
标准模板题
HDU 1232 ：http://acm.hdu.edu.cn/showproblem.php?pid=1232
基本思想是：N个节点，最少需要 N-1 根线连起来。 把输入的城镇全部进行Union，如果这两个城镇不连通，N--，最后输出 N - 1就是答案。
#include<stdio.h>
int Pre[1001];
//查找
int Find(int x) {
if (Pre[x] == x)return x;
return Pre[x] = Find(Pre[x]);
}
//合并
void Union(int x, int y, int &n) {
int fx = Find(x), fy = Find(y);
if (fx == fy) return;
Pre[fx] = fy;
n--;
}
int main() {
int n, m;
while (~scanf("%d", &n)) {
if (n == 0) break;
for (int i = 0; i <= n; i++) Pre[i] = i;
scanf("%d", &m);
while (m--) {
int c1, c2;
scanf("%d%d", &c1, &c2);
Union(c1, c2, n);
}
printf("%d\n", n - 1);
}
return 0;
}
最小生成树
题目链接：https://www.luogu.org/problem/P3366
题解： https://www.cnblogs.com/czc1999/p/11735791.html
求连通分量个数
并查集
求连通块数量，没找着简单的模板题，代码比较容易理解。
#include <iostream>
using namespace std;
int pre[maxn];
int Find(int x) {
if (pre[x] == x)return x;
return pre[x] = Find(pre[x]);
}
void Union(int x, int y) {
int fx = Find(x), fy = Find(y);
if (fx == fy) return;
pre[fx] = fy;
}
int main(){
int m, n;
cin >> n;
for (int i = 0; i < n; i++) pre[i] = i;
cin >> m;
while (m--) {
int c1, c2;
cin >> c1 >> c2;
Union(c1, c2);//两点之间有路的属于同一个连通块，合并起来
}
int cnt = 0;
for (int i = 0; i < n; i++) {
if (pre[i] == i) cnt++; //有多少个根结点，就有多少个连通块
}
cout << cnt << endl;
return 0;
}
DFS
写了玩的
#include <iostream>
using namespace std;
int m, n;
int cnt = 0;
int maze[500][500];
bool book[500][500];
int dir[4][2] = { {0,-1},{0,1},{-1,0},{1,0} };
void dfs(int x, int y) {
for (int i = 0; i < 4; i++) {
int tox = x + dir[i][0], toy = y + dir[i][1];
if (!book[tox][toy] && maze[tox][toy] == 1 && (tox >= 0 && tox < m && toy >= 0 && toy < n)) {
book[tox][toy] = true;
dfs(tox, toy);
}
}
return;
}
int main() {
cin >> m >> n;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
cin >> maze[i][j];
}
}
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (maze[i][j] && !book[i][j]) {
dfs(i, j);
cnt++;
}
}
}
cout << cnt;
return 0;
}
BFS
#include <iostream>
#include <stdio.h>
#include <queue>
using namespace std;
struct node {
int x;
int y;
node(int x, int y) :x(x), y(y) {};
};
queue<node> q;
int maze[105][105];
bool book[105][105];
int m, n;
int dir[4][2] = { {0,-1},{0,1},{-1,0},{1,0} };
void bfs(int x1, int y1) {
node nn;
nn.x = x1; nn.y = y1;
while (!q.empty()) q.pop();
q.push(nn);
book[x1][y1] = true;
while (q.empty() == false) {
node now = q.front();
q.pop();
for (int i = 0; i < 4; i++) {
int tox = now.x + dir[i][0];
int toy = now.y + dir[i][1];
if (!book[tox][toy] && maze[tox][toy] == 1 && (tox >= 0 && tox < m && toy >= 0 && toy < n)) {
book[tox][toy] = true;
q.push(node(tox, toy));
}
}
}
}
int main(int argc, char** argv) {
cin >> m >> n;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
cin >> maze[i][j];
}
}
int ans = 0;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (maze[i][j] && !book[i][j] ) {
bfs(i, j);
ans++;
}
}
}
cout << ans << endl;
return 0;
}
