赛时应该口胡了个大概,可惜没有转化成更纯粹的问题。
问题可以看做有多少不同的 \(x\) 满足 \(x=\sum_{i=1}^{m}(\frac{1}{k})^{a_i},1-x=\sum_{i=1}^{n}(\frac{1}{k})^{b_i}\)。
以上结论可以将平均数的过程建成一棵树,然后 \(x\) 代表权值为 \(1\) 的叶子结点,\(y\) 代表权值为 \(0\) 的叶子结点。
然后将 \(x,y\) 写成 \(k\) 进制小数,可以发现其中的规律。
于是答案变成 \([x^ny^m]\frac{\sum_{i=1}^{k-1}x^iy^{k-i}}{1-\sum_{i=1}^{k-2}x^iy^{k-i-1}}\)。
只需要求出 \(F(x,y)=\frac{1}{1-\sum_{i=0}^{k-1}x^iy^{k-i-1}}\) 即可,剩下的取个系数即可。
直接取系数,设 \(f[n][m]=[x^ny^m]F(x,y)\) 可以得到:
\[f[n][m]=\sum_{i=0}^{k-1}f[n-i][m-(k-1-i)] \]这个在斜方向上做前缀和即可,复杂度 \(O(nm)\)。
标签:iy,代表权,frac,AGC009E,sum,值为 From: https://www.cnblogs.com/lmpp/p/16655111.html