Traffic jam FZU - 2298 (最短路问题)

题目：Traffic jam FZU - 2298 

题意：给你n个地方，这n个地方有红绿灯，每个地方的红绿灯为时间为a[i],只有时间在[0, ai), [2*ai, 3*ai), [4*ai, 5*ai)...这些时间才可以通过，然后给你m条路，告诉你通过任意一条路的时间，xi, yi, ti,通过xi,yi这条路的时间为ti,问你最短的时间从s到达t。

思路：通过题意很容易想到最短路径，但是在这个过程中需要判断需不需要等红绿灯，以及等待的时间有多长。如何判断该点是否等红绿灯？仔细观察数据，你会发现把此时的时间加起来除以a[i],判断是奇数还是偶数，如果是奇数就得等红绿灯，。但是在这个过程中，如果到达的点为终点的话，就不需要等红绿灯，所以此时需要特判一下。

代码：

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<queue>
using namespace std;
#define Max 1<<30
struct P
{
    int pos;
    int val;
    friend bool operator <(P A, P B) //按照从小到大的顺序
    {
        return A.val > B.val;
    }
};

int ma[1005][1005], dis[1005];
int vis[1005], a[1005];
int n, m;
void Dijkstra(int st, int en)
{
    priority_queue<P>que;
    memset(vis, 0, sizeof(vis));
    for(int i = 1; i <= n; i++)
    {
        dis[i] = Max;
    }
    dis[st] = 0;
    P f;
    f.pos = st, f.val = 0;
    que.push(f);
    while(!que.empty())
    {
        P p = que.top();
        que.pop();
        int v = p.pos;
        if(vis[v])continue;
        vis[v] = 1;
        for(int i = 1; i <= n; i++) //更新
        {
            if(!vis[i])
            {
                if(a[i] == 0)
                {
                    if(dis[i] > dis[v] + ma[v][i])
                    {
                        dis[i] = dis[v] + ma[v][i];
                        P f;
                        f.pos = i;
                        f.val = dis[i];
                        que.push(f);
                    }
                }
                else
                {
                     int k = 0;
                     int t = (dis[v] + ma[v][i]) / a[i];
                     if(t % 2 ==  1){//判断是否等红绿灯
                        k = (t+1)*a[i] - (dis[v]+ma[v][i]);//等待时间
                     }
                     else k = 0;
                     if(i == en){//如果为重点特判一下
                        if(dis[i] > (dis[v] + ma[v][i]))
                        {
                            dis[i] = dis[v] + ma[v][i];
                            P f;
                            f.pos = i;
                            f.val = dis[i];
                            que.push(f);
                        }
                        continue;
                     }
                     if(dis[i] > (dis[v] + ma[v][i]+k))
                        {
                            dis[i] = dis[v] + ma[v][i]+k;
                            P f;
                            f.pos = i;
                            f.val = dis[i];
                            que.push(f);
                        }
                }

            }
        }
    }
}
int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d %d", &n, &m);
        for(int i = 1; i <= n; i++)
        {
            for(int j = 1; j <= n; j++)
                ma[i][j] = Max;
            ma[i][i] = 0;
        }
        for(int i = 1; i <= n; i++)
        {
            scanf("%d", &a[i]);
        }
        for(int i = 0; i < m; i++)
        {
            int t1, t2, t3;
            scanf("%d %d %d", &t1, &t2, &t3);
            ma[t1][t2] = ma[t2][t1] = t3;
        }
        int st, en;
        scanf("%d %d", &st, &en);
        Dijkstra(st, en);
        printf("%d\n", dis[en]);
    }

    return 0;
}