最小生成树专题

最下生成树专题

---

Prim（O(v^2),堆优化版O(ElogV)）

• 思想和dijkstra差不多，因kruskal常用且简单，这里不提供prim的堆优化版

int cost[maxv][maxv];//不存在边时为INF
int d[maxv];
bool vis[maxv];

int prim(){
    for(int i=0;i<v;i++){
        vis[i]=0,d[i]=INF;
    }
    d[0]=0;
    int ans=0;
    for(int i=0;i<v;i++){
        int u=-1,MIN=INF+2;
        for(int j=0;j<n;j++)
            if(!vis[j]&&d[j]<MIN)
                MIN=d[u=j];
        if(u==-1)return -1;
        vis[u]=1;
        ans+=d[u];
        for(int j=0;j<n;j++)
            if(!vis[j]&&cost[u][j]!=INF&&cost[u][j]<d[j])
                d[j]=cost[u][j];
    }
    return ans;
}

---

Kruskal（O(ElogV)）

struct edge{int from,cost,to;}es[maxe];
int p[maxv];

bool cmp(const edge& a,const edge& b){
    return a.cost<b.cost;
}

//并查集部分
int find(int x){
    if(p[x]==x)return x;
    else p[x]=find(p[x]);
}

//Kruskal部分
int kruskal(int v,int e){
    for(int i=0;i<v;i++)p[i]=i;
    sort(es,es+v,cmp);
    int ans=0;
    
    int cnte=0;
    for(int i=0;i<e;i++){
        int px=find(es[i].from);
        int py=find(es[i].to);
        if(px!=py){
            p[px]=py;
            ans+=es[i].cost;
            cnte++;
            if(cnte==v-1)break;
        }
    }
    if(cnte!=v-1)return -1;
    else return ans;
}