【树的直径】战略威慑

题意

求树上两条不相交的链乘积最大。

思路

枚举断开一条边，然后在两棵树上求直径更新答案。

代码

#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);
}