迪杰斯特拉70分,后三的点tle,已经跑反图了,求调
查看原帖
迪杰斯特拉70分,后三的点tle,已经跑反图了,求调
757101
junee楼主2023/11/7 21:52

dijkstra

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<queue> 
using namespace std;
typedef pair<int,int>PII;
const int N=1e4+10,M=1e5+10;
int h[N],ne[M],e[M],w[M],idx;
void add(int a,int b,int c){
	e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
int dist[N];
bool st[N];
void dijkstra(int s){
	memset(st,0,sizeof st);
	memset(dist,0x3f,sizeof dist);
	priority_queue<PII,vector<PII>,greater<PII> >heap;
	dist[s]=0;
	heap.push({0,s});
	while(heap.size()){
		int ver=heap.top().second;
		heap.pop();
		if(st[ver])continue;
		st[ver]=1;
		for(int i=h[ver];~i;i=ne[i]){
			int j=e[i];
			if(dist[j]>dist[ver]+w[i]){
				dist[j]=dist[ver]+w[i];
				heap.push({dist[j],j});
			}
		}
	}
}
int main(){
	freopen("1629.in","r",stdin);
	freopen("1629.out","w",stdout);
	memset(h,-1,sizeof h);
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=m;i++){
		int a,b,c;
		cin>>a>>b>>c;
		add(a,b,c);
		add(b+n,a+n,c);
	}	
	int res=0;
	dijkstra(1);
	for(int i=1;i<=n;i++)res+=dist[i];
	dijkstra(1+n);
	for(int i=1+n;i<=2*n;i++)res+=dist[i];
	cout<<res;
    return 0;
}
2023/11/7 21:52
加载中...