【图论】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;
}
相关推荐
向阳@向远方1 小时前
第二章 简单程序设计
开发语言·c++·算法
github_czy2 小时前
RRF (Reciprocal Rank Fusion) 排序算法详解
算法·排序算法
许愿与你永世安宁3 小时前
力扣343 整数拆分
数据结构·算法·leetcode
爱coding的橙子3 小时前
每日算法刷题Day42 7.5:leetcode前缀和3道题,用时2h
算法·leetcode·职场和发展
满分观察网友z3 小时前
从一次手滑,我洞悉了用户输入的所有可能性(3330. 找到初始输入字符串 I)
算法
YuTaoShao4 小时前
【LeetCode 热题 100】73. 矩阵置零——(解法二)空间复杂度 O(1)
java·算法·leetcode·矩阵
Heartoxx4 小时前
c语言-指针(数组)练习2
c语言·数据结构·算法
大熊背4 小时前
图像处理专业书籍以及网络资源总结
人工智能·算法·microsoft
满分观察网友z4 小时前
别怕树!一层一层剥开它的心:用BFS/DFS优雅计算层平均值(637. 二叉树的层平均值)
算法
杰克尼5 小时前
1. 两数之和 (leetcode)
数据结构·算法·leetcode