用topsort写有向无环图的最长路

最新推荐文章于 2025-05-27 21:52:10 发布

水中月569

最新推荐文章于 2025-05-27 21:52:10 发布

阅读量87

点赞数

文章标签：算法图论

本文链接：https://blog.youkuaiyun.com/qq_64073257/article/details/131604565

版权

该代码段展示了一个用C++编写的程序，用于处理有向图并执行拓扑排序。程序首先初始化图结构，然后通过添加边并设置权重构建图。接下来，它执行两次拓扑排序，一次从源节点开始计算所有节点的最大距离，另一次计算最短路径。如果从源到目标的最短路径大于等于负无穷的一半，则输出路径距离，否则输出-1表示无路径可达。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<iostream>
#include<unordered_map>
#include<vector>
#include<deque>
#include<algorithm>
#include<map>
#include<set>
#include<cstring>
#include<cmath>
#include<queue>
#include<stack>
#define int  long long

using namespace std;
typedef  long long ll;
const int N=4e5+10;
const int INF=0x3f3f3f;
typedef pair<ll,int>PII;
int n,m;
int h[N],ne[N],e[N],idx;
int din[N],w[N],dist[N];
bool st[N];
void init(){
	memset(h,-1,sizeof h);
	memset(dist,-INF,sizeof dist);
	idx=0;
}

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

void topsort(bool flag){
	queue<int>q;
	for(int i=2;i<=n&&flag;i++){
		if(!din[i]){
			q.push(i);
		}
	}
	if(!flag){
		q.push(1);
		dist[1]=0;
	}
	while(!q.empty()){
		int u=q.front();
		q.pop();
		for(int i=h[u];~i;i=ne[i]){
			int v=e[i];
			din[v]--;
//			cout<<"u = "<<u<<" v = "<<v<<endl;
			if(!flag)dist[v]=max(dist[u]+w[i],dist[v]);
//			cout<<dist[v]<<endl;
			if(!din[v]){
				q.push(v);
			}
		}
	}
}

void solve(){
	init();
	int a,b,c;
	cin>>n>>m;
	for(int i=1;i<=m;i++){
		scanf("%lld %lld %lld",&a,&b,&c);
		add(h,a,b,c);
		din[b]++;
	}
	topsort(1);
	topsort(0);
	if(dist[n]>=-INF/2)cout<<dist[n]<<endl;
	else cout<<-1<<endl;
}

signed main(){
	int t=1;
	//cin>>t;
	while(t--) solve();
}