Submission #319912

#	Time	Username	Problem	Language	Result	Execution time	Memory
319912		knon0501	Crocodile's Underground City (IOI11_crocodile)	C++14	46 / 100	194 ms	262148 KiB

crocodile

#include "crocodile.h"
#include <bits/stdc++.h>
using namespace std;

int n;
vector<pair<int,int>> a[100001];
long long dp[100001];
int chk[100001];
long long dfs(int v,int p){
    if(chk[v])return dp[v]=0;
    vector<long long> b;
    for(auto u: a[v]){
        if(u.first==p)continue;
        b.push_back(dfs(u.first,v)+u.second);
    }
    sort(b.begin(),b.end());
    return dp[v]=b[1];
}
int travel_plan(int N, int M, int R[][2], int L[], int K, int P[])
{
    n=N;
    int i,j;
    for(i=0 ; i<K ; i++)chk[P[i]]=1;
    for(i=0 ; i<M ; i++){
        int u=R[i][0];
        int v=R[i][1];
        a[u].push_back({v,L[i]});
        a[v].push_back({u,L[i]});
    }
    dfs(0,-1);

    return dp[0];
}

Compilation message (stderr)

crocodile.cpp: In function 'int travel_plan(int, int, int (*)[2], int*, int, int*)':
crocodile.cpp:22:11: warning: unused variable 'j' [-Wunused-variable]
   22 |     int i,j;
      |           ^

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