图论入门算法模板+分析+例题

本文详细介绍了图论中的最短路问题,包括Bellman-Ford算法、Dijkstra算法和Floyd-Warshall算法,讨论了它们的思路、适用条件和时间复杂度。此外,还讲解了最小生成树的Prim算法和Kruskal算法,并提供了例题和模板。

一、最短路问题

单源最短路指的是固定起点和终点,求最短路。
下文中,v表示点的个数,e表示边的个数。

1.1 单源最短路问题1(Bellman-Ford算法)

思路

只要之前有边更新过,就继续遍历所有的边,更新最小值。

适用条件

适用任意情况,可存在负边、可存在负圈。由于一些边不会即使更新,反复检查导致浪费了一些时间,下文会介绍SPFA队列优化。

模板 O ( V E ) O(VE) O(VE)

const int MAX_V=1e5+200,MAX_E=5e5+200;
struct edge{
	int from,to,cost;
};
edge es[MAX_E];

void shortest_path(int s){
	memset(d,0x3f,sizeof d);
	d[s]=0;
	while(true){
		bool update=false;
		for(auto e:es){
			if(d[e.from]!=INF&&d[e.to]>d[e.from]+e.cost){
				d[e.to]=d[e.from]+e.cost;
				update=true;
			}
		}
		if(!update)break;
	}
}

有步数限制的单源最短路

有边数限制的最短路 - AcWing题库

思路

每一次都遍历所有的边,但是要在之前的dis上进行比较和更新,因为本次遍历,上一条边可能对下一条边产生影响,这个影响是通过2步实现的,不符合我们一次遍历代表走一步的预期。所以每次遍历前,要copy dis数组。

代码
#include<bits/stdc++.h>
using namespace std;
const int M=1e4+200;
const int N=1e3;
const int INF=0x3f3f3f3f;
int n,m,k;
struct edge{
    int from,to,cost;
};
edge es[M];
int dis[N],last[N];

void bellman_ford(int s){
    memset(dis,0x3f,sizeof dis);
    dis[s]=0;
    int t=k;
    bool ud=1;
    while(ud&&t--){
        ud=0;
        memcpy(last,dis,sizeof dis);
        for(int i=1;i<=m;++i){
            edge e=es[i];
            if(last[e.from]!=INF&&dis[e.to]>last[e.from]+e.cost){
                dis[e.to]=last[e.from]+e.cost;
                //cout<<"dis["<<e.to<<"]="<<dis[e.to]<<'\n';
                ud=1;
            }
        }
    }
}

int main(){
    cin>>n>>m>>k;
    for(int i=1;i<=m;++i){
        cin>>es[i].from>>es[i].to>>es[i].cost;
    }
    bellman_ford(1);
    if(dis[n]==INF)cout<<"impossible";
    else cout<<dis[n];
    return 0;
}

判断负圈存在

负圈

负圈就是某个环路存在负边,且负边能对最短路的结果确定造成影响,使之能不断减下去。

若负圈存在,则求最短路是没有意义的。
判断是否存在负圈,看一个点到另一个点的最短路边数是否能达到N,因为一共N个点,最多N-1条边能走完,出现N表示有负环。

方法一

遍历所有的边,遍历次数若为N则存在负环

bool find_negative_loop(){
	memset(d,0,sizeof d);//其实初始值不重要(当然在不能越界条件下),判断的是他们的相对值 
	for(int i=1;i<=V;++i){
		for(auto e:es){
			if(d[e.to]>d[e.from]+e.cost){
				d[e.to]=d[e.from]+e.cost;
				if(i==V)return true; //V个点最多循环V-1次确定所有最短路,除非存在负圈 
			}
		}
	}
	return false; 
}
方法二

spfa判断负环 - AcWing题库
把所有点看成起点,循环遍历。因为题目问的是图中是否存在环,而不是求某点到某点的最短距离。

#include<bits/stdc++.h>
using namespace std;

const int N=2200;
const int M=N*4;
const int INF=0x3f3f3f3f;
int n,m;
struct edge{int to,cost;};
struct node{int id,d;};
bool vis[N];
int cnt[N];
vector<edge>G[N];
int dis[N];

bool SPFA(int s){
    fill(dis,dis+N,INF);
    fill(vis,vis+N,true);
    dis[s]=0;
    queue<node>q;
    for(int i=1;i<=n;++i)q.push({i,dis[i]});
    while(!q.empty()){
        node v=q.front();q.pop();
        vis[v.id]=false;
        for(edge e:G[v.id]){
            if(dis[e.to]>dis[v.id]+e.cost){
                dis[e.to]=dis[v.id]+e.cost;
                cnt[e.to]=cnt[v.id]+1;
                if(cnt[e.to]==N)return false;
                if(!vis[e.to]){
                    vis[e.to]=true;
                    q.push({e.to,dis[e.to]});
                }
            }
        }
    }
    return true;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=m;++i){
        int u,v,w;cin>>u>>v>>w;
        G[u].push_back({v,w});
    }
    cout<<(SPFA(1)?"No":"Yes");
  
    return 0;
}

SPFA O ( k E ) O(kE) O(kE)

思路

用队列优化bellman-ford写法,k表示每个点平均入队次数。此写法并不稳定,可能会退化成。如果比赛中题目为正边,建议用dijkstra队列优化。

代码

spfa求最短路 - AcWing题库

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+200;
const int INF=0x3f3f3f3f;
int n,m;
struct edge{
    int to,cost;
};
struct node{
    int id,d;
    bool operator<(const node&n)const{
        return d>n.id;
    }
};
vector<edge>G[N];
int dis[N];
void SPFA(int s){
    memset(dis,0x3f,sizeof dis);
    priority_queue<node>q;
    q.push({s,0});
    dis[s]=0;
    while(!q.empty()){
        auto v=q.top();q.pop();
        if(dis[v.id]<v.d)continue;
        for(auto x:G[v.id]){
            if(dis[x.to]>dis[v.id]+x.cost){
                dis[x.to]=dis[v.id]+x.cost;
                q.push({x.to,dis[x.to]});
            }
        }
    }
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=m;++i){
        int u,v,w;cin>>u>>v>>w;
        G[u].push_back({v,w});
    }
    SPFA(1);
    if(dis[n]==INF)cout<<"impossible";
    else cout<<dis[n];
    return 0;
}

总结

我先前在想如何既判断是否有负圈又求最短路,用一个函数实现两个功能,其实灵活变通就好了。因为是否有负圈和求最短路就是两个问题。因为图不一定是联通的。起点和终点在一边,负圈在另一边,哪怕负圈存在,也是存在最短路的。就算负圈不存在,如果起点和终点在两边,最短路也不存在,因为根本就没有从起点到终点的路。所以具体问题具体分析。

1.2 单源最短路问题2(Dijkstra算法)

思路

找到最短距离已经确定的点,然后更新与该点相邻的最短距离,以此循环。

适用条件

由于一旦确定某个点的最短距离,哪怕之后存在负边使得到该点的距离更小,也无法更改。

时间复杂度分析

队列优化dijkstra比bellman-ford高效,但是适用范围小,只能正边。

普通dijkstra时间复杂度,不管是用邻接表还是邻接矩阵都一样。

用邻接表访问边O(E),访问点O(V)

用邻接矩阵访问边O(1),访问点O(VE)

普通Dijkstra O ( E V ) O(EV) O(EV)

int cost[MAX_V][MAX_V];
bool used[MAX_V];
void dijkstra(int s){
	fill(d+1,d+V+1,INF);
	fill(used+1,used+V+1,false);
	d[s]=0;
	while(true){
		int v=-1;
		for(int u=1;u<=V;++u){
			if(!used[u]&&(v==-1||d[u]<d[v]))v=u;
		}
		if(v==-1)break;
		used[v]=true;
		for(int u=1;u<=V;++u){
			d[u]=min(d[u],d[v]+cost[v][u];
		}
	}
}

队列优化Dijkstra O ( E l o g V ) O(ElogV) O(ElogV)

利用priority_queue得到每次更新距离的节点当中最小距离的点

int V,E,s;
vector<edge>G[MAX_V];
int d[MAX_V];
struct edge{int to,cost;};
struct node{
	int id,d;//存储某点的id和到该点的最短距离 
	bool operator<(const node&no)const{
		return d>no.d;
	}
};
void dijkstra(int s){
	priority_queue<node>q;
	fill(d+1,d+V+1,INF);
	d[s]=0;
	q.push({s,0});
	while(!q.empty()){
		node no=q.top();q.pop();
		int v=no.id;
		if(d[v]<no.d)continue;
		for(int i=0;i<G[v].size();++i){
			edge e=G[v][i];
			if(d[e.to]>d[v]+e.cost){
				d[e.to]=d[v]+e.cost;
				q.push({e.to,d[e.to]});
			}
		}
	}
}

1.3 任意两点最短路问题(Floyd-Warshall算法)

Floyd求最短路 - AcWing

思路

利用动态规划求最短路。

dp[k][i][j]表示考虑到第k个点时,以i为起点,j为终点的最短路径。由于都是从上一层状态推导而来,可省略掉一维k。

递推公式为, d p [ i ] [ j ] = m i n ( d p [ i ] [ j ] , d p [ i ] [ k ] + d p [ k ] [ j ] ) dp[i][j]=min(dp[i][j],dp[i][k]+dp[k][j]) dp[i][j]=min(dp[i][j],dp[i][k]+dp[k][j])

全部初始化为最大值,然后dp[i][j]=0,在给边赋值的时候要注意,因为存在重边,所以要取最小值。因为存在自环,所以要判断,如果是到本身这个位置,值为0,如果取负,那就是负环,没有任何意义。

最后的结果如果大于INF/2,则判定最短路不存在。这是因为两个点之间本不存在路径,却因为一条负边值减小了。那么做这类题,就要对数据范围有一个大致的判断。

个人的一点思考,在现实生活中,重边或许还有一定意义,但是自环没有意义。题目中存在自环可能只是因为测试数据是随机的。

模板 O ( V 3 ) O(V^3) O(V3)

#include<bits/stdc++.h>
using namespace std;

const int N=300;
const int M=22000;
const int K=N*N;
const int INF=0x3f3f3f3f;
int n,m,k;
int dp[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){
                dp[i][j]=min(dp[i][j],dp[i][k]+dp[k][j]);
            }
        }
    }
}

int main(){
    ios::sync_with_stdio(false);
    cin>>n>>m>>k;
    memset(dp,0x3f,sizeof dp);
    for(int i=1;i<=n;++i)dp[i][i]=0;
    for(int i=1;i<=m;++i){
        int u,v,w;cin>>u>>v>>w;
        if(u==v)dp[u][v]=0;
        else dp[u][v]=min(dp[u][v],w);
    }
    floyd();
    while(k--){
        int i,j;cin>>i>>j;
        if(dp[i][j]>INF/2)cout<<"impossible"<<'\n';
        else cout<<dp[i][j]<<'\n';
    }
    return 0;
}

二、最小生成树问题

概念

最小生成树指的是用n-1条边将所有点联通,不存在环,路径最小的树。

2.1 最小生成树问题1(Prim算法)

思路

有一个集合X,用mincost数组存放,所有点到该集合的最短距离。刚开始放任意一个点到X,然后找出相邻最近的点,放入集合X,更新mincost数组,然后继续在集合X向外辐射的边当中找的最近的点,直到找完左右的点。如果图不连通,显然不存在最小生成树。

适用条件

用prim算法,可存在负边。

例题+模板

Prim算法求最小生成树 - AcWing题库

普通模板 O ( V E ) O(VE) O(VE)
#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
const int N=600;
const int M=1e5+200;
int cost[N][N];
int mincost[N];
bool used[N];
int n,m;

void init(){
    memset(cost,0x3f,sizeof cost);
    cin>>n>>m;
    for(int i=1;i<=m;++i){
        int u,v,w;cin>>u>>v>>w;
        cost[v][u]=cost[u][v]=min(cost[u][v],w);
    }
}
int prim(){
    memset(used,0,sizeof used);
    memset(mincost,0x3f,sizeof mincost);
    mincost[1]=0;
    int res=0;
    while(true){
        int t=-1;
        for(int i=1;i<=n;++i){
            if(!used[i]&&(t==-1||mincost[t]>mincost[i]))t=i;
        }
        if(t==-1)break;//遍历完所有的点就退出
        if(mincost[t]==INF)return INF;//剩余的点到最小生成树集合的最小距离为INF,说明不连通,最小生成树不存在。
        res+=mincost[t];
        used[t]=true;
        for(int i=1;i<=n;++i){
            mincost[i]=min(mincost[i],cost[i][t]);
        }
    }
    return res;
}
int main(){
    ios::sync_with_stdio(false);
    init();
    int res=prim();
    if(res==INF)cout<<"impossible";
    else cout<<res;
    return 0;
}
队列优化模板 O ( E l o g V ) O(ElogV) O(ElogV)

类似于dijkstra,可以用优先队列找最小mincost

需要注意的是,如果图不连通,res表示在集合中的最小生出树的值,还有一部分点在外面。因此,我们要统计在集合中的点的数量,小于n就表示图不连通,最小生成树不存在。

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> P;
const int INF=0x3f3f3f3f;
const int N=600;
const int M=1e5+200;
int cost[N][N];
int mincost[N];
bool used[N];
int n,m;

void init(){
    memset(cost,0x3f,sizeof cost);
    cin>>n>>m;
    for(int i=1;i<=m;++i){
        int u,v,w;cin>>u>>v>>w;
        cost[v][u]=cost[u][v]=min(cost[u][v],w);
    }
}
int prim(){
    priority_queue<P,vector<P>,greater<P> >q;
    memset(mincost,0x3f,sizeof mincost);
    mincost[1]=0;
    q.push({0,1});
    used[1]=true;
    int res=0,cnt=0;
    while(!q.empty()){
        P x=q.top();q.pop();
        int v=x.second;
        if(mincost[v]<x.first)continue;
        used[v]=true,cnt++;
        res+=mincost[v];
        for(int i=1;i<=n;++i){
            if(!used[i]&&mincost[i]>cost[i][v]){
                mincost[i]=cost[i][v];
                q.push({mincost[i],i});
            }
        }
    }
    if(cnt<n)return INF;
    return res;
}
int main(){
    ios::sync_with_stdio(false);
    init();
    int res=prim();
    if(res==INF)cout<<"impossible";
    else cout<<res;
    return 0;
}

2.2 最小生成树问题2(Kruskal算法)

把所有的边按权值从小到大排序,只要边的两个端点不在同一个集合,就加上这条边。在一个集合,就不能加入,否则会形成环。

例题+模板 O ( E l o g V ) O(ElogV) O(ElogV)

Kruskal算法求最小生成树 - AcWing题库

#include<bits/stdc++.h>
using namespace std;
const int INF=0x3f3f3f3f;
const int N=1e5+200;
const int M=N*2;
int p[N];
int n,m;
struct edge{
    int from,to,cost;
    bool operator<(const edge &a)const{
        return cost<a.cost;
    }
};
edge es[M];

int find(int v){
    if(p[v]!=v)p[v]=find(p[v]);
    return p[v];
}
int kruskal(){
    int cnt=0,res=0;
    sort(es+1,es+m+1);
    for(int i=1;i<=n;++i)p[i]=i;
    for(int i=1;i<=m;++i){
        int f=es[i].from,t=es[i].to;
        if(find(f)!=find(t)){
            res+=es[i].cost;
            p[find(f)]=find(t);
            cnt++;
        }
    }
    if(cnt<n-1)return INF;
    return res;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=m;++i){
        cin>>es[i].from>>es[i].to>>es[i].cost;
    }
    int res=kruskal();
    if(res==INF)cout<<"impossible";
    else cout<<res;
    return 0;
}

三、二分图判定

例题+模板

染色法判定二分图 - AcWing题库

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+200;
vector<int>G[N];
int n,m;
int color[N];
bool dfs(int i,int c){
    color[i]=c;
    for(int x:G[i]){
        if(color[x]==c)return false;//相邻点颜色相同则false
        if(color[x]==0&&!dfs(x,-c))return false;//没染过则染色
    }
    return true;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=m;++i){
        int a,b;cin>>a>>b;
        G[a].push_back(b);
        G[b].push_back(a);
    }
    for(int i=1;i<=n;++i){
        if(color[i]==0){
            if(!dfs(i,1)){
                cout<<"No";
                return 0;
            }
        }
    }
    cout<<"Yes";
    return 0;
}

四、匈牙利最大匹配

把两边看成男人和女人,要使匹配数最多,一种形象的描述方法是这样的:如果女人现在喜欢的男人有备胎,就把他绿了选新的男人,如果没有备胎就继续与这个男人在一起,如果没有喜欢的人,就直接牵手成功。

例题+模板

二分图的最大匹配 - AcWing题库

#include<bits/stdc++.h>
using namespace std;
const int N=600;
const int M=1e5+200;

int n1,n2,m;
vector<int>G[N];
bool st[N];
int match[N];
int res;
bool find(int x){//x为男人编号
    for(int i:G[x]){//遍历能匹配的女人
        if(!st[i]){
            st[i]=true;
            if(!match[i]||find(match[i])){
                match[i]=x;
                return true;
            }
        }
    }
    return false;
}

int main(){
    ios::sync_with_stdio(false);
    cin>>n1>>n2>>m;
    for(int i=1;i<=m;++i){
        int u,v;cin>>u>>v;
        G[u].push_back(v);
    }
  
    for(int i=1;i<=n1;++i){
        memset(st,0,sizeof st);//每一次新的查询,都需要重置。st的作用是为单次find服务的,以防递归死循环。
        if(find(i))res++;
    }
    cout<<res;
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值