【题解】2018.03.26测试 - FJWC2018正式赛

Source：

FJWC2018正式赛

Problem

T1 三元组

题意

求多少个三元组 $(a, b, c)$ 满足 $a + b^2 \equiv c^3(\bmod k)$， $T$ 组数据。

数据范围

$1 \leq n,k \leq 10^5, 1 \leq T \leq 400$

题解

30分解法

枚举 $a,b,c$ ，判断是否满足式子。

复杂度 $O(T\times n^3)$ 。

60分解法

先不考虑 $\bmod k$ ，则等式化为 $a+b^2=c^3$ 。

固定 $b$ ，根据 $1\leq a \leq b$ ，可以得到 $c$ 的取值范围是 $[b^2+1,b^2+b]$ 。

固定 $c$ ，枚举 $b$ ，统计 $c$ 落在 $[b^2+1,b^2+b]$ 的个数即三元组个数。

对于 $\bmod k$ 的情况处理方法类似。

首先考虑 $[b^2+1,b^2+b] \bmod k$ 之后会跨越 $[0, k - 1]$ 多次，其次再分别处理剩下的不完整区间。

跨越 $\lfloor\frac{i}{k}\rfloor$ 个完整区间 $[0, k - 1]$ 后，剩余的不完全区间有以下两种情况：

第一种情况，即$0 \leq (b^2+1)\bmod k \leq (b^2+b)\bmod k \leq k - 1$ ，即剩下相交的部分；

第二种情况，即$0\leq (b^2+b)\bmod k \leq (b^2+1)\bmod k \leq k - 1$，即剩下两边的部分。

那么对于第一种情况，处理 $[(b^2+1)\bmod k,(b^2+b)\bmod k]$；

对于第二种情况，分别处理 $[0, (b^2+b)\bmod k]$ 和 $[(b^2+1)\bmod k, k -1]$ 。

从 $1$ 到 $n$ 枚举 $b$ ，每次对区间 $[i^2+1,i^2+i]\bmod k$ 加一，累加$i^3$ 的值即最后的三元组数量。

复杂度$O(T\times n^2)$ 。

100分解法

根据数据范围，考虑 $O(n\log{n})$ 的数据结构维护区间。

想到树状数组/线段树。

由于是区间加法+单点查询，因此选择树状数组进行维护。

复杂度 $O(T\times n\log{n})$ 。

T2 攻略

题意

一棵树有 $n$ 个结点，每个结点有权值 $W_i$。现有 $m$ 条从根到叶子结点的路径，求它们路径结点的最大权值和（每个点权值只计算一次）。

数据范围

$1\leq n\leq 200000$，$1\leq W_i\leq 2^{31}-1$

题解

40分解法

树上背包，$F_{i,j}$ 表示结点 $i$ 为根的子树，$j$ 条路径的最大权值和。

则$F_{i,j}=\max\{F_{i,k}+F_{v,j-k}\} + W_i$ ，其中 $v$ 为 $i$ 的儿子，$0\leq k\leq j\leq \min\{size_i,m\}$ 。

复杂度 $O(n\times m)$ 。

100分解法

考虑贪心，每次选择当前树上权值和最大的那条链作为路径，然后将该链上的所有结点权值清零，往复 $m$ 次。

对原树进行深度优先遍历时，按照访问顺序对所有叶子节点进行编号，如此易得原树中任意一棵子树的叶子节点编号是连续的。

那么在深度优先遍历时记录每个节点访问时的叶子结点个数和访问结束（回溯）前的叶子结点个数（即最新编号），就可以得到它的子树中叶子节点的编号范围 $[L_i, R_i]$ 。

使用线段树维护路径权值和的最大值及对应叶子节点编号，每次取根节点记录的最大值，并通过记录的叶子节点编号和父节点编号来一步步更新线段树的维护的权值和。

预处理复杂度 $O(n)$，查询复杂度 $O(1)$，更新复杂度 $O(\log{n})$，总复杂度$O(m\times \log{n})$ 。

T3 迂回

题意

给定一个有向图，求图中长度小于 $k$ 的环个数（$\bmod m$）。

图中没有自环，旋转后相同的环也算作不同的环（即环中点的排列不完全相同即可）。

数据范围

$1\leq n\leq 100, k \leq 10^ 6, m\leq 10^9$

题解

30分解法

计算每个点在第 $1$ 到第 $k-1$ 步到每个点的方案数。所有自环（回到自己）的方案数的之和，即为总环数。

复杂度 $O(n^ 3 \times k)$ 。

100分解法

图 $G=\{a,b,c\}$ 中， $num(a,b)\times num(b,c)=num(a,c)$ ，$num(a,b)$ 即 $a$ 到 $b$ 的路径数量。

我们可以发现 $num(a, c) = \sum{num(a,b) \times num(b,c)}$ ，这和矩阵乘法完全相同。

考虑矩阵乘法，记 $A^{k}_{i,i}$ 为原始矩阵 $A$ 连乘 $k$ 次后 $(i,i)$ 的值，则$\sum_{i=1}^{k-1}\sum_{j=1}^{n}A^{i}_{j,j}$ 为总方案数。

考虑分治计算上式，记 $P(x)$ 为$\sum_{i = 1}^{x}{\prod_{1}^{i}{A}}$ ，则

若 $x$ 为偶数，则 $P(x) = P(x/2) + A^{x/2}\times P(x/2)$ ；

否则 $P(x) = P(x/2) + A^{x}\times P(x/2)+A^{x}$。

那么上式可以在 $O(n^3\times \log{k})$ 的时间内解出。

总复杂度 $O(n^3\times \log{k})$ 。

总结

考场状态及得分情况

考场上T1直接 $O(n^3)$ 暴力，T2 $O(n\times m)$ 树形dp，T3 $O(8\times n^3 \times \log{k})$构建 n n <script type="math/tex" id="MathJax-Element-99">n</script> 个虚拟点跑矩阵快速幂。

得分30+40+100=170，rk2。

问题

T1和T2都没有想得很清楚。

T1写完暴力之后就没有再思考，止步复杂度最高的暴力。

T2的贪心想到了但是没有去尝试，保守地写了个能拿到40分的树形dp。

T3常数有点大，好在最后A了，但是花的时间有点太长。

总的来说还可以，靠着T3把T1没拿到的一些分补上了。

题面

FJWC_Day5