제출 #999383

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
999383		Mr_Husanboy	악어의 지하 도시 (IOI11_crocodile)	C++17	46 / 100	145 ms	262144 KiB

crocodile

#include "crocodile.h"
#include <bits/stdc++.h>
 
using namespace std;
#define ll long long
 
template<typename T>
int len(T &a){return a.size();}
const int inf = 1e9 + 1;
 
int travel_plan(int n, int m, int r[][2], int L[], int K, int P[])
{
  vector<vector<pair<int,int>>> g(n);
  for(int i = 0; i < m; i ++){
    g[r[i][0]].push_back({r[i][1], L[i]});
    g[r[i][1]].push_back({r[i][0], L[i]});
  }
  vector<int> ex(n);
  for(int i = 0; i < K; i ++) ex[P[i]] = 1;
 
  vector<long long> dis(n);
 
  auto dfs = [&](auto &dfs, int i, int p = -1)->ll{
    if(ex[i]) return 0;
    vector<ll> res;
    for(auto [u, w] : g[i]){
      if(u == p) continue;
      res.push_back(dfs(dfs, u, i) + w);
    }
    
    if(len(res) < 2) return inf;
    sort(res.begin(), res.end());
    return res[1];
  };
 
  return dfs(dfs, 0);
}

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