题意
有长为 2 n 2n 2n 的序列 A , B A,B A,B,求有多少个单调不减的序列 C C C,要求 C C C 的每一位均为 A A A 或 B B B 中对应位置上的数,且恰有 n n n 个数来自 A A A。 n ≤ 5 × 1 0 4 n\leq 5\times 10^4 n≤5×104。3s。
题解
设 f [ i ] [ j = 0 / 1 ] [ k ] f[i][j=0/1][k] f[i][j=0/1][k] 表示 DP 确定了前 i i i 个数,第 i i i 个数是 A i / B i A_i/B_i Ai/Bi ,已经有 k k k 个数来自 A A A 的方案数。直接 DP 是 O ( n 2 ) O(n^2) O(n2) 的。
考虑把 k k k 这一维写成形式幂级数的形式,则由 f [ i ] f[i] f[i] 转移到 f [ i + 1 ] f[i+1] f[i+1] 的过程可以看作乘上一个矩阵 [ 0 / x 0 / x 0 / 1 0 / 1 ] \begin{bmatrix}0/x&\quad&0/x\\0/1&&0/1\end{bmatrix} [