HDU 6349 三原色图（最小生成树）

三原色图

Time Limit: 1500/1000 MS (Java/Others)    Memory Limit: 262144/262144 K (Java/Others)
Total Submission(s): 270    Accepted Submission(s): 111

 

Problem Description

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

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

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

 

 

Input

第一行包含一个正整数 T，表示有 T 组测试数据。

接下来依次描述 T 组测试数据。对于每组测试数据：

第一行包含两个整数 n 和 m，表示图的点数和边数。

接下来 m 行，每行包含三个整数 a,b,w 和一个字符 c，表示有一条连接点 a 与点 b 的权值为 w、颜色为 c 的无向边。

保证 1≤T≤100，1≤n,m≤100，1≤a,b≤n，1≤w≤1000，c∈{R,G,B}，这里 R,G,B 分别表示红色、绿色和蓝色。

 

 

Output

对于每组测试数据，先输出一行信息 "Case #x:"（不含引号），其中 x 表示这是第 x 组测试数据，接下来 m 行，每行包含一个整数，第 i 行的整数表示选出恰好 i 条满足条件的边的权值之和的最小值，如果不存在合法方案，输出 −1，行末不要有多余空格。

 

 

Sample Input

1
5 8
1 5 1 R
2 1 2 R
5 4 5 R
4 5 3 G
1 3 3 G
4 3 5 G
5 4 1 B
1 2 2 B

Sample Output

Case #1:
-1
-1
-1
9
10
12
17
22

Source

2018"百度之星"程序设计大赛 - 资格赛

 

还以为这道题特别难来着.......

两遍最小生成树就行了，每一次都要判断颜色，连通的边一定要满足题目的要求，如果满足要求了，就从剩下的边选权值最小的就行了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxm=110;
const int maxn=110;
struct Node
{
    int from;
    int to;
    int val;
    char col;
}edge[maxm];
int n,m;
int tree[maxn];
bool vis[maxn];
void init()
{
    for(int i=0;i<maxn;i++)
    {        
        tree[i]=i;
    }
    memset(vis,false,sizeof(vis));
    return;
}
int find(int x)
{
    return tree[x]==x?x:tree[x]=find(tree[x]);
}
void merge(int a,int b)
{
    int x=find(a);
    int y=find(b);
    if(x!=y)
    {
        tree[x]=y;
    }
    return;
}
bool cmp(Node a,Node b)
{
    return a.val<b.val;
}
int kruskal(char ch,int need)
{
    if(need<n-1) return -1;
    int cnt=0;
    int ans=0;
    for(int i=0;i<m;i++)
    {
        if(edge[i].col!=ch) 
        {
            int u=edge[i].from;
            int v=edge[i].to;
            if(find(u)!=find(v))
            {
                merge(u,v);
                vis[i]=true;
                ans+=edge[i].val;
                cnt++;
                if(cnt==n-1) break;
            }
        }
    }
    int edge_cnt=0;
    if(cnt==need) return ans;
    else if(cnt<n-1) return -1;
    for(int i=0;i<m;i++)
    {
        if(!vis[i])
        {
            ans+=edge[i].val;
            edge_cnt++;
            if(edge_cnt+cnt==need)
            {    
                return ans;
            }
        }
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int test;
    scanf("%d",&test);
    for(int cas=1;cas<=test;cas++)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
        {
            cin>>edge[i].from>>edge[i].to>>edge[i].val>>edge[i].col;
        }
        sort(edge,edge+m,cmp);
        printf("Case #%d:\n",cas);
        for(int i=1;i<=m;i++)
        {
            init();
            //red green
            int ansa=kruskal('B',i);
            init();
            //blue green
            int ansb=kruskal('R',i);
            if(ansa==-1&&ansb==-1)
            {
                puts("-1");
            }
            else if(ansa!=-1&&ansb==-1)
            {
                printf("%d\n",ansa);
            }
            else if(ansa==-1&&ansb!=-1)
            {
                printf("%d\n",ansb);
            }
            else
            {
                printf("%d\n",min(ansa,ansb));
            }
        }
    }
    return 0;
}