[luoguP3565] [POI2014]HOT-Hotels（dfs）

传送门

三点在树上距离相等的情况只有一种，就是以某一个点为中心，三个点到这个点的距离相等。

所以直接枚举每个点作为中心，dfs这个中心的子树，根据乘法原理统计答案即可。

时间复杂度 O(n²) (n <= 5000)

代码

#include <cstdio> #include <cstring> #include <iostream> #define N 5001 #define LL long long LL ans; int n, cnt; int head[N], to[N << 1], next[N << 1], one[N], two[N], dis[N], tmp[N]; inline int read() inline void add(int x, int y) inline void dfs(int u) } } int main() for(i = 1; i <= n; i++) } } printf("%lld\n", ans); return 0; }

　　

上一篇：[luoguP1076] 寻宝（模拟）

下一篇：[luoguP2626] 斐波那契数列（升级版）（模拟）

---

dfs