ARC212 C - ABS Ball
AtCoder Regular Contest 212 - ARC212 C - ABS Ball の解法です. shobonvip さんによるユーザ解説 そのままみたいなものですが, 式の導出のところを多めに書きました. 問題概要 $N$ 個の区別できない白いボールと,$M$ 個の区別できる箱がある. 「各ボールを赤・青どちらかの色に塗っていずれかの箱に入れる」方法それぞれに対して, $i$ 番目の箱に入った赤,青のボールの数を $a_i$, $b_i$ として $\prod_{1 \leq i \leq M} | a_i - b_i |$ を計算した場合の総和を mod 998244353 で求めよ. 制約: $1 \leq N \leq 10^7$, $1 \leq M \leq 10^7$ 問題へのリンク 解 $a + b = n$ である非負整数 $(a, b)$ についての $|b - a|$ の和を $c_n$ と書くことにすると, 冪級数 $f = \sum_n c_n x^n$ を使って,求める答が $[x^N] f^M$ と表せることに注意する....