这是今天xd弄的一个专题,,,,很是坑爹,,,c++过了,但是g++就是过不了,,,让我纠结了半个多小时,,,悲催,,,
prim算法:
#include<iostream> #include<string.h> #define N 505 #define M 99999999 #define FOR(i,s,t) for(int i=(s);i<=(t);++i) using namespace std; int n,m,p; int map[N][N]; bool visit[N]; int dist[N]; int prim() { FOR(i,1,n) { dist[i]=M; visit[i]=true; } int now=1; dist[now]=0; visit[now]=false; int ans=0; FOR(i,1,n) { FOR(j,1,n) if(visit[j]&&dist[j]>map[now][j]) dist[j]=map[now][j]; int minx=M; FOR(j,1,n) if(visit[j]&&minx>dist[j]) minx=dist[now=j]; visit[now]=false; } for(int i=1;i<=n;++i) if(dist[i]==M) return -1; else ans+=dist[i]; return ans; } int main() { int Case; scanf("%d",&Case); while(Case--) { scanf("%d%d%d",&n,&m,&p); for(int i=1;i<=n;++i) for(int j=1;j<=n;++j) map[i][j]=M; for(int i=0;i!=m;++i) { int a,b,c; scanf("%d%d%d",&a,&b,&c); if(map[a][b]>c) map[a][b]=map[b][a]=c; } for(int i=1;i<=p;i++) { int t,tt; scanf("%d%d",&t,&tt); for(int j=1;j<t;++j)一开始用数组整的,,听了kk了以后在,,,这样整。。 { int a; scanf("%d",&a); map[tt][a]=map[a][tt]=0; } } int ans=prim(); printf("%d\n",ans); } }krusal算法:
#include <stdio.h>
#include <string.h>
#include <set>
using namespace std;
const int MAXN = 501;
struct E{
int x,y , weight;
};
set<int> s;
E edge[25001];
int father[MAXN];
int cmp(const void *d1,const void *d2)
{ return (*(E*)d1).weight - (*(E*)d2).weight;}
void makeSet(int n)
{ for(int i = 0; i <= n; i++)
father[i] = i;
s.clear();
}
int find(int x)
{ return x==father[x]?x:father[x]=find(father[x]);}
int main()
{
int cas,n,m,k,cost;
scanf("%d",&cas);
while( cas-- )
{ scanf("%d %d %d",&n, &m, &k);
makeSet(n);
cost = 0;
for(int i = 1; i <= m; i++)
{
scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].weight);
}
int num,first,city;
for(int i = 1; i <= k;i++)
{
scanf("%d%d",&num,&first);
for(int j = 1; j < num; j++)
{ scanf("%d",&city);
int xx = find(first);
int yy = find(city);
if(xx != yy) father[yy]=xx;
}
}
qsort(edge+1,m,sizeof(E),cmp);
for(int i = 1; i <= m; i++)
{
int xx = find(edge[i].x);
int yy = find(edge[i].y);
if(xx != yy)
{ father[yy]=xx;
cost += edge[i].weight;
}
}
for(int i = 1; i <= n; i++) 整体进行压缩路径。。。
find(i);
for(int i = 1; i <= n; i++)
s.insert(father[i]); 判断根节点的个数,,,,,
if(s.size() > 1) printf("-1\n");
else printf("%d\n",cost);
}
return 0;
}