HDU-#4263 Red/Blue Spanning Tree（Kruskal）

来源：互联网发布：linux tcpdump 用法编辑：程序博客网时间：2024/05/30 22:54

题目大意：给一个无向图的关系，该图由红色边和蓝色边两种构成，问能否恰好用k条蓝色边构成一颗MST？

解题思路：要判断是否存在，额可以转化为：先求一次构成的MST最少需要多少条蓝色边a，再求一次最多需要多少条的蓝色边b，然后判断k属不属于该[a,b]区间即可。又因为是统计蓝色边的数量，因此输入的时候将红色边的权值置为0，蓝色边的权值置为1。

题目来源：http://acm.hdu.edu.cn/showproblem.php?pid=4263

code：

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int MAXN = 1000+10;int p[MAXN];int n,m,k,ans_a,ans_b;char ch[3];struct edge{    int u,v,w;}e[MAXN*MAXN];int cmp(const edge a,const edge b){return a.w<b.w;} //最小生成树的间接排序函数int recmp(const edge a,const edge b){return a.w>b.w;} //最大生成树的间接排序函数int find(int x){return x==p[x]?x:p[x]=find(p[x]);} //并查集find函数int Kruskal(bool flag){ //Kruskal算法    int ans=0;    for(int i=1;i<=n;i++) p[i]=i;    if(flag) sort(e+1,e+1+m,cmp); //蓝色的最小生成树排序    if(!flag) sort(e+1,e+1+m,recmp); //蓝色的最大生成树排序    for(int i=1;i<=m;i++){        int x=find(e[i].u);        int y=find(e[i].v);        if(x!=y){            p[x]=y;            ans+=e[i].w;        }    }    return ans;}int main(){    while(scanf("%d%d%d",&n,&m,&k)!=EOF && (n||m||k)){        for(int i=1;i<=m;i++){            scanf("%s%d%d",&ch,&e[i].u,&e[i].v);            if(ch[0]=='R') e[i].w=0; //为红色全置为0            else e[i].w=1; //蓝色置为1，这样的计数就是蓝色边的数量        }        ans_a=Kruskal(true); //计算最少用蓝色边的个数        ans_b=Kruskal(false); //计算最多用蓝色边的个数        if(k<=ans_b&& k>=ans_a) printf("1\n");        else printf("0\n");    }    return 0;}

0 0