제출 #292473

#제출 시각아이디문제언어결과실행 시간메모리
292473davooddkareshki경주 (Race) (IOI11_race)C++17
100 / 100
569 ms97272 KiB
//#include "race.h"
#include<bits/stdc++.h>

using namespace std;

//#define int long long
#define F first
#define S second
#define pii pair<ll,ll>
#define mpr make_pair

typedef long long ll;

const int maxn = 2e5+10;
const int inf = 1e9+10;
const int mod = 1e9+7;

ll K;
map<ll,ll> *mp[maxn];
ll st[maxn], h[maxn], sum[maxn];
vector<pii> g[maxn];

void pfs(ll v, ll p = -1)
{
   st[v] = 1;
   for(auto e : g[v])
   {
      ll u = e.F, w = e.S;
      if(u != p)
      {
         sum[u] = sum[v] + w;
         h[u] = h[v] + 1;

         pfs(u,v);
         st[v] += st[u];
      }
   }
}

ll ans = inf;
void dfs(ll v, ll p = -1)
{
   ll big = -1;
   for(auto e : g[v])
   {
      int u = e.F;
      if(u != p)
      {
         dfs(u,v);
         if(big == -1) big = u;
         else if(st[u] > st[big]) big = u;
      }
   }

   if(big != -1) mp[v] = mp[big];
   else mp[v] = new map<ll,ll>;
   (*mp[v])[sum[v]] = h[v];

   if((*mp[v]).count(K+sum[v]))
      ans = min(ans, (*mp[v])[K+sum[v]] - h[v]);

   for(auto e : g[v])
   {
      ll u = e.F, w = e.S;
      if(u != p && u != big)
      {
         for(auto X : (*mp[u]))
         {
            ll sm = X.F, Mn = X.S;
            if((*mp[v]).count(K-sm+2*sum[v]))
            {
               ll road = Mn + (*mp[v])[K-sm+2*sum[v]] - 2 * h[v];
               ans = min(ans,road);
            }
         }

         for(auto X : (*mp[u]))
         {
            ll sm = X.F, Mn = X.S;
            if(!(*mp[v]).count(sm)) (*mp[v])[sm] = Mn;
            else (*mp[v])[sm] = min((*mp[v])[sm], Mn);
         }
      }
   }
}

int best_path(int n, int k, int h[][2], int l[])
{
   //cin>> n >> K;
   K = k;
   for(int i = 0, u, v, w; i < n-1; i++)
   {
      //cin>> u >> v >> w;
      u = h[i][0], v = h[i][1], w = l[i];
      u++; v++;
      g[u].push_back({v,w});
      g[v].push_back({u,w});
   }

   pfs(1); dfs(1);
   if(ans == inf) return -1;
   else return ans;
}

/*int main()
{
}*/

컴파일 시 표준 에러 (stderr) 메시지

race.cpp: In function 'void dfs(ll, ll)':
race.cpp:64:19: warning: unused variable 'w' [-Wunused-variable]
   64 |       ll u = e.F, w = e.S;
      |                   ^
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...