Submission #649374

#	Time	Username	Problem	Language	Result	Execution time	Memory
649374		mychecksedad	Crocodile's Underground City (IOI11_crocodile)	C++17	46 / 100	22 ms	31956 KiB

crocodile

#include "crocodile.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
typedef long long int ll;
const int X = 1e6, MOD = 1e9+7;



int n, m;
vector<ll> dp(X);
vector<pair<int, ll>> g[X];
bitset<X> is, vis;

void dfs(int v, int p){
    if(is[v]){
        return;
    }
    vis[v] = 1;
    vector<pair<ll, int>> dist;
    for(auto k: g[v]){
        int u = k.first, e = k.second;
        if(u == p || vis[u]) continue;
        dfs(u, v);
    }
    for(auto k: g[v]){
        int u = k.first, e = k.second;
        if(u == p) continue;
        dist.pb({dp[u] + e, u});
    }
    sort(dist.begin(), dist.end());
    for(auto k: g[v]){
        int u = k.first, e = k.second;
        if(u == p) continue;

        ll d;
        if(dist[0].second == u){
            if(dist.size() > 1)
                d = dist[1].first;
            else
                d = MOD;
        }else{
            d = dist[0].first;
        }
        dp[v] = max(dp[v], d);
    }
}

int travel_plan(int F, int M, int R[][2], int L[], int K, int P[]){ 
    n = F;
    m = M;
    for(int i = 0; i < K; ++i) is[P[i]] = 1;
    for(int i = 0; i < m; ++i){
        g[R[i][0]].pb({R[i][1], L[i]});
        g[R[i][1]].pb({R[i][0], L[i]});
    }
    dfs(0, 0);

    return dp[0];
}

Compilation message (stderr)

crocodile.cpp: In function 'void dfs(int, int)':
crocodile.cpp:22:26: warning: unused variable 'e' [-Wunused-variable]
   22 |         int u = k.first, e = k.second;
      |                          ^
crocodile.cpp:33:26: warning: unused variable 'e' [-Wunused-variable]
   33 |         int u = k.first, e = k.second;
      |                          ^

• Subtask 1: 46 / 46
• Subtask 2: 0 / 43
• Subtask 3: 0 / 11