【2018百度之星资格赛】1006 三原色图

题意

度度熊有一张 nnn 个点 mmm 条边的无向图，所有点按照 1,2,⋯,n1,2,cdots,n1,2,⋯,n 标号，每条边有一个正整数权值以及一种色光三原色红、绿、蓝之一的颜色。

现在度度熊想选出恰好 kkk 条边，满足只用这 kkk 条边之中的红色边和绿色边就能使 nnn 个点之间两两连通，或者只用这 kkk 条边之中的蓝色边和绿色边就能使 nnn 个点之间两两连通，这里两个点连通是指从一个点出发沿着边可以走到另一个点。

对于每个 k=1,2,⋯,mk=1,2,cdots,mk=1,2,⋯,m，你都需要帮度度熊计算选出恰好 kkk 条满足条件的边的权值之和的最小值。

分析

好吧我真的是太太太太无聊了，因为发现写正在比赛的题的博客会让人气增加好多哈哈哈哈哈那我就再贴一篇

这个题就是拿并查集和最小生成树瞎搞一下。

分别求出两种情况的最小生成树（情况一只用红色和绿色的边，第二种情况只用绿色和蓝色的边），然后记录一下对于每种情况用了哪些边，当k>n-1的时候按照边权从小打到找没有被用过的边然后加上就可以惹

 下面是代码，写的很丑

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>

using namespace std;
const int INF=2147000000;
const int maxn=1000;
struct Edge{
    int from,to,w;
    char c;
    bool operator<(const Edge& rhs)const{
        return w<rhs.w;
    }
}edges[maxn];
int T,n,m;
int p[maxn];
int find(int x){
    return p[x]==x?x:p[x]=find(p[x]);
}
int vis[3][maxn];
int solve(int s){
    memset(vis[s],0,sizeof(vis[s]));
    int res=0;
    for(int i=1;i<=n;i++)p[i]=i;
    for(int i=1;i<=m;i++){
        if(s==1&&edges[i].c=='B')
            continue;
        if(s==2&&edges[i].c=='R')
            continue;
        int x=find(edges[i].from),y=find(edges[i].to);
        if(x!=y){
            p[x]=y;
            res+=edges[i].w;
            vis[s][i]=1;
        }
    }
    int foot=find(1);
    for(int i=2;i<=n;i++){
        if(foot!=find(i))
            return -1;
    }
    return res;
}
int ans1[maxn],ans2[maxn];
int main(){
    scanf("%d",&T);
    for(int t=1;t<=T;t++){
        memset(ans1,-1,sizeof(ans1));
        memset(ans2,-1,sizeof(ans2));
        printf("Case #%d:
",t);
        scanf("%d%d",&n,&m);
        for(int i=1;i<=m;i++){
            scanf("%d%d%d %c",&edges[i].from,&edges[i].to,&edges[i].w,&edges[i].c);
        }
        sort(edges+1,edges+1+m);
        for(int i=1;i<n-1;i++)
            printf("-1
");
        int flag=0;
        int res1=solve(1);
        int res2=solve(2);
        if(res1==-1&&res2==-1){
            for(int i=n-1;i<=m;i++){
                printf("-1
");
            }
            continue;
        }
        //printf("%d %d
",res1,res2);
        int num=n-1;
        if(res1==-1){
            ans2[num]=res2;
            ans1[num]=INF;

        }
        else if(res2==-1){
            ans1[num]=res1;
            ans2[num]=INF;
        }
        else{
            ans1[num]=res1;
            ans2[num]=res2;
        }
        if(res1!=-1){
            for(int i=1;i<=m;i++){
                if(!vis[1][i]){
                    num++;
                    ans1[num]=ans1[num-1]+edges[i].w;
                }
            }
        }
        else
            for(int i=n;i<=m;i++)ans1[i]=INF;

        num=n-1;
        if(res2!=-1){
            for(int i=1;i<=m;i++){
                if(!vis[2][i]){
                    num++;
                    ans2[num]=ans2[num-1]+edges[i].w;
                }
            }
        }
        else
            for(int i=n;i<=m;i++)ans2[i]=INF;
        for(int i=n-1;i<=m;i++){
                printf("%d
",min(ans1[i],ans2[i]));
        }
    }
return 0;
}