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

      题目大意：给一个无向图的关系，该图由红色边和蓝色边两种构成，问能否恰好用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;
}