[Luogu P3571] [BZOJ 3835] [POI2014]SUP-Supercomputer

最新推荐文章于 2022-09-12 17:42:32 发布

LPA20020220

最新推荐文章于 2022-09-12 17:42:32 发布

阅读量252

点赞数

分类专栏：动态规划斜率优化文章标签：动态规划斜率优化

本文链接：https://blog.csdn.net/lpa20020220/article/details/88304141

版权

动态规划同时被 2 个专栏收录

126 篇文章 1 订阅

订阅专栏

斜率优化

7 篇文章 0 订阅

订阅专栏

洛谷传送门

BZOJ传送门

题目描述

给定一棵 $N$ 个节点的有根树，根节点为 $1$ 。 $Q$ 次询问，每次给定一个 $K$ ，用最少的操作次数遍历完整棵树，输出最少操作次数。每次操作可以选择访问不超过 $K$ 个未访问的点，且这些点的父亲必须在之前被访问过。

输入输出格式

输入格式

第一行两个正整数 $N, Q$ 。

第二行 $Q$ 个正整数，分别表示每组询问中的 $K$ 。

第三行 $N - 1$ 个正整数，第 $i$ 个正整数表示 $i + 1$ 号节点的父节点。

输出格式

输出一行 $Q$ 个正整数，表示对于每组询问最小的操作次数。

输入输出样例

输入样例#1：

20 1
3
1 1 1 3 4 3 2 8 6 9 10 12 12 13 14 11 11 11 11

输出样例#1：

解题分析

实在太菜了，想不到 $d p$ 方程…

这个方程是这样的：

设 $d p [i]$ 表示每次最多选 $i$ 个节点， $s u m [i]$ 表示深度 $\ge i$ 的节点总数，就有：
$dp[i]=max(j+\lceil\frac{sum[j+1]}{i}\rceil)$
大概意思就是前 $j$ 次取完前 $j$ 层，后面每次都保证取满 $i$ 个节点。

为什么要取 $m a x$ ？因为有不合法的情况。

前 $j$ 次取不完，那么就有 $k < j$ 满足前 $k$ 次是取完的，容易证明此时不会比我们假设的不合法的 $j$ 次取完 $j $ 层更优。
后面取不满，那么就有 $k > j $ 满足后面是取完的，同样可得此时不会比这种不合法的情况更优。

然后如果有一个 $k > j$ 满足
$j+\lceil\frac{sum[j+1]}{i}\rceil<k+\lceil\frac{sum[k+1]}{i}\rceil$
就有
$i j + s u m [j + 1] < i k + s u m [k + 1]$
移项可得
$\\ i>\frac{sum[j+1]-sum[k+1]}{k-j}$
很简单的斜率优化。

注意这里所有的 $j, k$ 的取值范围都是 $[1, m a x d e p]$ ，所以先插入完，最后挨个查询即可。

代码如下：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cctype>
#include <cstdlib>
#include <vector>
#include <algorithm>
#define R register
#define IN inline
#define W while
#define gc getchar()
#define db double
#define MX 1000050
#define ll long long
template <class T>
IN void in(T &x)
{
	x = 0; R char c = gc;
	for (; !isdigit(c); c = gc);
	for (;  isdigit(c); c = gc)
	x = (x << 1) + (x << 3) + c - 48;
}
int n, q, head = 0, tail = -1, dpmx, qmx;
std::vector <int> nex[MX];
int s[MX], dp[MX], que[MX], dep[MX], ask[MX];
IN db slope(R int ini, R int tar) {return 1.0 * (s[ini + 1] - s[tar + 1]) / (tar - ini);}
IN void insert(R int id)
{
	W (head < tail && slope(que[tail - 1], que[tail]) >= slope(que[tail], id)) --tail;
	que[++tail] = id;
}
void DFS(R int now)
{
	dpmx = std::max(dpmx, dep[now]);
	++s[dep[now]];
	for (auto i : nex[now])
	{
		dep[i] = dep[now] + 1;
		DFS(i);
	}
}
int main(void)
{
	in(n), in(q); int foo;
	for (R int i = 1; i <= q; ++i)
	in(ask[i]), qmx = std::max(qmx, ask[i]);
	for (R int i = 2; i <= n; ++i)
	in(foo), nex[foo].push_back(i);
	dep[1] = 1; DFS(1);
	for (R int i = dpmx - 1; ~i; --i) s[i] += s[i + 1];
	for (R int i = 0; i < dpmx; ++i) insert(i);
	for (R int i = 1; i <= qmx; ++i)
	{
		W (head < tail && slope(que[head], que[head + 1]) < i) ++head;
		dp[i] = que[head] + (s[que[head] + 1] + i - 1) / i;
	}
	for (R int i = 1; i <= q; ++i) printf("%d ", dp[ask[i]]);
}