| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 672212 | lam | 경주 (Race) (IOI11_race) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "race.h"
#define ff first
#define ss second
#define int long long
using namespace std;
const int maxn = 2e5 + 10;
int n,k;
map<int,int> mp[maxn];
int id[maxn],d[maxn];
vector <pair<int,int>> adj[maxn];
int ans[maxn];
void dfs(int x, int p, int depth)
{
ans[x]=0LL;
mp[id[x]][depth]++;
for (int i=0; i<(int)adj[x].size(); i++)
{
int v = adj[x][i].ff; int w=adj[x][i].ss;
if (v==p) continue;
dfs(v,x,depth+w);
if (mp[id[i]].size()<mp[id[x]].size())
{
for (pair<int,int> j: mp[id[i]])
{
ans[x] += mp[id[x]][k - (j.ff - depth) +depth]*j.ss;
mp[id[x]][j.ff]+=j.ss;
}
}
else
{
for (pair<int,int> j: mp[id[x]])
{
ans[x] += mp[id[i]][k - (j.ff - depth) +depth]*j.ss;
mp[id[i]][j.ff]+=j.ss;
}
id[x]=id[i];
}
}
}
int best_path(int N, int K, int H[][2], int L[])
{
for (int i=1; i<=n; i++) id[i]=i;
for (int i=0; i<n-1; i++)
{
int u=H[i][0]; int v=H[i][1];
u++; v++;
adj[u].push_back({v,L[i]});
adj[v].push_back({u,L[i]});
}
dfs(1,1,0);
int res=0LL;
for (int i=1; i<=n; i++) res+=ans[i];
return res;
}