最优布线问题

问题 A: 最优布线问题

时间限制: 1 Sec  内存限制: 128 MB

题目描述

学校有n台计算机，为了方便数据传输，现要将它们用数据线连接起来。两台计算机被连接是指它们间有数据线连接。由于计算机所处的位置不同，因此不同的两台计算机的连接费用往往是不同的。

    当然，如果将任意两台计算机都用数据线连接，费用将是相当庞大的。为了节省费用，我们采用数据的间接传输手段，即一台计算机可以间接的通过若干台计算机（作为中转）来实现与另一台计算机的连接。

　　现在由你负责连接这些计算机，任务是使任意两台计算机都连通（不管是直接的或间接的）。

输入

输入文件wire.in，第一行为整数n（2<=n<=100），表示计算机的数目。此后的n行，每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。

输出

输出文件wire.out，一个整数，表示最小的连接费用。

样例输入

3
0 1 2
1 0 1
2 1 0

样例输出

2

提示

注：表示连接1和2，2和3，费用为2

prime 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<iomanip>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<map>
#define V 1509
#define mod 1000000007
using namespace std;
int n,vis[V],pre[V],md[V][V],mb,d[V];
inline int read()
{
int mdd=0,sbb=1;
char ch=getchar();
while(ch<'0'||ch>'9')
{
if(ch=='-')
sbb=-1;
ch=getchar();
}
while(ch>='0'&&ch<='9')
{
mdd=mdd*10+ch-'0';
ch=getchar();
}
return mdd*sbb;
}
inline void cd()
{

memset(d,1,sizeof(d));
d[1]=0;
int v,zz,vv,sd;
while(1)
{
sd=mod;vv=0;
for(int i=1;i<=n;i++)
if(d[i]<sd&&!vis[i])
{
vv=i;sd=d[i];
}
if(!vv)return ;
vis[vv]=1;
for(int i=1;i<=n;i++)
if(d[i]>md[vv][i]&&!vis[i])
{
d[i]=md[vv][i];
}

}
}
inline int hah()
{
//freopen("wire.in","r",stdin);freopen("wire.out","w",stdout);
int x;
n=read();
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
x=read();
md[i][j]=x;
}
cd();
int sb=0;
for(int i=1;i<=n;i++)
sb+=d[i];
printf("%d",sb);
return 0;
}
int sdd=hah();
int main()
{;}

Kruskal算法 并查集使用

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<queue>
#include<set>
#include<map>
#include<deque>
#include<cstdlib>
#include<algorithm>
using namespace std;
//map<int,int>mb;
int f[1600];
//int ss=2e6;
inline int read()
{
    int sum=0,mdd=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
  if(ch=='-')mdd=-1;
        ch=getchar();
    }
while(ch<='9'&&ch>='0')
    {
        sum=sum*10+ch-'0';
        ch=getchar();
    }
    return sum*mdd;
}
struct data
{
int x,y,z;
 bool operator<(const data & a)const{return a.z<z;}
}a[3090500];
priority_queue<data>q;
inline int find(int x)
{
     if(f[x]!=x)f[x]=find(f[x]);
     return f[x];
}
int main()
{ 
//freopen("wire.in","r",stdin);
  freopen("in.txt","r",stdin);freopen("wire.out","w",stdout);
   //cin>>n>>m;
int n,m;
int x;      
cin>>n;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
  
 // x=read();
  cin>>x;
  if(x!=0)
  {
  m++;
  a[m].x=i;
  a[m].y=j;
  a[m].z=x;      
  q.push(a[m]);         //排序   
  }     
}
}
 for(int i=1;i<=n;i++)
  f[i]=i;
int tot=0;
int k=0;
for(int i=1;i<=m;i++)
{
data o=q.top();
int xx=find(o.x);
int yy=find(o.y);
if(xx!=yy)
{
f[yy]=xx;
tot+=o.z;//合并点集
k++;
}
q.pop();
if(k==n-1)
 break;
}    
cout<<tot;
   return 0;
}