【图论】Linova and Kingdom—CF1336A

Linova and Kingdom---CF1336A
参考文章

思路

1 1 1 号节点为根节点。很容易想到,工业城市在树的下边,旅游城市在树的上边。具体来说,如果节点 u u u 是工业城市,那么它的子树的所有节点一定都是工业城市;如果节点 u u u 是旅游城市,那么它到根节点的路径上的所有城市都是旅游城市。

我们先把所有城市认定为工业城市,然后在与工业城市直接相连的旅游城市中选出"将其变为工业城市提供的贡献值"最大的城市,并将其变为工业城市。

城市的贡献的计算方法:当前点的子树的节点数 - (当前点的深度 - 1)。

可以利用用优先队列来选出最佳的城市,然后再将新产生的满足条件的城市压入队列。

由于我们选出的"贡献点最大的城市"的前提是这个点(设为 w w w 点)的子树上所有点都是工业城市,并且与根节点路径上的所有点都是旅游城市。那么如果后边的操作把 w w w 点的子节点给变成了旅游城市,那么不就不满足这个条件了吗?

请认真考虑我们是如何计算计算一个边界处的旅游城市如果变为工业城市后可以提供的贡献的:当前点的子树的节点数 - (当前点的深度 - 1)。 w w w 的贡献是相对于" w w w 点及 w w w 点到根节点路径上都是旅游城市的条件",也就是这种计算贡献的方法考虑了上一个状态,并非独立计算某一个点的贡献。

C o d e Code Code

cpp 复制代码
#include <bits/stdc++.h>
#define int long long
#define sz(a) ((int)a.size())
#define all(a) a.begin(), a.end()
using namespace std;
using PII = pair<int, int>;
using i128 = __int128;
const int N = 2e5 + 10;

int n, k;
int contri[N];

void solve() {
	cin >> n >> k;
	
	vector<vector<int>> g(n + 1);
	for (int i = 1; i <= n - 1; i ++) {
		int a, b; cin >> a >> b;
		g[a].push_back(b);
		g[b].push_back(a);
	}
	
	// 计算深度和子树中节点数量
	vector<int> deep(n + 1, -1), son_num(n + 1);
	auto dfs = [&] (auto dfs, int u) ->int {
		int sum = 1;
		for (auto v : g[u]) {
			if (deep[v] == -1) {
				deep[v] = deep[u] + 1;
				sum += dfs(dfs, v);
			}
		}
		son_num[u] = sum;
		return sum;
	};
	deep[1] = 1;
	dfs(dfs, 1);
	// 计算每个点的贡献
	for (int i = 1; i <= n; i ++) {
		son_num[i] --;
		contri[i] = son_num[i] - (deep[i] - 1);
	}
	
	int res = 0;
	k = n - k;
	vector<int> st(n + 1, 0); // 标记是否为旅游城市/已经来过
	
	priority_queue<PII, vector<PII>> q; // [贡献,节点] 降序优先队列
	q.emplace(contri[1], 1);
	while (not q.empty() && k --) {
		auto [f, s] = q.top(); q.pop();
		st[s] = 1;
		res += f;
		for (auto v : g[s]) {
			if (not st[v]) {
				q.emplace(contri[v], v);
			}
		}
	}
	
	cout << "         ";
	cout << res << "\n";
}

signed main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int T = 1;
//	cin >> T; cin.get();
	while (T --) solve();
	return 0;
}
相关推荐
吗~喽12 小时前
【LeetCode】四数之和
算法·leetcode·职场和发展
Net_Walke12 小时前
【散列函数】哈希函数简介
算法·哈希算法
卿言卿语12 小时前
CC1-二叉树的最小深度
java·数据结构·算法·leetcode·职场和发展
码流之上12 小时前
【一看就会一写就废 指间算法】执行操作后的最大 MEX —— 同余、哈希表
算法·面试
仰泳的熊猫13 小时前
LeetCode:889. 根据前序和后序遍历构造二叉树
数据结构·c++·算法
2025年一定要上岸13 小时前
【日常学习】10-15 学习re
学习·算法·正则表达式
aramae13 小时前
数据结构与算法(递归)
开发语言·经验分享·笔记·算法
小欣加油13 小时前
leetcode 329 矩阵中的最长递增路径
c++·算法·leetcode·矩阵·深度优先·剪枝
Emilia486.13 小时前
【Leetcode&nowcode&数据结构】单链表的应用(初阶)
c语言·数据结构·算法·leetcode
JAVA学习通14 小时前
JDK高版本特性总结与ZGC实践
java·jvm·算法