算法总结归纳（第十二天）（剩余的图论）

#include<iostream>
using namespace std;
#include<cstring>
#include<queue>
const int N = 1e6 + 10;
int e[N], ne[N], h[N], w[N], idx;
int dist[N];
bool st[N], cnt;
int n, m;

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

int spfa()
{
    memset(dist, 0x3f3f3f3f, sizeof(dist));
    dist[1] = 0;
    st[1] = true;
    queue<int> q;
    q.push(1);
    while(q.size())
    {
        int t = q.front();
        q.pop();
        st[t] = false;
        
        for(int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            
                if(dist[j] > dist[t] + w[i])
                {
                    dist[j] = dist[t] + w[i];
                    if(!st[j])
                    {
                        q.push(j);
                        st[j] = true;
                    }
                    
                }
        }
    }
    if(dist[n] == 0x3f3f3f3f){
        cnt = true;
        return 1;
    }
    
    return dist[n];
    
}

int main()
{
    cin>>n>>m;
    memset(h, -1, sizeof(h));
    while( m -- )
    {
        int x, y, z;
        cin>>x>>y>>z;
        add(x, y, z);
    }
    
    int t = spfa();
    if(cnt) puts("impossible");
    else cout<<t;
    return 0;
}

spfa判断负环

题目链接：spfa判断负环

思路：

本题判断负环就是根据判断是否有环来判断，cnt数组用来判断当前点经过的边数有几条，如果大于n条边，说明经过了n + 1个点。

代码：

#include<iostream>
using namespace std;
#include<cstring>
#include<queue>
const int N = 1e5 + 10;
int idx, e[N], ne[N], h[N], w[N];
int n, m;
bool st[N];
int cnt[N], dist[N];

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

bool spfa()
{
    queue<int> q;
    
    for(int i = 1; i<=n; i++){
        q.push(i);
        st[i] = true;
    }
    
    while(q.size())
    {
        int t = q.front();
        q.pop();
        st[t] = false;
        
        for(int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if(dist[j] > dist[t] + w[i]){
                dist[j] = dist[t] + w[i];
                cnt[j] = cnt[t] + 1;
                if(cnt[j] >= n) return true;
                if(!st[j])
                {
                    q.push(j);
                    st[j] = true;
                }
            }
        }
    }
    return false;
}

int main()
{
    cin>>n>>m;
    memset(h, -1, sizeof(h));
    while( m -- )
    {
        int a, b, c;
        cin>>a>>b>>c;
        add(a, b, c);
    }
    if(spfa()) puts("Yes");
    else puts("No");
    return 0;
}

Ⅱ、floyd算法

题目链接：floyd求最短路

思路：

floyd算法就是简单的三重循环，时间复杂度是n的三次方。

多源汇问题，具有多个源点，算法实现基于动态规划推出来的，直接背熟模板就可以。

for(int k = 1; k <= n; k++){
        for(int i = 1; i<=n; i++){
            for(int j = 1; j<=n; j++){
                d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
            }
        }
    }

上面这段是核心代码，只需要背熟就好。具体推导可问度娘。

代码：

#include<iostream>
using namespace std;

const int N = 210, INF = 1e9;

int n, m, k, x, y, z;
int d[N][N];

void floyd()
{ 
   for(int k = 1; k <= n; k++){
        for(int i = 1; i<=n; i++){
            for(int j = 1; j<=n; j++){
                d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
            }
        }
    }
}



int main()
{
    cin>>n>>m>>k;
    for(int i = 1; i <= n; i++){
        for(int j = 1; j<=n; j++){
            if(i == j) d[i][j] = 0;
            else d[i][j] = INF;
        }
    }
    
    while(m -- )
    {
        cin>>x>>y>>z;
        d[x][y] = min(d[x][y], z);
    }
    floyd();
    
    while( k -- )
    {
        cin>>x>>y;
        if(d[x][y] > INF / 2) puts("impossible");
        else cout<<d[x][y]<<endl;
    }
    return 0;
}

Ⅲ、prime算法

题目链接：prime求最小生成树

思路：

prime判断最小生成树，就是使用一个dt数组表示当前数组距离生成树的距离大小，同时使用邻接矩阵g进行图的存储，st数组表示的是否已经存储进树里面。

代码：

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

const int N = 510;
int g[N][N];//存储图
int dt[N];//存储各个节点到生成树的距离
int st[N];//节点是否被加入到生成树中
int n, m;//n 个节点，m 条边

void prim()
{
    memset(dt,0x3f, sizeof(dt));//初始化距离数组为一个很大的数（10亿左右）
    int res= 0;
    dt[1] = 0;//从 1 号节点开始生成 
    for(int i = 0; i < n; i++)//每次循环选出一个点加入到生成树
    {
        int t = -1;
        for(int j = 1; j <= n; j++)//每个节点一次判断
        {
            if(!st[j] && (t == -1 || dt[j] < dt[t]))//如果没有在树中，且到树的距离最短，则选择该点
                t = j;
        }

        //如果孤立点，直返输出不能，然后退出
        if(dt[t] == 0x3f3f3f3f) {
            cout << "impossible";
            return;
        }


        st[t] = 1;// 选择该点
        res += dt[t];
        for(int i = 1; i <= n; i++)//更新生成树外的点到生成树的距离
        {
            if(dt[i] > g[t][i] && !st[i])//从 t 到节点 i 的距离小于原来距离，则更新。
            {
                dt[i] = g[t][i];//更新距离
            }
        }
    }

    cout << res;

}


int main()
{
    memset(g, 0x3f, sizeof(g));//各个点之间的距离初始化成很大的数
    cin >> n >> m;//输入节点数和边数
    while(m --)
    {
        int a, b, w;
        cin >> a >> b >> w;//输出边的两个顶点和权重
        g[a][b] = g[b][a] = min(g[a][b],w);//存储权重
        //如果有重边，那么选择权重最小的那一个。
    }

    prim();
    return 0;
}

Ⅳ、kruskai算法

题目链接：kruskal算法求最小生成树

思路：

本题使用并查集的思路，开始每个点都是独立的，当合并之后，res加上边的权重，然后将两个点的祖宗合并，然后就相当于合并一条边，cnt表示合并的边的条数，n个点，说明至少有n - 1条边，如果小于n - 1,那说明图不连通。

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 2e5 + 10;
int p[N]; int n, m;
int cnt = 0, res = 0;//记录存进去的边长

struct E
{
  int a, b, w;
  bool operator < (const E& rhs){//通过边长进行排序
        return this->w < rhs.w;
    }
} edge[N];

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

void klskr()
{
    for(int i = 1; i<=m; i++){
        int pa = find(edge[i].a);
        int pb = find(edge[i].b);
        if(pa != pb){
            res += edge[i].w;
            p[pa] = pb;
            cnt ++;
        }
    }
}

int main()
{
    cin>>n>>m;
    for(int i = 1; i<=n; i++) p[i] = i;
    for(int i = 1; i<=m; i++){
        int a, b, w;
        cin>>a>>b>>w;
        edge[i] = {a, b, w};
    }
    sort(edge + 1, edge + m + 1);
    klskr();
    
    if(cnt < n - 1)
    {
        cout<<"impossible";
        return 0;
    }
    cout<<res;
    
    return 0;
}

Ⅴ、染色法判定二分图

题目链接：染色法判定二分图

思路：

使用邻接表存储图。dfs用来判断于结点染色与上一层结点是否相同。相同，说明不是二分图，不相同，说明是二分图。

代码：

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

const int N = 1e5 + 10, M = 2e5 + 10;

int idx, e[M], ne[M], h[N];
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 t = e[i];
        if(!color[t])
        {
            if(!dfs(t, 3 - c)) return  false;
        }
        else if(color[t] && color[t] != 3 - c)
        {
            return false;
        }
        
    }
    return true;
}

int main()
{
    memset(h, -1, sizeof(h));
    cin>>n>>m;
    while( m -- )
    {
        int a, b;
        cin>>a>>b;
        add(a, b); add(b, a);
    }
    
    for(int i = 1; i<=n; i++)
    {
        if(!color[i]){
            if(!dfs(i, 1)){
                puts("No");
                return 0;
            }
        }
    }
    puts("Yes");
    return 0;
}

Ⅵ、匈牙利算法（二分图）

题目链接：匈牙利算法

思路

本题也是二分图类型，本题，match表示当前点是否有配对的点，如果有其它配对点或者没有配对点则可以和当前点配对，否则就不配对。

代码：

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

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

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;
        cin>>a>>b;
        add(a, b);
    }
    int res = 0;
    
    for(int i = 1; i<=n1; i++){
        memset(st, false, sizeof(st));
        if(find(i)) res ++;
    }
    cout<<res;
    return 0;
}

算法总结归纳（第十二天）（剩余的图论）

一、图论

Ⅰ、spfa算法

spfa求最短路

思路：

代码：