分层图最短路径

通信线路 

#include<iostream>
#include<queue>
#include<cstring>
#include<string>
using namespace std;
typedef pair<int,int>PII;
int n,p,k;
const int N=1000050;
const int IFN=0x3f;
int h[N];
int e[N],ne[N],w[N];
int idx;
bool st[N];
int dist[N];
void add(int a,int b,int c){
	e[idx]=b;
	ne[idx]=h[a];
	w[idx]=c;
	h[a]=idx++;
}
void dijkstra(){
	memset(dist,0x3f,sizeof(dist));
	priority_queue<PII,vector<PII>,greater<PII> >q;
	PII t;//1为距离，2为结点 
	t.first=0;
	t.second=1;
	dist[1]=0;
	q.push(t);
	while(q.size()){
		PII temp=q.top();
		q.pop();
		int l=temp.first,jie=temp.second;
		if(st[jie])continue;
		st[jie]=true;
		for(int i=h[jie];i!=-1;i=ne[i]){
			int j=e[i];
			if(dist[j]>max(dist[jie],w[i])){
				dist[j]=max(dist[jie],w[i]);
				PII o;
				o.first=dist[j];
				o.second=j;
				q.push(o);
			}
		}
	} 
}
int main(){
	cin>>n>>p>>k;
	for(int i=0;i<=n;i++){
		dist[i]=IFN;
	}
	memset(h,-1,sizeof(h));
	while(p--){
		int a,b,c;
		cin>>a>>b>>c;
		add(a,b,c);
		add(b,a,c);
		for(int j=1,z=0;j<=k;j++){
			add(a+(j-1)*n,b+j*n,z);
			add(b+(j-1)*n,a+j*n,z);
			add(a+j*n,b+j*n,c);
			add(b+j*n,a+j*n,c);
		}
	}
	for(int i=1;i<=k;i++){
		add(i*n,(i+1)*n,0);
	}
	dijkstra();
	if(dist[n*(k+1)]!=0x3f3f3f3f){
		cout<<dist[n*(k+1)]<<"\n";
	}
	else cout<<"-1"<<"\n";
} 

小雨坐地铁

#include<iostream>
#include<queue>
#include<cstring>
#include<string>
using namespace std;
typedef pair<int ,int>PII;
const int N=100005;
int h[N];
int dist[N];
int e[N],w[N],ne[N],idx;
bool st[N];
int n,m,s,t; 
void add(int a,int b,int c){
	e[idx]=b;
	ne[idx]=h[a];
	w[idx]=c;
	h[a]=idx++;
}
void dijkstra(){
	memset(dist,0x3f,sizeof(dist));
	priority_queue<PII,vector<PII>,greater<PII> >q;
	PII o;//1为距离，0结点
	o.first=0;
	o.second=s;
	dist[s]=0;
	q.push(o);
	while(q.size()){
		PII temp=q.top();
		q.pop();
		int le=temp.first,jie=temp.second;
		if(st[jie])continue;
		st[jie]=true;
		for(int i=h[jie];i!=-1;i=ne[i]){
			int j=e[i];
			if(dist[j]>w[i]+le){
				dist[j]=w[i]+le;
				PII g;
				g.first=dist[j];
				g.second=j;
				q.push(g);
			}
		}
	}
	
}
int main(){
	cin>>n>>m>>s>>t;
	memset(h,-1,sizeof(h));
	for(int i=1;i<=m;i++){
		int a,b,c;
		cin>>a>>b>>c;
		int per=-1;
		for(int j=0;j<c;j++){
			int cnt;
			cin>>cnt;
			if(j){
				add(i*n+per,i*n+cnt,b);//前一个站和当前站建边
				add(i*n+cnt,i*n+per,b);
			}
			add(cnt,i*n+cnt,a);
			add(i*n+cnt,cnt,0);//将1-n作为中转站
			per=cnt;
		}
	}
	dijkstra();
	if(dist[t]!=0x3f3f3f3f){
		cout<<dist[t]<<endl;
	}
	else cout<<"-1"<<endl;
}

飞行路线

//和第一题类似
#include<iostream>
#include<queue>
#include<cstring>
#include<string>
using namespace std;
typedef pair<int ,int>PII;
const int N=2100010;
int h[N];
int dist[N];
int e[N],w[N],ne[N],idx;
bool st[N];
int n,m,s,t,k; 
void add(int a,int b,int c){
	e[idx]=b;
	ne[idx]=h[a];
	w[idx]=c;
	h[a]=idx++;
}
void dijkstra(){
	memset(dist,0x3f,sizeof(dist));
	priority_queue<PII,vector<PII>,greater<PII> >q;
	PII o;//1为距离，0结点
	o.first=0;
	o.second=s;
	dist[s]=0;
	q.push(o);
	while(q.size()){
		PII temp=q.top();
		q.pop();
		int le=temp.first,jie=temp.second;
		if(st[jie])continue;
		st[jie]=true;
		for(int i=h[jie];i!=-1;i=ne[i]){
			int j=e[i];
			if(dist[j]>w[i]+le){
				dist[j]=w[i]+le;
				PII g;
				g.first=dist[j];
				g.second=j;
				q.push(g);
			}
		}
	}
	
}
int main(){
	cin>>n>>m>>k;
	cin>>s>>t;
	memset(h,-1,sizeof(h));
	for(int i=0;i<m;i++){
		int a,b,c;
		cin>>a>>b>>c;
		add(a,b,c);
		add(b,a,c);
		for(int j=1;j<=k;j++){
			add((j-1)*n+a,j*n+b,0);
			add((j-1)*n+b,j*n+a,0);
			add(j*n+a,j*n+b,c);
			add(j*n+b,j*n+a,c);
		}
	}
	for(int i=1;i<=k;i++){
		add(i*n+t,(i+1)*n+t,0);
	}
	dijkstra();
	if(dist[n*k+t]!=0x3f3f3f3f){
		cout<<dist[n*k+t]<<"\n";
	}
	else cout<<"-1"<<"\n";
}