Invitation Cards——正反最短路

poj 1511: Invitation Cards

题意：点1到其他点的最短距离+其他点到点1的最短距离 的总和。

1.双向建图 正向建图是1到其他点的距离，反向建图则是其他点到1的距离。

2.数据比较多利用邻接表存图。

3.大量数据输入利用scanf

4.正反跑两边最短路模板即可。

spfa

#define __ ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<set>
#include<queue>
using namespace std;
typedef long long ll;
const int N=2e6+10;
int rh[N],h[N],e[N],w[N],ne[N],idx;
int d[N],b[N],rd[N];

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

int n;
void spfa2(){
	memset(d,0x3f ,sizeof d);
	memset(b,0,sizeof b);
	queue<int>q;
	d[1]=0;
	q.push(1);
	while(q.size()){
		int t=q.front();
		q.pop();
		b[t]=0;
		for(int i=h[t];i!=-1;i=ne[i]){
			int j=e[i];
			if(d[j]>d[t]+w[i]){
				d[j]=d[t]+w[i];
				if(!b[j]){
					b[j]=1;
					q.push(j);
				}
			}
		}
	}	
}
void spfa1(){
	memset(rd,0x3f ,sizeof rd);
	memset(b,0,sizeof b);
	queue<int>q;
	rd[1]=0;
	q.push(1);
	while(q.size()){
		int t=q.front();
		q.pop();
		b[t]=0;
		for(int i=rh[t];i!=-1;i=ne[i]){
			int j=e[i];
			if(rd[j]>rd[t]+w[i]){
				rd[j]=rd[t]+w[i];
				if(!b[j]){
					b[j]=1;
					q.push(j);
				}
			}
		}
	}	
}
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		memset(h,-1,sizeof h);
		memset(rh,-1,sizeof rh);
		int m;
		scanf("%d%d",&n,&m);
		while(m--){
			int a,b,c;
			scanf("%d%d%d",&a,&b,&c);
			add(h,a,b,c);
			add(rh,b,a,c);
		}
		long long res=0;
		spfa1();
		spfa2();
		for(int i=1;i<=n;i++)res+=d[i]+rd[i];
		cout<<res<<endl;
	}
	return 0;
}

Dijkstra-优化

#define __ ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include<iostream>
#include<cstring>
#define x first
#define y second
#include<cstdio>
#include<algorithm>
#include<set>
#include<queue>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N=3e6+10;
int rh[N],h[N],e[N],w[N],ne[N],idx;
int d[N],b[N],rd[N];

void add(int h[],int a,int b,int c){
	e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int n;
void dj1(){
	memset(d,0x3f ,sizeof d);
	memset(b,0,sizeof b);
	d[1]=0;
	priority_queue<PII,vector<PII>,greater<PII> > q;
	q.push({0,1});
	while(q.size()){
		PII t=q.top();
		q.pop();
		int x=t.y;
		if(b[x])continue;
		b[x]=1;
		for(int i=h[x];i!=-1;i=ne[i]){
			int j=e[i];
			if(d[j]>d[x]+w[i]){
				d[j]=d[x]+w[i];
				q.push({d[j],j});
			}
		}
	}
}
void dj2(){
	memset(rd,0x3f ,sizeof rd);
	memset(b,0,sizeof b);
	rd[1]=0;
	priority_queue<PII,vector<PII>,greater<PII> > q;
	q.push({0,1});
	while(q.size()){
		PII t=q.top();
		q.pop();
		int x=t.y;
		if(b[x])continue;
		b[x]=1;
		for(int i=rh[x];i!=-1;i=ne[i]){
			int j=e[i];
			if(rd[j]>rd[x]+w[i]){
				rd[j]=rd[x]+w[i];
				q.push({rd[j],j});
			}
		}
	}
}
int main(){
	int t;
	scanf("%d",&t);
	while(t--){
		idx=0;
		memset(h,-1,sizeof h);
		memset(rh,-1,sizeof rh);
		int m;
		scanf("%d%d",&n,&m);
		while(m--){
			int a,b,c;
			scanf("%d%d%d",&a,&b,&c);
			add(h,a,b,c);
			add(rh,b,a,c);
		}
		long long res=0;
		dj1();
		dj2();
		for(int i=1;i<=n;i++)res+=d[i]+rd[i];
		cout<<res<<endl;
	}
	return 0;
}