CF1399-Codeforces Round #661 (Div. 3)-E1. Weights Division (easy version)

传送门：题目链接
题意：给定一棵边权树，树的总价值是根节点到所有叶子节点所经过的距离之和$\sum _{v\in leaves}w(root,v)\le S$，距离即根节点到叶子节点经过所有边的边权和。给定一个操作：选择任意一条边将其边权值除以二（向下取整）（即边权$w_i:=\lfloor \frac{w_i}{2}\rfloor$）。让我们求最少的操作次数，使得树的总价值不大于$k$。
思路：我们简单思考一下求树的总价值的过程，显然每一条边被加的次数是固定的，我们假设其为$time{s}_i$，那么$time{s}_i$的值是从根节点到所有叶子节点经过本条边的次数（也就是从根节点通过这条边可达的叶子数量），这一步我们可以通过$dfs$预处理得到（代码中的$dfs1$），然后把这些边的权值$w_i$和它对应的$time{s}_i$存于优先队列(代码中的$dfs2$)，优先队列的排序规则是按$w_i-\lfloor \frac{w_i}{2}\rfloor$从大到小排序。然后贪心的选择最大的直到小于$k$就行了。

AC代码：
#include<bits/stdc++.h>

using namespace std;
#define ll long long
#define pb push_back
#define rep(i, a, b) for(int i=a;i<=b;i++)
#define frep(i, a, b) for(int i=a;i>=b;i--)
const int N = 1E5 + 10;

ll n, k, vis[N], times[N], vis2[N];

struct edge {
    int val, to;
    edge(int t, int w) {
        val = w, to = t;
    }
};

struct cmp {
    bool operator()(pair<ll, ll> a, pair<ll, ll> b) {
        return (a.first-a.first/2) * a.second < (b.first-b.first/2) * b.second;
    }
};

vector<edge> G[N];
priority_queue<pair<ll, ll>, vector<pair<ll, ll> >, cmp> nums;
ll sum = 0;

void dfs(int x) {
    vis[x] = 1;
    if (G[x].size() == 1 && x > 1) {
        times[x] = 1;
        return;
    } else {
        for (auto i : G[x]) {
            if (!vis[i.to]) {
                dfs(i.to);
                vis[i.to] = 1;
                times[x] += times[i.to];
            }
        }
    }
    return;
}

void dfs2(int x) {
    vis2[x] = 1;
    if (G[x].size() == 1 && x > 1) return;
    else {
        for (auto e:G[x]) {
            if (!vis2[e.to]) {
                nums.push(make_pair(e.val, times[e.to]));
                sum += e.val * times[e.to];
                dfs2(e.to);
            }
        }
    }
    return;
}

int main() {
    //freopen("1.txt","r",stdin);
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int t;
    cin >> t;
    string str;
    while (t--) {
        cin >> n >> k;
        for (int i = 0; i <= n; i++) G[i].clear(), times[i] = 0, vis[i] = 0, vis2[i] = 0;
        sum = 0;
        while (!nums.empty()) nums.pop();
        for (int i = 0; i < n - 1; i++) {
            int u, v, w;
            cin >> u >> v >> w;
            G[u].emplace_back(v, w);
            G[v].emplace_back(u, w);
        }
        dfs(1);
        dfs2(1);
        ll cnt = 0;
        pair<ll, ll> mm;
        while (sum > k) {
            mm = nums.top();
            nums.pop();
            cnt++;            
            sum -= (mm.first - mm.first / 2) * mm.second;
            nums.push(make_pair(mm.first / 2, mm.second));
        }
        cout << cnt << endl;
    }
    return 0;
}

写在最后：这题还是写了很久，对于一些细节的处理非常不到位，导致卡了非常久，第一个错误是每次从总价值$sum$减掉$max(w_i)\ast time{s}_i$之后往优先队列里push的应该是$\lfloor \frac{w_i}{2}\rfloor$，我写成了$w_i-\lfloor \frac{w_i}{2}\rfloor$，这在$w_i$是奇数的时候显然不对；第二个错误是一开始我并没有将$w_i$和$time{s}_i$分开来存，而是简单的把他们的乘积存到优先队列，这样会由下式原因造成误差$$w_i\ast time{s}_i-\lfloor \frac{w_i\ast time{s}_i}{2}\rfloor \ne (w_i-\lfloor \frac{w_i}{2}\rfloor )\ast time{s}_i$$故大顶堆应同时维护每一条边的权重$w_i$和计算次数$time{s}_i$。