Bad Cowtractors POJ 2377(最小生成树)

原题

题目链接

题目分析

题目看似要求最大生成树,其实不然,输入时把所有边乘个-1,在求出来的最小生成树其实就是最大生成树了,只需要注意一下有可能无法生成连接所有点的树,这时候要输出-1.

代码

 1 #include <iostream>
 2 #include <algorithm>
 3 #include <utility>
 4 #include <cstdio>
 5 #include <cmath>
 6 #include <cstring>
 7 #include <string>
 8 #include <vector>
 9 #include <stack>
10 #include <queue>
11 #include <map>
12 #include <set>
13 
14 using namespace std;
15 typedef long long LL;
16 const int INF_INT=0x3f3f3f3f;
17 const LL INF_LL=0x3f3f3f3f3f3f3f3f;
18 
19 typedef pair<int,int> P;
20 int n,m;
21 struct edge{int to,cost;};
22 vector<edge> es[2000];
23 bool used[2000];
24 
25 int prim()
26 {
27     int ans=0;
28     priority_queue<P,vector<P>,greater<P> > que;
29     for(int i=0;i<es[1].size();i++) que.push(P(es[1][i].cost,es[1][i].to));
30     used[1]=true;
31     while(que.size())
32     {
33         P p=que.top();que.pop();
34         if(used[p.second]) continue;
35         ans+=-1*p.first;
36         used[p.second]=true;
37         for(int i=0;i<es[p.second].size();i++)
38         {
39             int t=es[p.second][i].to,c=es[p.second][i].cost;
40             if(!used[t]) que.push(P(c,t));
41         }
42     }
43     for(int i=1;i<=n;i++)
44     if(!used[i]) return -1;
45     return ans;
46 }
47 
48 int main()
49 {
50 //    freopen("black.in","r",stdin);
51 //    freopen("black.out","w",stdout);
52     cin>>n>>m;
53     for(int i=0;i<m;i++)
54     {
55         int a,b,c;
56         scanf("%d %d %d",&a,&b,&c);
57         c*=-1;
58         es[a].push_back(edge{b,c});
59         es[b].push_back(edge{a,c});
60     }
61     printf("%d\n",prim());
62     return 0;
63 }

 

上一篇:python中的模块和包


下一篇:shell脚本报错:-bash:/bin/sh^M: bad interpreter: