三原色图
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
还以为这道题特别难来着.......
两遍最小生成树就行了,每一次都要判断颜色,连通的边一定要满足题目的要求,如果满足要求了,就从剩下的边选权值最小的就行了
#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;
}