제출 #896604

#제출 시각아이디문제언어결과실행 시간메모리
896604NotLinuxCyberland (APIO23_cyberland)C++17
49 / 100
1038 ms11980 KiB
#include "cyberland.h"
#include <bits/stdc++.h>
using namespace std;

typedef pair < double , pair < int , int > > VAR;

const int MAXK = 31;

double solve(int n, int m, int k, int h, std::vector<int> x, std::vector<int> y, std::vector<int> c, std::vector<int> arr) {
    bitset < 31 > vis[n];
    fill(vis , vis + n , 0);

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

    priority_queue < VAR > pq;//dist , node , how much k used
    pq.push({0,{0,0}});

    function < void (int) > dfs = [&](int cur){
        if(vis[cur] == 1 or cur == h)return;
        vis[cur] = 1;
        if(arr[cur] == 0)pq.push({0,{cur,0}});
        for(auto itr : graph[cur]){
            dfs(itr.first);
        }
    };

    dfs(0);
    fill(vis , vis + n , 0);

    double ans = 1;
    while(pq.size()){
        VAR tmp = pq.top();
        pq.pop();
        if(vis[tmp.second.first][tmp.second.second])continue;
        vis[tmp.second.first][tmp.second.second] = 1;
        //cout << "pq : " << tmp.first << " " << tmp.second.first << " " << tmp.second.second << endl;
        if(tmp.second.first == h){
            if(ans == 1 or ans < tmp.first){
                ans = tmp.first;
            }
        }
        else{
            for(auto itr : graph[tmp.second.first]){
                if(arr[tmp.second.first] == 2 and tmp.second.second < k){
                    pq.push({tmp.first/2 - itr.second , {itr.first , tmp.second.second+1}});
                }
                pq.push({tmp.first - itr.second , {itr.first , tmp.second.second}});
            }
        }
    }
    return -ans;
}
#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...