SGU 174 (並查集+map)

題目鏈接:SGU 174

分析:每個點爲並查集的一個元素,兩點形成邊時,把兩點歸併成同一個集合。當添加一條新邊,該邊的兩點已在同一集合時,封閉區域形成。


代碼如下:

#include <cstdio>
#include <map>
#include <utility>
using namespace std;

typedef pair<int, int> pii;
map<pii, int> mp;

const int N = 444444;
int p[N], m;

void init() { for(int i = 0; i < 2 * m; i++) p[i] = i; }
int find(int i) { return p[i] == i ? i : (p[i] = find(p[i])); }
void merge(int a, int b) { p[find(b)] = find(a); }

int main()
{
    // freopen("in", "r", stdin);
    int a, b, x, y;
    while(~scanf("%d", &m)) {
        init();
        int ok = 0, ans = 0;
        for(int i = 0; i < m; i++) {
            scanf("%d%d%d%d", &a, &b, &x, &y);
            if(ok) continue;
            if(mp[pii(a, b)] == 0) mp[pii(a, b)] = mp.size();
            if(mp[pii(x, y)] == 0) mp[pii(x, y)] = mp.size();
            if(find(mp[pii(a, b)]) == find(mp[pii(x, y)])) { ans = i + 1; ok = 1; }
            merge(mp[pii(a, b)], mp[pii(x, y)]);
        }
        printf("%d\n", ans);
    }
    return 0;
}




發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章