Problem D: CX and girls

Description

  CX是要赶去上课，为了不迟到必须要以最短的路径到达教室，同时CX希望经过的路上能看到的学妹越多越好。现在把地图抽象成一个无向图，CX从1点出发，教室在N号点，告诉每个点上学妹的数量，每条边的长度。请你求出CX以最短路径赶到教室最多能看到多少学妹。

Input

  多组输入数据(最多20组)，输入到文件结束。

  每组数据第一行两个正整数N,M其中N代表点的个数(2<=N<=1000),M代表边的个数(1<=M<=10000)。

  接下来一行N个数，代表着1~N每个点上学妹的个数，(0<=Ni<=50)。 接下来M行，每行三个数A B C(1<=A,B<=N,0<c<=100)代表A,B间有边，长度为C。(可能存在重边)

Output

  输出CX以最短距离从1到n的情况下能看到的最多学妹数量，若从点1无法到达点N输出-1。

Sample Input

4 41 2 3 41 2 11 3 12 4 23 4 2

Sample Output

8

代码：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
typedef pair<int, int> PII;
 
const int MAXN = 1050;//节点数
const int MAXE = 20050;//边数
int N,M;
 
int head[MAXN];
int to[MAXE], next[MAXE], cost[MAXE];
int ecnt;
int mei[MAXN];
 
 
void init()   //记住要初始化！
{
    memset(head, 0, sizeof(head));
    ecnt = 1;
}
 
inline void add_edge(int u, int v, int c)//加边
{
    to[ecnt] = v;
    cost[ecnt] = c;
    next[ecnt] = head[u];
    head[u] = ecnt++;
}
 
int dis[MAXN],meizi[MAXN];
bool vis[MAXN];
 
void Dijkstra(int st, int ed)
{
    memset(dis, 0x7f, sizeof(dis));
    memset(vis, 0, sizeof(vis));
    memset(meizi,0,sizeof(meizi));
    priority_queue<PII> que;  
    que.push(make_pair(0, st));
    dis[st] = 0;   meizi[st]=mei[st];
    while(!que.empty())
    {
        int u = que.top().second;
        que.pop();
        if(vis[u]) continue;
        if(u == ed) return ;
        vis[u] = true;
        for(int p = head[u]; p; p = next[p])
        {
            int &v = to[p];
            if(dis[v] > dis[u] + cost[p])
            {
                dis[v] = dis[u] + cost[p];
                meizi[v]=meizi[u]+mei[v];
                que.push(make_pair(-dis[v], v));
            }
            else if(dis[v]==dis[u]+cost[p]){
                if(meizi[v]<meizi[u]+mei[v])
                {
                    meizi[v]=meizi[u]+mei[v];
                    que.push(make_pair(-dis[v],v));
                }
            }
        }
    }
}
 
 
int main()
{
    while(scanf("%d%d",&N,&M)!=EOF)
    {
        init();
        for(int i=1;i<=N;i++)  scanf("%d",&mei[i]);
        int x,y,z;
        for(int i=1;i<=M;i++){
            scanf("%d%d%d",&x,&y,&z);
            add_edge(x,y,z);
            add_edge(y,x,z);
        }
        Dijkstra(1,N);
        if(dis[N]>100000000) printf("-1\n");
        else printf("%d\n",meizi[N]);
    }
    return 0;
}