hdu 4705 Y (계수)
코드
#pragma comment(linker, "/STACK:16777216")
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
typedef unsigned long long ll;
const int maxn = 1e5 + 5;
ll ans;
int N, sz[maxn];
vector<int> G[maxn];
void init () {
for (int i = 1; i <= N; i++) G[i].clear();
int u, v;
for (int i = 1; i < N; i++) {
scanf("%d%d", &u, &v);
G[u].push_back(v);
G[v].push_back(u);
}
ans = 0;
}
void dfs (int u, int f) {
sz[u] = 0;
for (int i = 0; i < G[u].size(); i++) {
int v = G[u][i];
if (v == f) continue;
dfs(v, u);
sz[u] += sz[v];
}
int nd = sz[u], fz = N-1-sz[u];
ll sum = 1LL * fz * nd;
for (int i = 0; i < G[u].size(); i++) {
int v = G[u][i];
if (v == f) continue;
nd -= sz[v];
sum += 1LL * sz[v] * nd;
}
ans += 1LL * fz * (sum - 1LL * fz * (N-1-fz));
for (int i = 0; i < G[u].size(); i++) {
int v = G[u][i];
if (v == f) continue;
ans += 1LL * sz[v] * (sum - 1LL * sz[v] * (N-1-sz[v]));
}
sz[u]++;
}
int main () {
while (scanf("%d", &N) == 1) {
init();
dfs(1, 0);
printf("%I64u
", ans / 3);
}
return 0;
}
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
다양한 언어의 JSONJSON은 Javascript 표기법을 사용하여 데이터 구조를 레이아웃하는 데이터 형식입니다. 그러나 Javascript가 코드에서 이러한 구조를 나타낼 수 있는 유일한 언어는 아닙니다. 저는 일반적으로 '객체'{}...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.