每日一题 - 231031 - P4568 [JLOI2011] 飞行路线

• P4568 [JLOI2011] 飞行路线

//
#include<bits/stdc++.h>
using namespace std;
//#define int long long

const int INF=0x3f3f3f3f;
const int U=1e4+5;				// units
const int N=11*U;				// n*k+n
const int M=5*U*(2+4*10)+10;	// m*(2+4*k)+10
int h[N],idx;
struct A{ int y,data,next; }e[M];
int n,m,k,st,ed;

inline void init()
{
	memset( h,-1,sizeof( h ) );
}

inline void add( int x,int y,int data )
{
	e[idx]=(A){ y,data,h[x] };
	h[x]=idx++;
}

struct B
{
	int x,dis;
	bool operator < ( const B& in ) const
	{
		return in.dis < dis ;
	}
};

int dis[N];
bool used[N];

void dijkstra()
{
	memset( dis,0x3f,sizeof( dis ) );
	dis[st]=0;
	
	priority_queue<B> q;
	q.push( (B){ st,0 } );
	
	while( !q.empty() )
	{
		int x=q.top().x; 
		q.pop();
	
		if( used[x] ) continue;
		used[x]=1;
		
		for( int i=h[x];~i;i=e[i].next )
		{
			int y=e[i].y;
			int data=e[i].data;
			
			if( dis[y]>dis[x]+data )
			{
				dis[y]=dis[x]+data;
				q.push( (B){ y,dis[y] } );
			}
		}
	}
}

void solve()
{
	init();
	
	scanf("%d%d%d",&n,&m,&k );
	scanf("%d%d",&st,&ed );
	while( m-- )
	{
		int x,y,data;
		scanf("%d%d%d",&x,&y,&data );
		add( x,y,data );
		add( y,x,data );
		
		for( int i=1;i<=k;i++ )
		{
			add( x+i*n,y+i*n,data );
			add( y+i*n,x+i*n,data );
			add( x+(i-1)*n,y+i*n,0 );
			add( y+(i-1)*n,x+i*n,0 );
		}
	}	
	for( int i=1;i<=k;i++ ) add( ed+(i-1)*n,ed+i*n,0 );
	
	dijkstra();
	
	printf("%d\n",dis[ed+k*n] );
}

signed main()
{
	solve();
	return 0;
}

---

作者 | 乐意奥AI