Submission #1023216

#	Time	Username	Problem	Language	Result	Execution time	Memory
1023216		nam6	Cyberland (APIO23_cyberland)	C++17	0 / 100	33 ms	33660 KiB

cyberland

#include "cyberland.h"
#include <bits/stdc++.h>
using namespace std; 
 
const long long INFINI = 1e17; 
const int MAXI_NOEUDS = 100*1000 + 100; 
const int MAXI_K = 70; 


vector<int> pouvoirs; 
int fin;
set<int> zeroAccess; 
vector<int> dejaVu;
 
struct Arc{
    int dest; 
    int poids; 
};
 
struct Sit{
    int node; 
    int nbDiv; 
    double dis; 
    bool operator <(const Sit &other) const{
        return dis > other.dis; 
    }
};

vector<vector<Arc>> adj;
 
void dfs(int n){
    if(dejaVu[n] == 1)
        return; 
    dejaVu[n] = 1; 
    if(n == fin){
        return;
    } 
    if(pouvoirs[n] == 0)
        zeroAccess.insert(n);
        
    for(int v=0; v<(int)adj[n].size(); v++){
        dfs(adj[n][v].dest); 
    }
}
 
double solve(int N, int M, int K, int H, vector<int> x, vector<int> y, vector<int> c, vector<int> arr) {
    
    zeroAccess.clear(); 
    zeroAccess.insert(0);
    pouvoirs = arr; 
    fin = H;;  
    K = min(K, MAXI_K);

    dejaVu = vector<int>(N, 0);
    adj = vector<vector<Arc>>(N);

    for(int n=0; n<N; n++){
        adj[x[n]].push_back({y[n], c[n]}); 
        adj[y[n]].push_back({x[n], c[n]}); 
    }
    dfs(0);
    if(!dejaVu[fin])
        return -1;  
 
    vector<vector<int>> deja(N, vector<int>(K+1, 0)); 
    vector<vector<double>> distances(N, vector<double>(K+1, 1e17)); 
    distances[fin][0]=0;
    priority_queue<Sit> aTraiter; 
    aTraiter.push({fin, 0, 0});

    while(!aTraiter.empty()){
        Sit curSit = aTraiter.top(); 
        aTraiter.pop(); 
        if(curSit.node == fin && curSit.nbDiv != 0)
            continue; 
        if(deja[curSit.node][curSit.nbDiv])
            continue; 
        if(zeroAccess.count(curSit.node)){
            return curSit.dis; 
        }
        deja[curSit.node][curSit.nbDiv] = 1; 

        for(int i=0; i<(int)adj[curSit.node].size(); i++){
            Arc voisin = adj[curSit.node][i];
            int v = voisin.dest; 
            int w = voisin.poids; 
            // pas diviser par 2
            if( distances[curSit.node][curSit.nbDiv] + (double)w/(double)pow(2, curSit.nbDiv) < distances[v][curSit.nbDiv]){
                distances[v][curSit.nbDiv] = distances[curSit.node][curSit.nbDiv] + (double)w/(double)pow(2, curSit.nbDiv);
                aTraiter.push({v, curSit.nbDiv, distances[v][curSit.nbDiv]});
            }
            // diviser par 2
            if(pouvoirs[v] == 2 && curSit.nbDiv < K){
                if(distances[curSit.node][curSit.nbDiv] + (double)w/(double)pow(2, curSit.nbDiv) <  distances[v][curSit.nbDiv + 1]){
                    distances[v][curSit.nbDiv + 1] = distances[curSit.node][curSit.nbDiv] + (double)w/(double)pow(2, curSit.nbDiv);
                    aTraiter.push({v, curSit.nbDiv+1, distances[v][curSit.nbDiv + 1]});
                }
            }        
        }
    }
    return -1; 

}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 8
• Subtask 3: 0 / 13
• Subtask 4: 0 / 19
• Subtask 5: 0 / 7
• Subtask 6: 0 / 16
• Subtask 7: 0 / 29
• Subtask 8: 0 / 3