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

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

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

 

 

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

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值