天梯赛 L3-007 天梯地图(最短路/两关键字dijkstra)

题目

不同时候的样例

相同时候的样例

题解

就是裸的dijkstra，

按第一关键字比较，相同的话再比较第二关键字

只是我写了两个dijkstra，代码非常的丑就是了……

自己姿势匮乏，命名的时候各种重名

还和库函数time冲突那就叫tim叭……

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<vector>
using namespace std;
const int maxn=505;
const int maxm=maxn*maxn+10;
typedef long long ll;
int n,m;
int S,e;
int u,v,op;
int pre1[maxn],pre2[maxn];
ll dis[maxn],tim[maxn],cnt[maxn];
ll d,t;
bool vis[maxn];
ll Dis,Time;
vector<int>ans1,ans2;
char s[]=" => ";
void init()
{
	for(int i=0;i<maxn;++i)
	dis[i]=tim[i]=cnt[i]=1e18;
	memset(vis,0,sizeof vis);
}
struct node1
{
	int s;
	ll d,t;
	node1(int ss,ll dd,ll tt):s(ss),d(dd),t(tt){
	}
};
bool operator<(node1 a,node1 b)
{
	return a.t>b.t||(a.t==b.t&&a.d>b.d);
}
struct node2
{
	int s;
	ll num,d;
	node2(int ss,ll n,ll dd):s(ss),num(n),d(dd){
	}
};
bool operator<(node2 a,node2 b)
{
	return a.d>b.d||(a.d==b.d&&a.num>b.num); 
}
struct edge
{
	int v;
	ll d,t;
	edge(int vv,ll dd,ll tt):v(vv),d(dd),t(tt){
	}
};
vector<edge>E[maxm];
void add(int u,int v,ll d,ll t)
{
	E[u].push_back(edge(v,d,t));
}
void dijkstra1(int u)
{
	init();
	memset(pre1,-1,sizeof pre1);
	priority_queue<node1>q;
	q.push(node1(u,0,0));
	dis[u]=tim[u]=0;
	while(!q.empty())
	{
		node1 tmp=q.top();
		q.pop();
		int s=tmp.s;
		//printf("%d:time:%lld dis:%lld\n",s,tmp.t,tmp.d);
		if(s==e)
		{
			Time=tim[e];
			return;
		}
		if(vis[s])continue;
		vis[s]=1;
		int len=E[s].size();
		for(int i=0;i<len;++i)
		{
			edge &v=E[s][i];
			int to=v.v;
			if(vis[to])continue; 
			ll d=v.d,t=v.t;
			//printf("ndis:%lld ntime:%lld\n",d,t);
			if(tim[to]>tim[s]+t||(tim[to]==tim[s]+t&&dis[to]>dis[s]+d))
			{
				dis[to]=dis[s]+d;
				tim[to]=tim[s]+t;
				pre1[to]=s;
				q.push(node1(to,dis[to],tim[to]));
			}
		}
	}
}
void dijkstra2(int u)
{
	init();
	memset(pre2,-1,sizeof pre2);
	priority_queue<node2>q;
	q.push(node2(u,0,0));
	dis[u]=cnt[u]=0;
	while(!q.empty())
	{
		node2 tmp=q.top();
		q.pop();
		int s=tmp.s,num=tmp.num;
		if(s==e)
		{
			Dis=dis[e];
			return;
		}
		if(vis[s])continue;
		vis[s]=1;
		int len=E[s].size();
		for(int i=0;i<len;++i)
		{
			edge &v=E[s][i];
			int to=v.v;
			if(vis[to])continue; 
			ll d=v.d;
			if(dis[to]>dis[s]+d||(dis[to]==dis[s]+d&&cnt[to]>num+1))
			{
				dis[to]=dis[s]+d;
				cnt[to]=num+1;
				pre2[to]=s;
				q.push(node2(to,cnt[to],dis[to]));
			}
		}
	}
	
}
bool judge()
{
	int len=ans1.size();
	if(ans2.size()!=len)return 0;
	for(int i=len-1;i>=0;--i)
	if(ans1[i]!=ans2[i])return 0;
	return 1;
}
void output(bool same)
{
	if(same)
	{
	    printf("Time = %lld; ",Time);
	    printf("Distance = %lld: ",Dis);
	    for(int i=ans1.size()-1;i>=0;--i)
	    printf("%lld%s",ans1[i],i?s:"\n");
    }
    else
    {
    	printf("Time = %lld: ",Time);
    	for(int i=ans1.size()-1;i>=0;--i)
	    printf("%lld%s",ans1[i],i?s:"\n");
    	printf("Distance = %lld: ",Dis);
    	for(int i=ans2.size()-1;i>=0;--i)
	    printf("%lld%s",ans2[i],i?s:"\n");
    }
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;++i)
	{
		scanf("%d%d%d%lld%lld",&u,&v,&op,&d,&t);
		add(u,v,d,t);
		if(!op)add(v,u,d,t);
	}
	scanf("%d%d",&S,&e);
	dijkstra1(S);
	for(int i=e;~i;i=pre1[i])
	ans1.push_back(i);
	//for(int i=ans1.size()-1;i>=0;--i)
	//printf("%d ",ans1[i]);puts("");
	dijkstra2(S);
	for(int i=e;~i;i=pre2[i])
	ans2.push_back(i);
	//for(int i=ans2.size()-1;i>=0;--i)
	//printf("%d ",ans2[i]);puts("");
	output(judge());
	return 0;
}