您的位置:首页 > 运维架构

hdu5438邻接表+topsprt+并查集

2015-09-16 13:12 357 查看
    1.题目要求:给一个无向图可以是森林,要求去掉叶子节点直到再也没有叶子节点后,求森林中包含奇数的节点的树的权值和

     2.解法:去节点直到没有叶子节点也就说只剩环,那可以topsort一直让度<=1(注意是<=)的节点入列直到不存在,那么剩下的都是环,一次topsort之后用vis标记入过列的点也就是废点。再用并查集维护环的点数的权值和,最后扫一遍父亲是本身并且没标记过而且是奇数的环求和即可。具体细节自己写的时候细细品读,比如topsort和合并都忽略标记过的点。#include<iostream>
#include<fstream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#define ms(x) memset(x,0,sizeof(x))
#define FOR(i,s,t) for(int i=(s);i<(t);i++)
#define CFOR(i) for(;i>0;i--)
#define DFOR(i,s,t) for(int i=(s);i>=(t);i--)
#define PFOR(i,s,t,p) for(int i=(s);i<=(t);i+=p)
#define SD3(i,j,l) scanf("%d%d%d",&i,&j,&l);
#define SLLD(i) scanf("%lld",&i)
#define SD(i) scanf("%d",&i)
#define MEM(a,t) memset(a,t,sizeof(a))
#define PIA system("pause")
#define MAXINT 0xfffffff
#define maxn 10000+10
#define MARK cout<<"MARK"<<endl;
#define BUG cout<<"error!"<<endl;
#define ll long long
using namespace std;
int t,p,m,v[maxn],vis[maxn],d[maxn],a,b,fa[maxn],num[maxn];
vector<int>g[maxn];
ll sum[maxn],ans;
void topsort()
{
MEM(vis,0);
queue<int>q;
for(int i=1;i<=p;i++)
if(d[i]<=1)
{q.push(i);vis[i]=1;}
while(!q.empty())
{
int tmp=q.front();
q.pop();
for(int j=0;j<g[tmp].size();j++)
{
if(vis[g[tmp][j]]==0){///因为是双向边,所以有可能会重复访问
d[g[tmp][j]]--;
if(d[g[tmp][j]]<=1)
{q.push(g[tmp][j]);vis[g[tmp][j]]=1;}
}
}
}
}
int fin(int x)
{
if(fa[x]!=x)
fa[x]=fin(fa[x]);
return fa[x];
}
int main()
{
cin>>t;
while(t--)
{
ans=0;
SD(p),SD(m);
for(int i=1;i<=p;i++)
SD(v[i]);
for(int i=1;i<=p;i++)
{fa[i]=i;sum[i]=v[i],num[i]=1;g[i].clear(),d[i]=0;}
for(int j=0;j<m;j++)
{
scanf("%d%d",&a,&b);
g[a].push_back(b);
g[b].push_back(a);
d[a]++,d[b]++;
}
topsort();
for(int i=1;i<=p;i++)
{
if(!vis[i])
{
for(int j=0;j<g[i].size();j++)
{
if(!vis[g[i][j]])
{
int k1=fin(i),k2=fin(g[i][j]);
if(k1!=k2)
{
sum[k1]+=sum[k2];
num[k1]+=num[k2];
fa[k2]=k1;
}
}
}
}
}
for(int i=1;i<=p;i++)
if(i==fa[i]&&(!vis[i])&&(num[i]&1))
ans+=sum[i];
printf("%lld\n",ans);
}
return 0;
}

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: