zoj 1586 QS Network

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=586

解题思路：最小生成树（kruskal 算法 + 优先队列 + 并查集）

  1 /**************************************************************************
  2 user_id: SCNU20102200088
  3 problem_id: zoj 1586
  4 problem_name: QS Network
  5 **************************************************************************/
  6 
  7 #include <algorithm>
  8 #include <iostream>
  9 #include <iterator>
 10 #include <iomanip>
 11 #include <sstream>
 12 #include <fstream>
 13 #include <cstring>
 14 #include <cstdlib>
 15 #include <climits>
 16 #include <bitset>
 17 #include <string>
 18 #include <vector>
 19 #include <cstdio>
 20 #include <cctype>
 21 #include <ctime>
 22 #include <cmath>
 23 #include <queue>
 24 #include <stack>
 25 #include <list>
 26 #include <set>
 27 #include <map>
 28 using namespace std;
 29 
 30 //线段树
 31 #define lson l,m,rt<<1
 32 #define rson m+1,r,rt<<1|1
 33 
 34 //手工扩展栈
 35 #pragma comment(linker,"/STACK:102400000,102400000")
 36 
 37 const double EPS=1e-9;
 38 const double PI=acos(-1.0);
 39 const double E=2.7182818284590452353602874713526;  //自然对数底数
 40 const double R=0.5772156649015328606065120900824;  //欧拉常数:(1+1/2+...+1/n)-ln(n)
 41 
 42 const int x4[]={-1,0,1,0};
 43 const int y4[]={0,1,0,-1};
 44 const int x8[]={-1,-1,0,1,1,1,0,-1};
 45 const int y8[]={0,1,1,1,0,-1,-1,-1};
 46 
 47 typedef long long LL;
 48 
 49 typedef int T;
 50 T max(T a,T b){ return a>b? a:b; }
 51 T min(T a,T b){ return a<b? a:b; }
 52 T gcd(T a,T b){ return b==0? a:gcd(b,a%b); }
 53 T lcm(T a,T b){ return a/gcd(a,b)*b; }
 54 
 55 ///////////////////////////////////////////////////////////////////////////
 56 //Add Code:
 57 int n,s[1005],a[1005][1005];
 58 
 59 struct Node{
 60     int u,v,w;
 61     bool operator <(const Node &a) const{
 62         return w>a.w;
 63     }
 64 }node;
 65 
 66 priority_queue<Node> pq;
 67 
 68 void Union(int x,int y){
 69     s[y]=x;
 70 }
 71 
 72 int Find(int x){
 73     if(s[x]<0) return x;
 74     return s[x]=Find(s[x]);
 75 }
 76 
 77 int Kruskal(){
 78     int num=0,ret=0;
 79     memset(s,-1,sizeof(s));
 80     while(!pq.empty() && num<n-1){
 81         node=pq.top();
 82         pq.pop();
 83         if(Find(node.u)!=Find(node.v)){
 84             Union(Find(node.u),Find(node.v));
 85             ret+=node.w;
 86             num++;
 87         }
 88     }
 89     while(!pq.empty()) pq.pop();
 90     return ret;
 91 }
 92 ///////////////////////////////////////////////////////////////////////////
 93 
 94 int main(){
 95     std::ios::sync_with_stdio(false);
 96     //freopen("in.txt","r",stdin);
 97     //freopen("out.txt","w",stdout);
 98     ///////////////////////////////////////////////////////////////////////
 99     //Add Code:
100     int Case,i,j,x[1005];
101     scanf("%d",&Case);
102     while(Case--){
103         scanf("%d",&n);
104         for(i=1;i<=n;i++) scanf("%d",&x[i]);
105         for(i=1;i<=n;i++){
106             for(j=1;j<=n;j++) scanf("%d",&a[i][j]);
107         }
108         for(i=1;i<n;i++){
109             for(j=i+1;j<=n;j++){
110                 node.u=i;
111                 node.v=j;
112                 node.w=a[i][j]+x[i]+x[j];
113                 pq.push(node);
114             }
115         }
116         int ans=Kruskal();
117         printf("%d\n",ans);
118     }
119     ///////////////////////////////////////////////////////////////////////
120     return 0;
121 }
122 
123 /**************************************************************************
124 Testcase:
125 Input:
126 1
127 3
128 10 20 30
129 0 100 200
130 100 0 300
131 200 300 0
132 Output:
133 370
134 **************************************************************************/

posted on 2013-10-09 09:55  SCNU20102200088  阅读(129)  评论(0)  编辑  收藏  举报

导航