[DFS] 树上删点使两连通块构成的点对最少 CFgym/101673F

最新推荐文章于 2020-05-17 16:57:56 发布

原创最新推荐文章于 2020-05-17 16:57:56 发布 · 333 阅读

0 ·

CC 4.0 BY-SA版权

搜索专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种高效的算法，用于解决在树形结构中寻找包含特定点对的最大数量的问题。通过两次深度优先搜索（DFS），算法首先确定了树的重心，并以此为基准进行第二次搜索，最终计算出最大点对数及其次优解。该算法适用于图论和竞赛编程中的复杂问题。

#include <bits/stdc++.h>
using namespace std;

const int mn = 10010;

int cnt;
int to[2 * mn], nx[2 * mn], fr[mn];
void add_edge(int u, int v)
{
	to[cnt] = v;
	nx[cnt] = fr[u];
	fr[u] = cnt++;
}

int n;
int pans, ans1;

bool vis[mn];
int dfs(int u)
{
	vis[u] = 1;
	int tol = 0;
	int sum = 0;
	for (int i = fr[u]; i != -1; i = nx[i])
	{
		if (!vis[to[i]])
		{
			int t = dfs(to[i]);
			tol += t;
			sum += t * (n - t);		// 子节点间点对数重复计算
		}
	}
	
	// tol 叶子连通块节点总数
	sum += tol * (n - tol);		// 子节点与其他节点点对数重复计算
	sum /= 2;
	
	if (ans1 < sum)
	{
		ans1 = sum;
		pans = u;
	}
	return tol + 1;
}

int dfs2(int u)
{
	vis[u] = 1;
	int tol = 0;
	for (int i = fr[u]; i != -1; i = nx[i])
	{
		if (!vis[to[i]])
			tol += dfs2(to[i]);
	}
	return tol + 1;
}

int main()
{
	memset(fr, -1, sizeof fr);
	
	scanf("%d", &n);
	for (int i = 0; i < n; i++)
	{
		int a, b;
		scanf("%d %d", &a, &b);
		add_edge(a, b);
		add_edge(b, a);
	}
	
	dfs(0);
	
	memset(vis, 0, sizeof vis);
	vis[pans] = 1;
	
	int ma1 = 0, ma2 = 0;
	for (int i = fr[pans]; i != -1; i = nx[i]) // 每个子节点连通块上的点数
	{
		int t = dfs2(to[i]);
		if (t > ma1)
		{
			ma2 = ma1;
			ma1 = t;
		}
		else if (t > ma2)
			ma2 = t;
	}
	
	printf("%d %d\n", ans1, ans1 - ma1 * ma2);
	
	return 0;
}