CF623E Transforming Sequence

一、题目

点此看题

二、解法

首先有一个显然的$dp$，设$dp[i][j]$为前$i$轮选$j$个的方案数：
$$dp[i][k]=\sum dp[i-1][j]\times C(n-j,j-k)\times 2^j$$发现这样很不好优化，我们设$f[i][j]=\frac{dp[i][j]}{C(n,j)}$，也就是先钦定这些要选的东西，那么转移：
$$f[i][k]=\sum f[i-1][j]\times C(k,j)\times 2^j$$发现两个$dp$可以合并：
$$f[n+m][i]=\sum f[n][j]\times f[m][i-j]\times C(i,j)\times 2^{mj}$$主要解释一下后面的$2^{mj}$，其实合并就相当于分部选，我们要选$m$次，每次都可以在大小为$k$的子集中随便选，还有：
$$f[2n][i]=\sum f[n][j]\times f[n][i-j]\times C(i,j)\times 2^{nj}$$还可以用$\text{ntt}$优化，化成这样的形式：
$$f[n+m][i]=i!\sum \frac{2^{mj}f[n][j]}{j!}\times \frac{f[m][i-j]}{(i-j)!}$$那么时间复杂度$O(n\log n\log n)$

咕咕咕