D. Paths on the Tree

题目链接：https://codeforces.com/contest/1746/problem/D

题意：给定一颗n个节点的树，根节点是1，每个节点有一个权值s[i]，给定一个整数k，定义k条简单路径集合有效为任意集合中两个父节点相同的点，被覆盖次数之差小于等于1.

求最大值，s[i]为点权值，c[i]为i节点被覆盖次数。

分析：

假设一个节点被覆盖k次，设该节点有size个儿子，为了满足题目条件且尽量增大s[i] * c[i]，显然

每个儿子至少经过k / size次，多出来的k % size次，分在多经过一次增量最多的前 k % size个节点上。那么我们就可以进行树形dp了，设f[u][0]表示没多走一次, 即经过k次，f[u][1]表示多经过一次， 即k + 1次， 不妨设其儿子有sum个。

则f[u][0] += (k % sum 个 f[v][1] + 剩下的 f[v][0]), f[u][1] += (k % sum + 1 个 f[v][1] + 剩下的 f[v][0]), 

代码:
 

# include <bits/stdc++.h>
#define ios ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
#define x first
#define y second
#define endl '\n'
#define int long long
using namespace std;
typedef pair<int, int> PII;
const int N = 1e6 + 10;
const int M = 2e5 + 10;
const int INF = 0x3f3f3f3f;
const double INFF = 0x7f7f7f7f7f7f7f7f;
const int mod = 1e9 + 7;
int h[N], e[N], ne[N], idx, t, n, k, a[N];
int f[N][2], se[N];
void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

bool cmp(PII a, PII b)
{
    return a.x > b.x;
}

void dfs(int u, int v)
{
    f[u][0] = v * a[u];
    f[u][1] = v * a[u] + a[u];
    int res = 0;//儿子数量
    for(int i = h[u];i != -1;i = ne[i])
    {
        int j = e[i];
        res ++;
    }
    vector<PII>val1, val2;
    for(int i = h[u];i != -1;i = ne[i])
    {
        int j = e[i];
        dfs(j, v / res);
        val1.push_back({f[j][1] - f[j][0], j});
    }
    if(res == 0)return;
    int m = v / res;
    int mm = v % res;
    sort(val1.begin(), val1.end(), cmp);//按增量排序
     for(int i = 0;i < val1.size();i ++)
        {
            if(i < mm)
            f[u][0] += f[val1[i].y][1];
            else f[u][0] += f[val1[i].y][0]; 
        }
        for(int i = 0;i < val1.size();i ++)
        {
            if(i <= mm)
            f[u][1] += f[val1[i].y][1];
            else f[u][1] += f[val1[i].y][0]; 
        }
    
}

signed main()
{
    ios;
    cin >> t;
    while(t --)
    {
        idx = 0;
        cin >> n >> k;
        for(int i = 0;i <= n + 1;i ++)f[i][0] = f[i][1] = 0, se[i] = 0;
        for(int i = 0;i <= n;i ++)h[i] = -1;
        for(int i = 2;i <= n;i ++)
        {
            int p;cin >> p;
            add(p, i);
        }
        for(int i = 1;i <= n;i ++)cin >> a[i];
        dfs(1, k);
        cout << f[1][0] << endl;
    }
    return 0;
}