本来差点都要放入”存疑“归档了,午休之后回来仔细检查一下代码,发现测试用例”1 1 2 6“通不过,结果是因为自己在遍历边时用的是
for(i=0;i<n;i++) 而边是有n*(n-1)/2个的。
每个人写代码一段时间后都会有固定的规范,比如这道题用并查集+kruscal就看到多种写法,有的人是多不喜欢main()里面的函数啊。
Kuruscal代码如下:
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1000;
int pre[maxn];
struct edge
{
int x;
int y;
int weight;
bool operator<(const edge& cur)const
{
return weight<cur.weight;
}
}edge[maxn*(maxn-1)/2];
int find(int x)
{
int r=x;
while(pre[r]!=r)
r=pre[r];
int i=x,j;
while(pre[i]!=r)
{
j=pre[i];
pre[i]=r;
i=j;
}
return r;
}
bool judge(int x,int y) //判断兼合并
{
int fx=find(x);
int fy=find(y);
if(fx!=fy) {pre[fx]=fy;return true;}
return false;
}
int main(void)
{
int n,m;
int a,b;
int i,j,k;
while(cin>>n&&n!=0)
{
for(i=0;i<n*(n-1)/2;i++)
cin>>edge[i].x>>edge[i].y>>edge[i].weight;
memset(pre,0,sizeof(pre));
for(i=1;i<=n;i++) pre[i]=i;
int weight_sum=0;
sort(edge,edge+n*(n-1)/2);
for(i=0;i<n*(n-1)/2;i++) //这里出现的错误真是,,写代码的时候走心了
{
if(judge(edge[i].x,edge[i].y)) weight_sum+=edge[i].weight;
}
cout<<weight_sum<<endl;
}
}
关于prim()算法,实际上就是选点,并且在每次选完点后更新<集合>到各个点的最短距离。
最开始时用dist[]来存储map[1][](因为一定会选中一个点,那么不妨从第一个点开始),之后每当选中一个新的点pos时,比较map[pos][]和dist[]的关系。
代码如下:
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1000;
int map[maxn][maxn],visit[maxn],dist[maxn];
<pre name="code" class="cpp">const int INF=999999;
int n,m;
int prim()
{
int i,j,k;
int pos,sum=0;
for(i=1;i<=n;i++) dist[i]=map[1][i];//初始化
visit[1]=1;
for(i=1;i<n;i++)
{
int min=INF;
for(j=1;j<=n;j++)
{
if(!visit[j]&&dist[j]<min)
{
pos=j;
min=dist[j];
}
}
visit[pos]=1;
for(j=1;j<=n;j++)
{
if(!visit[j]&&map[pos][j]<dist[j])
dist[j]=map[pos][j]; //进行更新
}
}
for(i=1;i<=n;i++) //<=?
sum+=dist[i];
return sum;
}
int main(void)
{
int i,j,k;
while(cin>>n)
{
if(n==0) break;
memset(visit,0,sizeof(visit));
/* for(i=1;i<n;i++)
for(j=i+1;j<=n;j++)
{
cin>>map[i][j];
map[j][i]=map[i][j];
}*/
int x,y,z;
for(i=1;i<=n*(n-1)/2;i++)
{
cin>>x>>y>>z;
map[x][y]=map[y][x]=z;
}
for(i=1;i<=n;i++) map[i][i]=0; //说明无法访问
cout<<prim()<<endl;
}
}