湖南多校对抗赛(2015.05.03)Problem B: War

并查集。从后往前加边。

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
const int maxn = 1000000 + 10;
int father[maxn], u[maxn], v[maxn], Q, q[maxn], ans[maxn], ff[maxn];
int find(int x)
{
    if (x != father[x]) father[x] = find(father[x]);
    return father[x];
}
int main()
{
    int n, m, i;
    while (~scanf("%d%d", &n, &m))
    {
        memset(ff, 0, sizeof(ff));
        for (i = 0; i <= n; i++) father[i] = i;
        for (i = 1; i <= m; i++) scanf("%d%d", &u[i], &v[i]);
        scanf("%d", &Q);
        for (i = 1; i <= Q; i++){ scanf("%d", &q[i]); ff[q[i]] = 1; }
        int tot = n;//没有边的时候有n个集合
        for (i = 1; i <= m; i++)
        {
            if (ff[i] == 0)
            {
                int uf, vf;
                uf = find(u[i]);
                vf = find(v[i]);
                if (vf != uf)
                {
                    father[vf] = uf;
                    tot--; //加入一条边,集合少一个
                }
            }
        }
        ans[1] = tot;
        int j = 2;
        for (i = Q; i > 1; i--)
        {
            int uf, vf;
            uf = find(u[q[i]]);
            vf = find(v[q[i]]);
            if (vf != uf)
            {
                father[vf] = uf;
                tot--; //加入一条边,集合少一个
            }
            ans[j] = tot; j++;
        }
        for (i = Q; i >= 1; i--)
        {
            if (i > 1) printf("%d ", ans[i]);
            else printf("%d\n", ans[i]);
        }
    }
    return 0;
}

 

posted @ 2015-05-03 16:17  Fighting_Heart  阅读(165)  评论(0编辑  收藏  举报