1416 两点 并查集

http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1416

对于每一个坐标(x, y)

要使得它变成一个整数，一个方法就是x * max(n, m) + y

这样就不会相同了。可以幻想一下吧。

这样坐标的大小只会去到n * m + y这个级别，不会很大。

然后并查集那里，每一次都走下面和右边的点，看看能否合并，能的话就合并。然后

合并到相同的爸爸，就是有了。

AA

AA

其中(4, 4)这个点被(1, 2)合并了一次，爸爸是（1， 1）

然后又被(2, 1)合并一次，爸爸是相同了。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <assert.h>
#define IOS ios::sync_with_stdio(false)
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;

#include <iostream>
#include <sstream>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <string>
#include <bitset>
const int maxn = 50 + 20;
int fa[maxn * maxn * maxn];
char str[maxn][maxn];
int tofind(int u) {
if (fa[u] == u) return fa[u];
else return fa[u] = tofind(fa[u]);
}
bool tomerge(int x, int y) {
x = tofind(x);
y = tofind(y);
if (x == y) return false;
fa[y] = x;
return true;
}
int n, m;
int calc(int x, int y) {
return max(n, m) * x + y;
}
void work() {
cin >> n >> m;
for (int i = 1; i <= n; ++i) cin >> str[i] + 1;
for (int i = 1; i <= maxn * maxn * maxn - 1; ++i) fa[i] = i;
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= m; ++j) {
if (j + 1 <= m && str[i][j] == str[i][j + 1]) {
if (!tomerge(calc(i, j), calc(i, j + 1))) {
//                    cout << i << endl;
//                    cout << j << endl;
cout << "YES" << endl;
return;
}
}
if (i + 1 <= n && str[i][j] == str[i + 1][j]) {
if (!tomerge(calc(i, j), calc(i + 1, j))) {
cout << "YES" << endl;
return;
}
}
}
}
cout << "NO" << endl;
}
int main() {
#ifdef local
freopen("data.txt", "r", stdin);
//    freopen("data.txt", "w", stdout);
#endif
work();
return 0;
}

View Code