E - Come Back Quickly(最短回路)

题意

​ 求有向带环(无负环)图中每个点的最短回路。


思路

​ 因为没有负环,直接 d i j k s t r a dijkstra dijkstra跑最短路,与一般的最短路的唯一的不同是先将所有与 s t a r t start start点相连的点加入队列,然后再跑,返回的 d [ s t ] d[st] d[st]就是最短回路了。

时间复杂度: O ( n m l o g m ) O(nmlogm) O(nmlogm)


代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e3+5,M=2e4+5,inf=0x3f3f3f3f,mod=1e9+7;
#define mst(a,b) memset(a,b,sizeof a)
#define PII pair<int,int>
#define fi first
#define se second
#define pb push_back
vector<PII>e[N];
int n,m,d[N];
int dij(int st){	//mlogm 
	mst(d,0x3f);
	priority_queue<PII>q;
	for(auto p:e[st]) if(d[p.fi]>p.se) d[p.fi]=p.se,q.push({-d[p.fi],p.fi});
	while(!q.empty()){
		PII now=q.top();q.pop();
		int u=now.se;
		if(d[u]<-now.fi) continue;
		for(auto p:e[u])
			if(d[p.fi]>d[u]+p.se){
				d[p.fi]=d[u]+p.se;
				q.push({-d[p.fi],p.fi});
			}
	}
	return d[st]==inf?-1:d[st];
}
int main(){	
	scanf("%d%d",&n,&m);for(int i=1,u,v,w;i<=m;i++) scanf("%d%d%d",&u,&v,&w),e[u].pb({v,w});
	for(int i=1;i<=n;i++) printf("%d\n",dij(i)); 
	return 0;
}

做的时候,以为 d i j k s t r a dijkstra dijkstra不能求环的最短路,去 f l o y d floyd floyd强行跑了,允悲。