第三章 搜索与图论（三）

本章简介：最小生成树和二分图

一、Prim算法求最小生成树

题目：

最小生成树的实际应用：

城市之间铺设光缆，使得各个城市之间两两之间都能通信，铺设光缆的成本很高，需要设计一条线路使得光缆铺设总长度最小。

朴素prim算法：

dist[i] =INF;

for(int i = 0;i<n;i++)

        找到距离集合最近的点并将其赋值给t

        用更新其他点到集合的距离；

        st[t] = true;

证明：

 代码：

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

const int N = 510,INF = 0x3f3f3f3f;

int n,m;
int dist[N],g[N][N];
bool st[N];

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];
        st[t]=true;
        
        for(int j = 1;j<=n;j++){
            dist[j] = min(dist[j],g[t][j]);
        }
    }
    return res;
}


int main(){
    scanf("%d%d",&n,&m);
    memset(g,0x3f,sizeof g);
    for(int i = 0;i < m;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        g[u][v] =g[v][u] = min(g[u][v],w);
    }
    int t = prim();
    if(t == INF) puts("impossible");
    else printf("%d\n",t);
    
    return 0;
    
}

二、Kruskal算法求最小生成树

算法思路：

1. 将所有边按权重从小到大排序。
2. 枚举每一条边a,b,权重是c，                                                                                                                     如果a，b不连通         将这条边加入集合中。

代码：

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;
const int N = 100010,M = 200010,INF = 0x3f3f3f3f;

int n,m;
int p[N];

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

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

int kruskal(){
    sort(edges,edges+m);
    for(int i = 1;i<=n;i++) p[i] = i;
    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) return INF;
    return res;
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i = 0;i<m;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        edges[i] = {a,b,c};
    }
    
    int t = kruskal();
    if(t==INF) puts("impossible");
    else printf("%d\n",t);
    return 0;
}

三、染色法判定二分图

二分图的定义：顶点分为两个不同的子集，同时图中的每条边的两端点分别属于不同点的集合。

性质：二分图当且仅当图中不含奇数环。

证明

代码：

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

const int N = 100010,M = 200010;

int n,m;
int e[M],ne[M],h[N],idx;
int color[N];

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)) //假如不能染色成1集合
            {
                flag = false;
                break;
            }
        }
    }
    
    if(flag) puts("Yes");
    else puts("No");
    
    return 0;
}

四、匈牙利算法：

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 510,M = 100010;

int n1,n2,m;
int h[N],e[M],ne[M],idx;
bool st[N];//表示右半部分有没有被匹配
int match[N];//表示右半部分被匹配的是左半部分的哪一个点

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

//判断左半集合中的i能不能找到右半集合中的匹配。
bool find(int x){
    for(int i = h[x];i != -1;i = ne[i])
    {
        int j =e[i];
        if(!st[j])//x连接的边没有匹配,如果被匹配了就跳到下一个与之匹配的边。
        {
            st[j] = true;
            if(match[j] == 0 || find(match[j]))//虽然被匹配了但是因为每一次对左半边执行find都会重置st，也会进入判断，此时match[j]表示真正的是否匹配，寻找和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 u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    
    int res = 0;
    for(int i = 1;i <= n1;i++){
        memset(st,false,sizeof st);
        if(find(i)) res++;
    }
    
    printf("%d\n",res);
    
    return 0;
}