题意
求树上两条不相交的链乘积最大。
思路
枚举断开一条边,然后在两棵树上求直径更新答案。
代码
#include <cstdio>
#include <cstring>
#include <algorithm>
int n, ans, p1, p2, tot;
int head[201], ver[401], next[401], d[201], v[201];
void add(int u, int v) {
ver[++tot] = v;
next[tot] = head[u];
head[u] = tot;
}
void dp(int p) {
v[p] = 1;
for (int i = head[p]; i; i = next[i]) {
if (v[ver[i]]) continue;
dp(ver[i]);
p1 = std::max(p1, d[p] + d[ver[i]] + 1);
d[p] = std::max(d[p], d[ver[i]] + 1);
}
}
int main() {
scanf("%d", &n);
for (int i = 1, x, y; i < n; i++)
scanf("%d %d", &x, &y), add(x, y), add(y, x);
for (int i = 1; i <= tot; i += 2) {
memset(v, 0, sizeof(v));
memset(d, 0, sizeof(d));
p1 = p2 = 0;
v[ver[i]] = v[ver[i + 1]] = 1;
dp(ver[i]);
p2 = p1;
p1 = 0;
dp(ver[i + 1]);
ans = std::max(ans, p1 * p2);
}
printf("%d", ans);
}