首先,什么是最小生成树?
他就是无向图G中的所有生成树中树枝权值总和最小的。
如何求?
我们不妨采用以下的贪心策略:
Prim算法(复杂度:(n+m)logm):
我们对于把上述的点看成两个集合,一个是确定了最小生成树的点,一个还没有确定,我们只要不断把距离已经确定的集合的最短的边添加进去即可。假如我们加的距离不是最小的,那么当我们假设未确定的点已经构成了他们点的最小生成树,那么我们此时用距离最小的去添加他们肯定更优。(我们对于那先未确定的点的集合,不管用什么边去联系他们任何一个点,都不会影响他们以后的最小生成树的形状,这也是贪心当前最优解可以推出全局最优解的保证)
来道模板题:
因为传递消息,至少连n-1条边,又要距离min,相当于求最小生成树,下面是AC代码(我们可以优化一下,对于还未拿出的边,若有一个比他长的则不放入队列):
#include<bits/stdc++.h>
using namespace std;
int n,m,head[100010],a,b,v,cnt,sum;
struct node{
int len,dian,next;
}edge[1000005];
void addedge(int x,int y,int v){
edge[++cnt].len=v;
edge[cnt].dian=y;
edge[cnt].next=head[x];
head[x]=cnt;
}
int dis[100010];
struct ty{
int bian,name;
bool operator<(const ty &a) const{
return bian>a.bian;
}
};
bool vis[1000001];
priority_queue<ty> q;
int prim(){
q.push({0,1});
while(!q.empty()){
ty ck=q.top();
q.pop();
if(vis[ck.name]==1) continue;
vis[ck.name]=1;
sum+=ck.bian;
for(int i=head[ck.name];i!=-1;i=edge[i].next){
if(vis[edge[i].dian]==1) continue;
if(dis[edge[i].dian]<=edge[i].len) continue;
dis[edge[i].dian]=edge[i].len;
q.push({edge[i].len,edge[i].dian});
}
}
return sum;
}
int main(){
memset(head,-1,sizeof(head));
memset(vis,0,sizeof(vis));
memset(dis,0x3f,sizeof(dis));
cin>>n>>m;
for(int i=1;i<=m;i++){
scanf("%d%d%d",&a,&b,&v);
addedge(a,b,v);
addedge(b,a,v);
}
cout<<prim();
}
Kruskal算法(复杂度:mlogm):
还是采取贪心策略,只不过这次是直接选所有边下的最短边,若他们连起来还是树,就连起来,反之舍弃,用并查集维护即可。
首先,我们注意到如果每一次都可以选min的n-1条边就是最优的情况。
但是,在实际上,可能边会在同一个并查集中,说明这条边可以发挥构成树的作用,当时已经存在一点,他的作用是一样的,但是它的距离更小,因此更优。换句话说,我们就是在选n-1个在构建生成树的发挥不同作用的边,而之所以要放弃,是因为功能的重叠。
综上,这样选取的策略最优。
下面给出AC代码:
#include<bits/stdc++.h>
using namespace std;
int n,m,fa[100010],a,b,v,cnt,sum;
struct node{
int len,x,y;
}edge[1000005];
bool cmp(node a,node b){
return a.len<b.len;
}
int find(int x){
if(fa[x]==x) return x;
else return fa[x]=find(fa[x]);
}
void merge(int x,int y){
fa[find(x)]=find(y);
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++) fa[i]=i;
for(int i=1;i<=m;i++){
scanf("%d%d%d",&a,&b,&v);
edge[++cnt].x=a;
edge[cnt].y=b;
edge[cnt].len=v;
}
sort(edge+1,edge+1+m,cmp);
for(int i=1;i<=m;i++){
int xx=find(edge[i].x);
int yy=find(edge[i].y);
if(xx==yy) continue;
sum+=edge[i].len;
merge(xx,yy);
}
cout<<sum;
}