码迷,mamicode.com
首页 > Windows程序 > 详细

3624: [Apio2008]免费道路

时间:2018-07-23 23:36:27      阅读:214      评论:0      收藏:0      [点我收藏+]

标签:struct   puts   pre   git   int   两种   for   code   lin   

3624: [Apio2008]免费道路

https://www.lydsy.com/JudgeOnline/problem.php?id=3624

 

题意:

  一张无向图,每种边有两种类型0和1。求一个最小生成树使得有k条0边。

 

分析:

  为了满足有k条0边的限制,先考虑0边哪些必选,如果所有1边都加入后,还有0边可以使得图不连通,那么这些0边必须选。

  把必须选的加上然后再加到k,然后再加1边。中间判一下是否必选的大于k,0边是否大于等于k。

 

代码:

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 typedef long long LL;
 4 
 5 inline int read() { 
 6     int x=0,f=1;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch==-)f=-1;
 7     for(;isdigit(ch);ch=getchar())x=x*10+ch-0;return x*f;
 8 }
 9 
10 const int N = 20010;
11 const int M = 100010;
12 
13 struct Edge{
14     int u,v;
15     Edge() {}
16     Edge(int a,int b) {u = a, v = b;}
17 }e[M];
18 int fa[N], q[M];
19 bool vis[M];
20 
21 int find(int x) {
22     return x == fa[x] ? x : fa[x] = find(fa[x]);
23 }
24 
25 int main () {
26     
27     int n = read(), m = read(), k = read();
28     
29     int B = 0, W = m + 1;
30     for (int i=1; i<=m; ++i) {
31         int u = read(), v = read(), c = read();
32         if (c == 1) e[++B] = Edge(u,v);
33         else e[--W] = Edge(u,v);
34     }
35     
36     for (int i=1; i<=n; ++i) fa[i] = i;
37     
38     int tot = 0;
39     for (int i=1; i<=m; ++i) {
40         int u = find(e[i].u), v = find(e[i].v);
41         if (u == v) continue;
42         fa[u] = v;
43         if (i > B) vis[i] = true, q[++tot] = i;
44     }
45     
46     if (tot > k) {
47         puts("no solution");
48         return 0;
49     }
50     
51     for (int i=1; i<=n; ++i) fa[i] = i;
52     
53     for (int i=1; i<=tot; ++i) 
54         fa[find(e[q[i]].u)] = find(e[q[i]].v);
55     
56     for (int i=W; i<=m; ++i) {
57         if (tot == k) break;
58         int u = find(e[i].u), v = find(e[i].v);
59         if (u == v) continue;
60         fa[u] = v;
61         tot ++;
62         vis[i] = true;
63     }
64     
65     if (tot < k) {
66         puts("no solution");
67         return 0;
68     }
69     
70     for (int i=1; i<=B; ++i) {
71         int u = find(e[i].u), v = find(e[i].v);
72         if (u == v) continue;
73         fa[u] = v;
74         vis[i] = true;
75     }
76     
77     for (int i=1; i<=m; ++i) 
78         if (vis[i]) printf("%d %d %d\n", e[i].u, e[i].v, i<=B?1:0);
79 
80     return 0;
81 }

 

3624: [Apio2008]免费道路

标签:struct   puts   pre   git   int   两种   for   code   lin   

原文地址:https://www.cnblogs.com/mjtcn/p/9357005.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!