1010: 好坑的电子地图

题目描述
小明是今年参加复试的外校考生，他要去民主楼小礼堂签到。由于对中南大学校本部很不熟悉，小明找到了这边读书的好朋友鲁大师，不巧，鲁大师在忙着自由探索项目的结题工作，不能给他带路，只好给他发了一份半成品的电子地图。地图上只列出了校本部内的N个点，M条路，小明处于S点，民主楼小礼堂是T点。小明感谢鲁大师，当然只是在拿到地图的一瞬间，后面的情况让他知道这半成品到底有多坑。鲁大师制作的电子地图是带有语音提示功能的，但是在编号为奇数的点他要等1分钟才能告诉他具体怎么走，而在编号为偶数的点要等2分钟。现在告诉你地图的具体情况，小明想知道他能不能在A分钟内赶到民主楼小礼堂。
输入
输入数据有多组，每组占M+1行，第一行有5个数字N，M，S，T，A，接下来M行，每行三个数字u,v,t，代表每条路的两个顶点和步行时间。（输入数据保证不含重边0<N<M<1000）
输出
对于每组输入数据，输出一行，小明能在A分钟内赶到民主楼小礼堂输出YES和总共花费的时间，否则输出KENG
样例输入
4 3 1 4 10
1 2 1
3 2 2
3 4 3
5 4 2 4 7
1 2 5
5 4 2
3 5 1
2 3 1
样例输出
YES 10
KENG

//单源最短路径
#include<stdio.h>
#define INF 0x7ffff
int MG[1001][1001]; 
bool vis[1001]; 
int dist[1001];
int dij(int N,int S,int T){//求S到T的最短距离
	//初始化数组dist[1001],每一点到源点的距离皆为无穷大，并标记未访问 
	for(int i=1;i<1001;i++){
		dist[i]=INF;
		vis[i]=false;
	} 
	//将源点加入，起点到起点的距离为0（起点的wait也计算，终点不用计算wait值） 
	dist[S]=0;
	//vis[S]=true;若vis不加入，S必然会在第一趟被加入，wait也处理了，统一了处理的过程 
	for(int i=1;i<=N;i++){
		int min=INF;//标记最短距离 
		int index;//存放新的要更新的点的，下标正好与点对应
		int wait;//wait表示等待距离
		//找出到当前加入点的最短距离的点
		for(int j=1;j<=N;j++){
			if(!vis[j]&&dist[j]<=min){
				min=dist[j];
				index=j;
			}
		} 
		//将该点加入 
		vis[index]=true;
		if(index%2==0){
			wait=2;
		}
		else {
			wait=1;
		}
		//更新其余未被访问过的点，到新更新的点的距离 
		for(int j=1;j<=N;j++){
			if(dist[j]>dist[index]+MG[index][j]+wait){
				dist[j]=dist[index]+MG[index][j]+wait;
			}
		}
	} 
	return dist[T];
}
int main(void){
	int N,M,S,T,A;
	while(scanf("%d%d%d%d%d",&N,&M,&S,&T,&A)){
		//初始化邻接矩阵
		for(int i=1;i<1001;i++){
			for(int j=1;j<1001;j++){
				MG[i][j]=INF;
			}
		} 
		//将路径存入图中
		for(int i=1;i<=M;i++){
			int A,B,D;
			scanf("%d%d%d",&A,&B,&D);
			MG[A][B]=D;
			MG[B][A]=D;
		}
		//调用调整后的dijstra算法，获取时间
		int time=dij(N,S,T);
		if(time>A){
			printf("KENG\n");
		} 
		else{
			printf("YES %d\n",time);
		}
	}
	return 0;
}