luogu#P1342 请柬

题意:

求源点到所有点的最短路和所有点到源点的最短路之和

解法:

反向建图,跑两次从源点开始的最短路

数据比较大,spfa无法跑过,需要dijkstra

//结果都比较大,需要开long long

就是UVA721的翻译版

//double exp
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int MAXP=1000005,MAXQ=1000005;
struct Edge{int next,to,dis;}edge[MAXQ][2]; 
struct node{long long dis,u;
bool operator < (const node &x)const{return x.dis<dis;}};
long long n,p,q,head[MAXP][2],n_e[2],dis[MAXP][2];
bool vis[MAXP][2];

void dij(int s,int x);
void init();
void addedge(int from,int to,int dis,int x);
int main()
{
	long long ans=0,u,v,w;
	scanf("%d%d",&p,&q);
	memset(dis,0x7f,sizeof(dis));
	for(int i=1;i<=q;i++)
	{
		scanf("%lld%lld%lld",&u,&v,&w);
		addedge(u,v,w,1);
		addedge(v,u,w,0);
	}
	dij(1,1);
	dij(1,0);
	for(int i=2;i<=p;i++)
		ans+=dis[i][1]+dis[i][0];
	printf("%lld\n",ans);
	return 0;
}
void addedge(int from,int to,int dis,int x)
{
	edge[++n_e[x]][x].next=head[from][x];
	edge[n_e[x]][x].to=to;
	edge[n_e[x]][x].dis=dis;
	head[from][x]=n_e[x];
}
void dij(int s,int x)
{
	priority_queue<node> q;
	dis[s][x]=0;
	q.push((node){0,s});
	while(!q.empty())
	{
		node tmp=q.top();
		q.pop();
		int now=tmp.u;
		if(vis[now][x])
			continue;
		vis[now][x]=1;
		for(int i=head[now][x];i;i=edge[i][x].next)
		{
			int v=edge[i][x].to;
			if(dis[v][x]>dis[now][x]+edge[i][x].dis)
			{
				dis[v][x]=dis[now][x]+edge[i][x].dis;
				if(!vis[v][x])
					q.push((node){dis[v][x],v});
			}
		}
	}
}

posted @ 2019-10-17 20:51  nenT  阅读(89)  评论(0)    收藏  举报