最小生成树和二分图常用算法汇总

最小生成树与二分图

Prim算法（普利姆算法）- 多用于稠密图

朴素 prim算法

和dijstra算法很相似，变化的是更新是最小dist的点到集合的距离

链接:https://www.acwing.com/problem/content/860/

代码

#include <iostream>
#include <cstring>
using namespace std;

const int N = 510, INF = 0x3f3f3f3f;
int g[N][N];
int dist[N];
bool st[N];
int n,m;

int prim(){
    memset(dist, 0x3f, sizeof dist);
    
    int res = 0;
    for(int i = 0;i < n;i ++){
        int t = -1;
        for(int j = 1;j <= n;j ++)
            if(!st[j] && (t == -1 || dist[t] > dist[j]))
                t = j;
        
        if(i && dist[t] == INF) return INF;
        
        if(i) res += dist[t];
        
        for(int j = 1;j <= n;j ++) dist[j] = min(dist[j],g[t][j]);
        
        st[t] = true;
        
    }
    
    return res;
}

int main()
{
    scanf("%d%d",&n,&m);
    
    memset(g, 0x3f,sizeof g);
    while(m --){
        int a, b, c;
        scanf("%d%d%d",&a,&b,&c);
        g[a][b] = g[b][a] = min(g[a][b],c);
    }
    
    int t = prim();
    if(t == INF) puts("impossible");
    else printf("%d\n",t);
    
    return 0;
}

Kruskal算法(克鲁斯卡尔算法) - 多用于稀疏图

先对边排序，然后从小到大枚举所有的边，如果边对于的两端点不在同一连通块中（并查集判断），则将该边加入为最小生成树的生成集合。

链接:https://www.acwing.com/problem/content/861/

代码

#include <iostream>
#include <algorithm>
using namespace std;

const int N = 200010;
int n,m;

int p[N];
struct Edge {
    int a, b, w;
    
    bool operator< (const Edge &W) const
    {
        return w < W.w;
    }
}edges[N];

int find(int x){
    
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
    
}


int main()
{
    scanf("%d%d",&n,&m);
    
    for(int i = 0;i < m;i ++){
        int a, b, w;
        scanf("%d%d%d",&a,&b,&w);
        edges[i] = {a,b,w};
    }
    
    for(int i = 0;i < n;i ++) p[i] = i;
    
    sort(edges, edges + m);
    
    int res = 0,cnt = 0;
    for(int i = 0;i < m;i ++){
        int a = edges[i].a, b = edges[i].b, w = edges[i].w;
        
        a = find(a),b = find(b);
        if(a != b){
            p[a] = b;
            res += w;
            cnt ++;
        }
        
    }
    
    if(cnt < n - 1) puts("impossible");
    else printf("%d\n",res);
    
    return 0;
}

染色法判断二部图

只要二染色不出现矛盾，即任意两个顶点之间都是不同的颜色，全部染一遍，那么就是二部图，否则不是二部图

链接:https://www.acwing.com/problem/content/862/

代码

#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

const int N = 100010,M = 200010;
int h[N],e[M],ne[M],idx;
int color[N];
int n,m;

void add(int a, int b){
    e[idx] = b,ne[idx] = h[a],h[a] = idx ++;
}

bool dfs(int u,int c){
    color[u] = c;
    
    for(int i = h[u];i != -1;i = ne[i]){
        int j = e[i];
        if(!color[j])
        {
            if(!dfs(j,3 - c)) return false;
        }
        else
            if(color[j] == c) return false;
    }
    
    return true;
    
}
 
int main()
{
    scanf("%d%d",&n,&m);
 
 
    memset(h,-1,sizeof h);   
    while(m --){
        int a, b;
        scanf("%d%d",&a,&b);
        add(a,b),add(b,a);
    }
    
    bool flag = true;
    for(int i = 1;i <= n;i ++){
        if(!color[i]){
            if(!dfs(i,1)){
                flag = false;
                break;
            }
        }
    }
    
    if(flag) puts("Yes");
    else puts("No");
    
    return 0;
}

匈牙利算法求极大匹配

匈牙利算法求极大匹配的思想就是遍历二部图的一半部分，如果该部分所连的右半部分没有被匹配，则匹配，如果被匹配，则查看该点匹配的左半部分的点是否能按上述办法找到下一个可以匹配的点，来空出原来匹配的点。明显这是个递归的过程

链接:https://www.acwing.com/problem/content/863/

代码c

#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

const int N = 510,M = 100010;
int h[N],e[M],ne[M],idx;
int match[N];
bool st[N];
int n1, n2, m;


void add(int a,int b){
    e[idx] = b,ne[idx] = h[a],h[a] = idx ++;
}

bool find(int x){
    
    for(int i = h[x];i != -1;i = ne[i]){
        int j = e[i];
        if(!st[j]){
            st[j] = true;
            if(match[j] == 0 || find(match[j])){
                match[j] = x;
                return true;
            }
        }
    }
    
    return false;
    
}

int main()
{
    scanf("%d%d%d",&n1,&n2,&m);
    memset(h, -1, sizeof h);
    
    while(m --){
        int a, b;
        scanf("%d%d",&a,&b);
        add(a,b);
    }
    
    int ans = 0;
    for(int i = 1;i <= n1; i ++){
        memset(st, false, sizeof st);
        if(find(i)) ans ++;
    }
    
    printf("%d\n",ans);
    
    return 0;
}