hdu6349

题目链接：hdu6349

思路：求两个最小生成树，一个只走红色（R），绿色的路(G)，然后按权值从小到大的顺序加入边，即可生成一个答案的数组，同理生成另一个数组，每次输出答案是比较即可。

可用Kruskal求最小生成树。

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<stack>
#include<cstdio>
#include<map>
#include<set>
#include<string>
#include<queue>
using namespace std;
#define inf 0x3f3f3f3f
const int maxn=100+50;
int cnt=0;
int fa[maxn];//使用并查集求kruskal
struct edg{
    int u,v,w;
    char color;
}s[maxn];
int flag[maxn][2];//标记某条边是否已被使用
int ans_2[maxn][2];//两种情况的最小值
void add(int u,int v,int w,char ch){
    s[cnt].u=u;
    s[cnt].v=v;
    s[cnt].w=w;
    s[cnt].color=ch;
    cnt++;
}
int find(int i){
    if(fa[i]==-1)
    return i;
    else
    return find(fa[i]);
}
bool cmp(const edg & a,const edg & b){
    return a.w<b.w;
}
void kruskal(int n,int m){
    memset(fa,-1,sizeof(fa));
    memset(flag,0,sizeof(flag));
    memset(ans_2,-1,sizeof(ans_2));
    int ans1=0,ans2=0;
    int num1=0,num2=0;
    int flag1=0;
    for(int i=0;i<m;i++){
        if(s[i].color=='G'||s[i].color=='R'){
        int a=find(s[i].u);
        int b=find(s[i].v);
        if(a!=b){
            ans1+=s[i].w;
            fa[a]=b;
            num1++;
            flag[i][0]=1;
        }
        if(num1==n-1){
            flag1=1;
            break;    
        }
        }    
    }
    if(flag1==1){
        ans_2[n-2][0]=ans1;
        for(int i=n-1;i<m;i++){
            for(int j=0;j<m;j++){
                if(flag[j][0]==0){
                    ans_2[i][0]=ans_2[i-1][0]+s[j].w;
                    flag[j][0]=1;
                    break;
                }    
            }
        }
    }
    flag1=0;
    memset(fa,-1,sizeof(fa));
    for(int i=0;i<m;i++){
        if(s[i].color=='G'||s[i].color=='B'){
        int a=find(s[i].u);
        int b=find(s[i].v);
        if(a!=b){
            ans2+=s[i].w;
            fa[a]=b;
            flag[i][1]=1;
            num2++;
        }
        if(num2==n-1){
            flag1=1;
            break;
        }
        }    
    }
    if(flag1==1){
        ans_2[n-2][1]=ans2;
        for(int i=n-1;i<m;i++){
            for(int j=0;j<m;j++){
                if(flag[j][1]==0)
                {
                    ans_2[i][1]=ans_2[i-1][1]+s[j].w;
                    flag[j][1]=1;
                    break;
                }
            }
        }
    }
}
int main(){
    int t,a,b,w;
    char ch;
    cin>>t;
    int n,m,count=1;
    while(t--){
        cnt=0;
        cin>>n>>m;
        for(int i=0;i<m;i++){
            cin>>a>>b>>w>>ch;
            add(a,b,w,ch);
        }
        sort(s,s+m,cmp);
        kruskal(n,m);
        printf("Case #%d:\n",count++);
        for(int i=0;i<m;i++){
            if(ans_2[i][0]==-1&&ans_2[i][1]==-1){
                cout<<-1<<endl;
            }
            else if(ans_2[i][0]==-1&&ans_2[i][1]!=-1){
                cout<<ans_2[i][1]<<endl;
            }
            else if(ans_2[i][1]==-1&&ans_2[i][0]!=-1){
                cout<<ans_2[i][0]<<endl;
            }
            else
            cout<<min(ans_2[i][0],ans_2[i][1])<<endl;
        }
    }
    return 0;
}