题目链接: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;
}