제출 #1194098

#제출 시각아이디문제언어결과실행 시간메모리
1194098zh_h사이버랜드 (APIO23_cyberland)C++17
0 / 100
16 ms6980 KiB
#include<bits/stdc++.h>
#define pb push_back
#define lint long long int
using namespace std;

double solve(int n, int m, int k, int h, vector<int> x, vector<int> y, vector<int> c, vector<int> abi) {
    if (n == 2) {
        return c[0];
    }

    vector<vector<pair<int, double>>> edge(n);
    for (int i = 0; i < m; i ++) {
        edge[x[i]].pb({y[i], c[i]});
        edge[y[i]].pb({x[i], c[i]});
    }

    double ans = -1;
    vector<int> visited(n, 0);
    vector<double> dis(n, 1e9);
    priority_queue<tuple<double, int, int>> pq;
    pq.push(make_tuple(0, 0, 0)); // -w, v, k
    visited[0] ++;
    dis[0] = 0;
    while (!pq.empty()) {
        tuple<double, int, int> cur = pq.top(); pq.pop();
        double w = get<0>(cur); w *= -1;
        int v = get<1>(cur), k = get<2>(cur);

        if (visited[v] == 2) {continue;}
        visited[v] ++;

        if (abi[v] == 0) {dis[v] = 0;}

        for (auto [cv, cw] : edge[v]) {
            if (dis[cv] > dis[v] + cw) {
                dis[cv] = dis[v] + cw;
                pq.push({make_tuple(-cw, cv, k)});
            }
        }
    }

    if (dis[h] == 1e9) return -1;
    else return dis[h];
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...