Atcoder Regular Contest 116(ARC116) A-E 题解

A

令 $n$ 含有 $k$ 个质因子 $2$ 。

①若 $k\ge 2$ ，则偶因子多；
②若 $k=1$ ，则两者相等；
③若 $k=0$ ，则奇因子多。

B

我们将 $a$ 从小到大排序，不难发现答案是
$$\sum _{i=1}^n\sum _{j=i+1}^n{2}^{j-i-1}{a}_i{a}_j+\sum _{i=1}^n{a}_i^2$$

对于前面的那一部分，分治计算就好了。

C

令 $a_m=x$ 的方案数为 $f_x$ 。

我们将 $x$ 质因数分解为 $x={\prod }_{i=1}^t{p}_i^{q_i}$ 的形式，不难发现
$$f_x=\prod _{i=1}^t\left(\genfrac{}{}{0ex}{}{n+q_i-1}{q_i}\right)$$

筛出每一个 $[1,m]$ 中的质数 $p$ ，然后向所有含有至少一个质因子的 $p$ 的数产生贡献，最后对于每一个 $x$ 求出 $f_x$ 即可。

时间复杂度 $O(m\log m)$ 。

D

答案为
$$[x^m]\sum _{i=0}^{{\log }_{2}m}\sum _{j=0}^n\left(\genfrac{}{}{0ex}{}{n}{2j}\right)[x^{2^i\times j}]$$

直接做是 $O(m^2\log m)$ 的，可以采用 NTT 优化到 $O(m{\log }^{2}m)$ 或分治 NTT 优化到 $O(m\log m\log \log m)\approx O(m\log m)$ 。

E

考虑二分答案。令当前二分的值为 $mid$ 。

为了方便叙述，令关键点分别为 $k_1,k_2,\cdots ,k_t$ ，节点 $i$ 的权值为 ${\min }_{1\le j\le k}dis(i,k_j)$ 。我们称一个点不合法，当且仅当该点的权值超过了 $mid$ 。

考虑求出至少需多少个关键点才能使得所有点权不超过 $mid$ 。

考虑贪心地去选择关键点。我们在树上做深搜，在回溯时做一个判断——若 $i$ 到子树内所有不合法点的最大距离为 $mid$ ，则在 $i$ 上放一个关键点。

现在，关键在于求出，点 $i$ 到子树内所有不合法点的最大距离。注意到这是一个单点修改，子树查询的问题，我们可以采用dfs序+线段树来维护。

但是这样的复杂度会多一个 $\log$，我们并不满意。考虑通过递推来代替线段树维护这个值。

令 $f_i$ 表示点 $i$ 到子树内所有关键点的最小距离。回溯的时候：

①若 $f_i\ge mid$ ，则点 $i$ 本身不合法，所以 $g_i:=\max (g_i,0)$
②若 $f_i+g_i\le mid$ ，则点 $i$ 的子树中所有点均合法，所以 $g_i:=-\infty$
③若 $g_i=mid$ ，就在点 $i$ 上摆一个关键点，令 $f_i=0$ 且 $g_i=-\infty$

这里 $g_i=-\infty$ 表示在 $i$ 的子树中不存在不合法点。

注意，最终还要判断一下 $g_{root}$ 是否为 $-\infty$ ，如果不是的话，关键点的数量还要加一。若最终关键点的数量不超过 $mid$ ，则答案一定不大于 $\text{mid}$ ，反之同理。

总时间复杂度 $O(n\log n)$ 。

F

咕咕咕