제출 #498517

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
498517		gratus907	주유소 (KOI16_gas)	C++17	34 / 100	2067 ms	4236 KiB

gas

#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
#include <cstdlib>
#include <unordered_map>

#pragma GCC optimize("Ofast")
#define ll long long
#define eps 1e-7
#define all(x) ((x).begin()),((x).end())
#define usecppio ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
using namespace std;
using pii = pair<int, int>;

const ll INF = 987654321987654321;
const int MX = 2625;

struct Edge {
    int src, dst, w;
    bool operator<(const Edge &p) const {
        return w > p.w;
    }
};
int fuel[MX];
unordered_map<int, ll> cost[MX];
vector <Edge> G[MX];

void dijkstra(int start, vector<Edge> graph[])
{
    cost[start][fuel[start]] = 0;
    priority_queue<Edge> pq;
    pq.push({0, start,0});
    while(!pq.empty()) {
        Edge x = pq.top();
        pq.pop();
        for (auto ed : graph[x.dst]) {
            bool flag = false;
            for (pii p : cost[ed.src]) {
                int j = p.first;
                int fcost = min(fuel[ed.dst], j);
                int v = ed.dst;
                int w = ed.w * j;
                if ((cost[v].find(fcost) == cost[v].end()) or (cost[v][fcost] > p.second + w)) {
                    flag = true;
                    cost[v][fcost] = p.second + w;
                }
            }
            if (flag) pq.push(ed);
        }
    }
}

int32_t main() {
    usecppio
    int n, m; cin >> n >> m;
    for (int i = 1; i <= n; i++) cin >> fuel[i];
    for (int i = 0; i < m; i++) {
        int u, v, w; cin >> u >> v >> w;
        G[u].push_back({u, v, w});
        G[v].push_back({v, u, w});
    }
    dijkstra(1, G);
    ll ans = INF;
    for (auto p : cost[n]) ans = min(ans, p.second);
    cout << ans << '\n';
}

• Subtask 1: 18 / 18
• Subtask 2: 0 / 15
• Subtask 3: 16 / 16
• Subtask 4: 0 / 24
• Subtask 5: 0 / 27