题目链接: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;
}