Codeforces Round #635 (Div. 2)补题
C
题目
题目大意
有n个城市,分为两类,工业城市和旅游城市
每年每一个工业城市派出一个人去1号城市
途径的快乐值是从这个工业城市路过的旅游城市的个数
问你快乐值最大是多少
思路
我们发现,如果一条路只有一个工业城市,那么这个工业城市的快乐值就是他的深度
如果有其他的工业城市,那么我们考虑如何计算一个工业城市的快乐贡献
如图,叶子节点都被选为工业城市,还需要再选一个工业城市,我们应该选择1还是2。
很容易发现,如果我们选择1,那么1号点的子树节点就会失去2点快乐值,如果选择2,那么2号点的子树节点会损失3快乐值
所以我们得出一个结论,一个节点的快乐值和他的深度和子树大小有关
dep[i]表示i的深度,s[i]表示i的子树的大小
那么我们依据dep[i] - s[i]进行排序,选取贡献值大的作为工业城市即可
代码
#include <iostream>
#include <cstdio>
#include <set>
#include <list>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <string>
#include <sstream>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <cctype>
#include <cmath>
#include <fstream>
#include <iomanip>
//#include <unordered_map>
using namespace std;
#define dbg(x) cerr << #x " = " << x << endl;
typedef long long ll;
typedef pair<ll, ll> P;
const int MAXN = 2e5 + 5;
vector<ll> v[MAXN];
int dis[MAXN];
bool vis[MAXN];
int n, k;
int cnt[MAXN];
int s[MAXN];
int dfs(int st, ll dep)
{
dis[st] = dep;
vis[st] = 1;
ll res = 0;
//dbg(st);
for(int i = 0; i < v[st].size(); i++)
{
int next = v[st][i];
if(!vis[next])
{
res++;
res += dfs(next, dep+1);
}
}
s[st] = res;
return res;
}
bool cmp(int a, int b)
{
return dis[a] - s[a] > dis[b] - s[b];
}
int main()
{
//freopen("in.txt", "r", stdin);
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
ll ans = 0;
cin >> n >> k;
for(int i = 0; i < n-1; i++)
{
int u, vv;
cin >> u >> vv;
v[u].push_back(vv);
v[vv].push_back(u);
}
s[1] = dfs(1, 0);
int tmp[MAXN];
/*for(int i = 1; i <= n; i++)
{
dbg(i);
cout << "DEP = " << dis[i] << " SIZE = " << s[i] << endl;
}*/
for(int i = 0; i < n; i++)
{
tmp[i] = i + 1;
}
sort(tmp, tmp+n, cmp);
/* for(int i = 0; i < k; i++)
{
cout << tmp[i] << ' ';
cout << "CHOISE" << endl;
}
cout << endl;*/
for(int i = 0; i < k; i++)
{
ans += dis[tmp[i]] - s[tmp[i]];
}
cout << ans << endl;
}