多校A层冲刺NOIP2024模拟赛18

T1 选彩笔（rgb）

签到题，但是没签上。。.

没想到三维前缀和，直接上了个bitset.

就是直接二分答案，然后枚举这三维每维的区间的起点，前缀和查数量是否 大于等于 $ K $ 即可，也可以把二分答案改为第一维的双指针，少一个 $ log $ .

T2 兵蚁排序（sort）

比T1还签到，但是没写 freopen 挂 100.

从 $ b $ 数组里挨个去在 $ a $ 数组里匹配，然后 $ O(n) $ 拉过来就行.

T3 人口局 DBA（dba）

数学推式子.

但是有很多部分分的数位DP，骗了 97 分.

简单来说就是解一个方程满足 $ x_1 + x_2 + x_3 + ······ + x_{n-1} + x_{n} = sum , x1 \lt p , \forall \ \ i ∈ [1,n] \ \ 0 \le x_i \lt m $ 的解的个数.

首先不考虑 $ x1 \lt p $ 的限制，然后你就可以容斥，钦定有 $ k $ 个 $ x $ 是 大于 $ m $ 的，把他们斥掉，然后就有一个式子（直接插板）:

如果考虑上 $ x1 \lt p $ 的限制，那就可以直接枚举 $ x1 $ ，所以有:

那个后面求和式的化简就是因为 组合数的下面那个不变，所以在杨辉三角上就是竖着的一列 。

如图:

T4 银行的源起（banking）

因为讲过了，所以这篇讲的简单点。 算了，还是稍微详细点吧.

首先考虑只有一个银行的时候怎么求，那肯定是所有人都往一个银行跑，所以枚举点然后算贡献?

其实这种题有 $ O (n) $ 做法.

首先以 1 为根，我们直接考虑每条边的贡献，对于一条边，他的权值是 $ w $ ，连接着点 $ j $ 和他的父亲 $ fa_j $ .

对于这条边，要么是 $ j $ 和他的子树里面的点往上走经过这条边，要么是上面的点下来，这时候我们设 $ sz_j $ 是 $ j $ 及其子树里面的点的权值之和， $ tot $ 表示整棵树的权值和，直接取 $ w \times \min( sz_j , tot - sz_j) $ ， 作为这条边的贡献即可.

为什么对，来证明一下:

上面的式子只能保证对于一条边来说是最优的，但是不能保证全局都是这样的，但其实全局最优可以被证明只有在每个都最优的时候才成立.

证毕。此时我们得到了 $ O(n) $ 解决一个银行的问题的方法。那如果有两个银行呢，这个时候也一定存在一条边没有人走，所以我们枚举一条边并把它断开，然后对于两棵树分别求贡献加和，即可 $ O(n^2) $ 解决.

$ O(n^2) $ 解决不了，那就看看能不能上数据结构优化.

优化肯定要从贡献方面入手，那就再看看那个贡献的式子， $ ans_j = w_j \times \min( sz_j , tot - sz_j ) $ ，这个 min 就很难直接优化，那就分讨， 如果 $ sz_j \le \frac{tot}{2} , ans_j = w_j \times sz_j $ ，否则 $ ans_j = w_j \times (tot - sz_j ) $ ，那么这东西可以上值域线段树，具体来说，对于 $ sz $ 开一颗值域线段树，然后维护好 $ \sum w_j 、 \sum w_j \times sz_j $ 然后就可以做了.

主体思路有了，具体的分析一下:

在这个图里面，我们将这棵树分成了两部分，一颗红的，一颗绿的，我们分别求贡献.

对于绿色的部分很好求，就用刚才的至于线段树就行，dfs时线段树合并 $ n( log (n) ) $ ，对于红色的部分稍微麻烦一点.

首先考虑红色的部分相对于原树有哪些变化:

红树的 $ tot = sz_1 - sz_j $ 。

对于绿树中的根 $ j $ ，从他的父亲 $ x $ 到 1 的这条链上面的点 $ i $ ， $ sz_i $ 都减少了 $ sz_j $ .

首先链上的点的贡献肯定得单求，可以维护一颗树状数组，然后每次查询时先链减 ，然后问，但是一个技巧可以不用链减，现在的分讨条件是 $ sz_i - sz_j \le \frac{tot}{2} $ ，那么直接移项，分讨条件变为 $ sz_i \le \frac{tot}{2} + sz_j $ ，同时最后求的答案要用 减去 $ sz_j $ 之后的值。（写出式子来就会了） 。

接下来就是红树中除了链上的部分了，我们发现这部分很难直接求，线段树也不好维护，但是他就是全局除了链上和绿树上以外的所有点，所以我们求出全局对于 $ \frac{tot}{2} $ 的贡献，这部分可以弄一颗静态的全局线段树，也可以直接离散化后前缀和 ，然后减去 链上以及绿树中对于 $ \frac{tot}{2} $ 的贡献即可，这部分都可以用刚才维护的东西直接求.

那我们就求完了，这道题主要是多注意注意细节就行了.