PIPIOJ_1010：好坑的电子地图

Problem Description

小明是今年参加复试的外校考生，他要去民主楼小礼堂签到。由于对中南大学校本部很不熟悉，小明找到了这边读书的好朋友鲁大师，不巧，鲁大师在忙着自由探索项目的结题工作，不能给他带路，只好给他发了一份半成品的电子地图。地图上只列出了校本部内的N个点，M条路，小明处于S点，民主楼小礼堂是T点。小明感谢鲁大师，当然只是在拿到地图的一瞬间，后面的情况让他知道这半成品到底有多坑。鲁大师制作的电子地图是带有语音提示功能的，但是在编号为奇数的点他要等1分钟才能告诉他具体怎么走，而在编号为偶数的点要等2分钟。现在告诉你地图的具体情况，小明想知道他能不能在A分钟内赶到民主楼小礼堂。

Input

输入数据有多组，每组占M+1行，第一行有5个数字N，M，S，T，A，接下来M行，每行三个数字u,v,t，代表每条路的两个顶点和步行时间。（输入数据保证不含重边0<N<M<1000）

Output

对于每组输入数据，输出一行，小明能在A分钟内赶到民主楼小礼堂输出YES和最少花费的时间，否则输出KENG

Sample Input

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

Sample Output

YES 10
KENG

解题思路

1. Dijkstra算法带点权（算法笔记p377）

经验总结

1. 多组数据输入时，未对全局变量初始化导致WA

AC代码（C）

// http://pipioj.online/problem.php?id=1010
#include <stdio.h>
#include <limits.h>

#define false 0
#define true 1
#define MaxV 1000
typedef int bool;

const int INF = INT_MAX;

// 邻接矩阵
int G[MaxV][MaxV];
// 保存到该节点所需的走路时间
int time[MaxV];
// 标记当前节点是否访问过
bool visit[MaxV] = {false};
// 保存到该节点指路所需的等待时间
int wait[MaxV];
// 每个节点指路所需的等待时间
int w[MaxV];

// 初始化
void init(int N) {
    for (int i = 1; i <= N; ++i) {
        // 领接矩阵初始化
        for (int j = 1; j <= N; ++j) {
            if (i == j)
                G[i][j] = 0;
            else
                G[i][j] = INF;
        }
        // 节点访问数组初始化
        visit[i] = false;
        // 节点等待时间初始化
        wait[i] = INF;
        if (i % 2)
            w[i] = 1;
        else
            w[i] = 2;
    }
}

// Dijkstra算法部分
void Dijkstra(int N, int S) {
    // 初始化,将time数组内的元素(最短距离)都置为最大
    for (int i = 1; i <= N; ++i)
        time[i] = INF;
    // 源点的最短距离为0
    time[S] = 0;
    // 从源点出发指路所需的等待时间
    wait[S] = w[S];
    // 在N个点中找到最短距离最小且未被访问过的点
    for (int i = 0; i < N; ++i) {
        // u表示最短距离最小的点,MIN表示最短距离
        int u = -1, MIN = INF;
        // 循环N次
        for (int v = 1; v <= N; ++v) {
            // v未被访问 && 最短距离小于MIN
            if (visit[v] == false && time[v] < MIN) {
                u = v;
                MIN = time[v];
            }
        }
        // u未被找到
        if (u == -1) return;
        // 标记u为已访问
        visit[u] = true;
        // 循环N次
        for (int v = 1; v <= N; ++v) {
            // v未被访问 && u->v之间存在路径
            if (visit[v] == false && G[u][v] != INF) {
                // 以u为中介点,S到v的最短距离可被优化
                if (time[u] + G[u][v] < time[v]) {
                    // 更新S到v的最短距离
                    time[v] = time[u] + G[u][v];
                    // 更新S到v指路所需的等待时间
                    wait[v] = wait[u] + w[v];
                } else if (time[u] + G[u][v] == time[v] && wait[v] > wait[u] + w[v]) {  // S到v的最短距离相同时,选择指路等待时间短的路径
                    // 更新S到v指路所需的等待时间
                    wait[v] = wait[u] + w[v];
                }
            }
        }
    }
}

int main() {
    // N个点,M条路,源点为S,目的地为T,规定时间为A
    int N, M, S, T, A;
    while (~scanf("%d%d%d%d%d", &N, &M, &S, &T, &A)) {
        // 初始化
        init(N);
        // 录入路径
        int u, v, t;
        for (int i = 0; i < M; ++i) {
            scanf("%d%d%d", &u, &v, &t);
            G[u][v] = G[v][u] = t;
        }
        // 使用Dijkstra算法
        Dijkstra(N, S);
        // 计算总时间,需减去到达目的地T后,T的指路时间
        int ans = time[T] + wait[T] - w[T];
        if (ans <= A) {
            printf("YES %d\n", ans);
        } else
            printf("KENG\n");
    }
    return 0;
}