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;
}