Codeforces Round #635 (Div. 2)补题

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