hdu1102
#include<iostream>
#include<string>
#define MAXINT 9999999
using namespace std;
int map[101][101],s[101],dis[101],n;
int prim()
{
int sum=0;
memset(s,0,sizeof(s));
s[1]=1;
for(int i=2;i<=n;i++)
dis[i]=map[1][i];
dis[1]=0;
for(int i=1;i<n;i++)
{
int min=MAXINT,k=1;
for(int j=1;j<=n;j++)
if(!s[j]&&dis[j]<min)
k=j,min=dis[j];
sum+=min;
s[k]=1;
for(int j=1;j<=n;j++)
{
min=map[k][j];
if(min<dis[j])
dis[j]=min;
}
}
return sum;
}
int main()
{
int a,b;
while(scanf("%d",&n)==1&&n)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
scanf("%d",&map[i][j]);
int m;
scanf("%d",&m);
for(int i=0;i<m;i++)
{
scanf("%d %d",&a,&b);
map[a][b]=0;
map[b][a]=0;
}
cout<<prim()<<endl;
}
return 0;
}
呵呵,差点就被坑了,路分为已修和未修,只要把已修的路的成本改为0,就变回原原本本的最小生成树了
附上hdu1102的代码,题意也是查不多的
#include<iostream> #include<string> #define MAXINT 9999999 using namespace std; int map[101][101],s[101],dis[101],n; int prim() { int sum=0; memset(s,0,sizeof(s)); s[1]=1; for(int i=2;i<=n;i++) dis[i]=map[1][i]; dis[1]=0; for(int i=1;i<n;i++) { int min=MAXINT,k=1; for(int j=1;j<=n;j++) if(!s[j]&&dis[j]<min) k=j,min=dis[j]; sum+=min; s[k]=1; for(int j=1;j<=n;j++) { min=map[k][j]; if(min<dis[j]) dis[j]=min; } } return sum; } int main() { int a,b,c,d; while(scanf("%d",&n)==1&&n) { int m=(n*(n-1))/2; for(int i=0;i<=n;i++) for(int j=0;j<=n;j++) map[i][j]=MAXINT; for(int i=0;i<m;i++) { scanf("%d %d %d %d",&a,&b,&c,&d); if(d) c=0; if(map[a][b]>c) { map[a][b]=c; map[b][a]=c; } } cout<<prim()<<endl; } return 0;
}