| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 166890 | wmrmr | 공장들 (JOI14_factories) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "factories.h"
using namespace std;
const int MAX = 5e5, LOG = 22;
const long long int INF = 1e18;
vector<int> g[MAX], p[MAX];
vector<int> rList;
int troidProf[MAX], troidPai[LOG][MAX], sub[MAX];
long long int troidDist[LOG][MAX], mnDist[MAX];
bool isTroid[MAX], active[MAX];
void CalcSub(int v, int pai)
{
sub[v] = 1;
for(int i=0;i<g[v].size();i++)
{
int prox = g[v][i];
if(prox == pai || isTroid[prox]) continue;
CalcSub(prox,v);
sub[v] += sub[prox];
}
return;
}
int FindTroid(int v, int pai, int sz)
{
for(int i=0;i<g[v].size();i++)
{
int prox = g[v][i];
if( prox == pai || isTroid[prox] ) continue;
if( sub[prox] > sz/2 ) return FindTroid(prox,v,sz);
}
return v;
}
void CalcDist(int v, int pai, int prof, int troid)
{
troidPai[prof][v] = troid;
for(int i=0;i<g[v].size();i++)
{
int prox = g[v][i], peso = p[v][i];
if(prox == pai || isTroid[prox]) continue;
troidDist[prof][prox] = troidDist[prof][v] + peso;
CalcDist(prox,v,prof,troid);
}
return;
}
void Decompose(int v, int prof)
{
CalcSub(v,v);
int tempN = sub[v];
int troid = FindTroid(v,v,tempN);
isTroid[v] = 1;
troidProf[troid] = troid;
CalcDist(troid,troid,prof,troid);
for(int i=0;i<g[troid].size();i++)
{
int prox = g[troid][i];
if(isTroid[prox]) continue;
Decompose(prox,prof+1);
}
}
void Init(int N, int A[], int B[], int D[])
{
for(int i=0;i<n-1;i++)
{
g[ A[i] ].push_back( B[i] ); g[ B[i] ].push_back( A[i] );
p[ A[i] ].push_back( D[i] ); p[ B[i] ].push_back( D[i] );
}
Decompose(0,0);
for(int i=0;i<N;i++) sub[i] = 0;
return;
}
long long int Query(int S, int X[], int T, int Y[])
{
long long int resp = INF;
for(int i=0;i<S;i++)
{
int v = X[i];
for(int j=0;j<=troidProf[v];j++)
{
int troid = troidPai[j][v];
if(active[troid]) mnDist[troid] = min( mnDist[troid] , troidDist[j][v] );
else active[troid] = 1, mnDist[troid] = troidDist[j][v];
rList.push_back(troid);
}
}
for(int i=0;i<T;i++)
{
int v = Y[i];
for(int j=0;j<=troidProf[v];j++)
{
int troid = troidPai[j][v];
if(!active[troid]) continue;
resp = min( resp , mnDist[troid] + troidDist[j][v] );
}
}
for(int i=0;i<rList.size();i++) active[rList[i]] = 0;
rList.clear();
return resp;
}