yukicoder No.523 LED
解法
LEDをポチポチする順番を順列にして考えてみると, 例えばN=3のとき, ①①②②③③の並べ方を数える問題になります. このとき, 同じ番号のものは区別せず, 場合の数はとなります. 一般のNに対して答えは次の式で表されることがわかります.
制約からはループを回しながら剰余をとっても間に合います. は, まず繰り返し二乗法を用いてとなるを求めた後, さらに繰り返し二乗法を用いてフェルマーの小定理
からの逆元を求め, を計算して答えです.
その他
実は答えの式は
より
と変形できるため, 一回のfor文で求めることもできます. スゴイ