HDU5438 Ponds 拓扑+并查集

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5438

题目大意：n个顶点m条边的无向图，每个顶点有一个权值，现在要删除所有的叶子节点，重复此操作直至形成的“森林”中不存在叶子节点，然后找出森林中顶点个数为奇数的树的权值的和。

分析：删除操作可以模拟拓扑过程，首先我们让度数为1的顶点进队，然后每出队一个顶点，就让该顶点相邻顶点的度数减1，碰到度数为1的顶点就入队，重复操作直至队列为空，这样我们就删除了所有的叶子节点，此时图可能变为了森林，然后只需在剩下的森林中找出符合条件的树即可，可以用并查集纪录每个树中顶点的个数和树的权值。要注意的是图不能用邻接矩阵来存储（交了一次MLE），这里用的是前向星。

实现代码如下：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
typedef long long ll;
const int maxn=10005;

int u[maxn],v[maxn]; //纪录每条边
int w[maxn],ind[maxn]; //w纪录每个点的权值，ind纪录每个点的度数
bool vis[maxn]; //标记remove操作后还剩下的点
//前向星存储图
struct EDGE
{
    int to,nex;
}edge[maxn*20];
int head[maxn],num;
void add(int u,int v)
{
    edge[num].to=v;
    edge[num].nex=head[u];
    head[u]=num++;
}
//并查集
int par[maxn],ran[maxn];
long long sum[maxn];
void Init(int n)
{
    for(int i=1;i<=n;i++)
    {
        par[i]=i;
        ran[i]=1;
        sum[i]=w[i];
    }
}
int Find(int x)
{
    if(par[x]!=x) return par[x]=Find(par[x]);
    return x;
}
void Union(int a,int b)
{
    int x=Find(a);
    int y=Find(b);
    if(x==y) return ;
    par[y]=x;
    ran[x]+=ran[y];
    sum[x]+=sum[y];
}

void Top_sort(int n)
{
    queue <int> que;
    for(int i=1;i<=n;i++)
      if(ind[i]<=1)
      {
          vis[i]=false;
          que.push(i);
      }
    while(!que.empty())
    {
        int cnt=que.front();
        que.pop();
        for(int i=head[cnt];i!=-1;i=edge[i].nex)
        {
            int x=edge[i].to;
            if(vis[x])
            {
                ind[x]--;
                if(ind[x]==1)
                {
                    vis[x]=false;
                    que.push(x);
                }
            }
        }
    }
}
int main()
{
    int t,n,m;
    scanf("%d",&t);
    while(t--)
    {
        memset(vis,true,sizeof(vis));
        memset(head,-1,sizeof(head));
        memset(ind,0,sizeof(ind));
        num=0;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
          scanf("%d",&w[i]);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&u[i],&v[i]);
            add(u[i],v[i]);
            add(v[i],u[i]);
            ind[ u[i] ]++,ind[ v[i] ]++;
        }
        Top_sort(n);
        Init(n);
        for(int i=0;i<m;i++)
        {
            if(vis[ u[i] ]&&vis[ v[i] ])
              Union(u[i],v[i]);
        }
        long long ans=0;
        for(int i=1;i<=n;i++)
          if(i==par[i]&&vis[i]&&(ran[i]&1))
            ans+=sum[i];
        printf("%I64d\n",ans);
    }
    return 0;
}