穷游？“穷”游？

Description

贫穷的小A有一个梦想，就是到t国去一次穷游，但现实是残酷的。小A所在的世界一共有n(n<=500)个国家，国家与国家之间总共有E(E<=50000)条道路相连，第i个国家对于进入它的外国人都要收取Bi的费用，而小A家住在s国，他必须通过这些道路在各个国家之间中转最终到达t国（除非他运气够好可以直接从s国到达t国）。但是贫穷的小A只剩下M(M<=100)元家底了，因此他必须精打细算旅途的费用，同时小A对于t国实在太向往了，因此他希望能够走最短的路尽快到达t国。这个问题难倒了小A，现在他请你帮他算一算他到达t国的最短路径有多长。

Input

第一行输入T(T<=10)表示有T组数据。每组数据第一行输入n、E、s、t、M，分别表示小A所在世界的国家数、国家之间的总道路数、小A的国籍、小A向往的国家以及小A的家底；接下来一行输入n个正整数Bi，表示第i个国家收取的过路费（由于小A是s国人，因此s国不会收取，但t国会）；接下来输入E行每行三个正整数u(1<=u<=n)、v(1<=v<=n)、w，表示u国和v国之间存在着一条长度为w的无向边（可能有重边）。输入保证最终结果不会使int溢出。

Output

输出T行正整数，第i行表示第i组数据小A花费不超过M元到达t国的最短路。若小A无法到达t国，输出-1.

Sample Input

3
2 2 1 2 10
20 10
1 2 1
1 2 2
3 1 1 3 10
1 1 1
2 3 1
3 3 1 3 10
1 11 1
1 2 1
1 2 3
2 3 1

Sample Output

1
-1
-1

#include <iostream>
#include <vector>
#include <queue>
#include <climits>
using namespace std;

struct Node {
    int cost;       // 距离
    int country;    // 到达了哪个国家
    int spent;      // 花费了多少钱
    Node(int c, int co, int s) : cost(c), country(co), spent(s) {}
    bool operator>(const Node& other) const {
        return cost > other.cost;
    }
};

typedef pair<int, int> pii;

vector<vector<int>> dijkstra(vector<vector<pii>>& graph, int start, int max_cost, vector<int>& countries) {
    int n = graph.size();
    vector<vector<int>> dist(n+1, vector<int>(max_cost + 1, INT_MAX));
    dist[start][0] = 0;
    priority_queue<Node, vector<Node>, greater<Node>> pq;
    pq.push(Node(0, start, 0));

    while (!pq.empty()) {
        int cost = pq.top().cost;
        int u = pq.top().country;
        int spent = pq.top().spent;
        pq.pop();
        if (cost > dist[u][spent]) continue;
        for (auto edge : graph[u]) {
            int v = edge.first;
            int w = edge.second;
            int new_spent = spent + countries[v];
            if (new_spent <= max_cost && cost + w < dist[v][new_spent]) {
                dist[v][new_spent] = cost + w;
                pq.push(Node(dist[v][new_spent], v, new_spent));
            }
        }
    }

    return dist;
}

int main() {
    int T;
    cin >> T;
    while (T--) {
        int n, E, s, t, M;
        cin >> n >> E >> s >> t >> M;
        vector<int> countries(n);
        for (int i = 0; i < n; i++) {
            cin >> countries[i];
        }
        vector<vector<pii>> graph(n);
        for (int i = 0; i < E; i++) {
            int u, v, w;
            cin >> u >> v >> w;
            graph[u-1].push_back({v-1, w});
            graph[v-1].push_back({u-1, w});
        }

        vector<vector<int>> dist = dijkstra(graph, s-1, M, countries);
        int min_cost = INT_MAX;
        for (int spent = 0; spent <= M; spent++) {
            min_cost = min(min_cost, dist[t-1][spent]);
        }
        if (min_cost == INT_MAX) {
            cout << -1 << endl;
        } else {
            cout << min_cost << endl;
        }
    }
    return 0;
}