PAT A 1013. Battle Over Cities

PAT A 1013. Battle Over Cities

就是并查集
id[x] = fd(id[x]) 写成 id[x] = fd(x)
导致段错误
ZZ

#include <bits/stdc++.h>
using namespace std;
struct UnionFind
{
    vector<int> id;
    vector<int> sz;
    int n;
    int cnt;
    UnionFind(int n) : n(n)
    {
        cnt = n;
        id.resize(n);
        sz.resize(n);
        for (int i = 0; i < n; i++) {
            id[i] = i;
            sz[i] = 1;
        }
    }
    int fd(int x)
    {
        if (x == id[x]) return x;
        return id[x] = fd(id[x]);
    }
    void unite(int x, int y)
    {
        int fx = fd(x);
        int fy = fd(y);
        if (fx == fy) return;
        if (sz[fx] < sz[fy]) swap(fx, fy);
        id[fy] = fx;
        sz[fx] += sz[fy];
        cnt--;
    }
};
int main()
{
    int n, m, k;
    scanf("%d%d%d", &n, &m, &k);
    vector< pair<int, int> > a;
    for (int i = 0; i < m; i++) {
        int u, v;
        scanf("%d%d", &u, &v);
        u--;
        v--;
        a.push_back(make_pair(u, v));
    }
    while (k--) {
        int x;
        scanf("%d", &x);
        x--;
        UnionFind uf(n);
        for (int i = 0; i < m; i++) {
            if (a[i].first != x && a[i].second != x) {
                uf.unite(a[i].first, a[i].second);
            }
        }
        cout << uf.cnt - 2 << endl;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值