No.3698 くじ引きでチーム分け
問題文最終更新日: 2026-09-09 23:34:12
チーム機能テストコンテスト
(順位表)
の他の問題:
問題文
$N$ 人の参加者を、無作為に $K$ 個のチームへ分けます。 $N$ は $K$ の倍数で、どのチームもちょうど $N/K$ 人になります。
そのような分け方は同様に確からしく選ばれます。 参加者 $i$ には $2$ つの整数 $A_i, B_i$ が定まっていて、参加者 $i$ と $j$ が同じチームになったとき $A_i B_j + A_j B_i$ の仲良し度を得ます。
得られる仲良し度の合計の期待値を $998244353$ で割った余りで求めてください。
入力
$N\ K$ $A_1\ A_2\ \cdots\ A_N$ $B_1\ B_2\ \cdots\ B_N$
$2 \le N \le 2 \times 10^5$
$1 \le K \le N$ かつ $K$ は $N$ を割り切る
$0 \le A_i, B_i < 998244353$
入力はすべて整数
出力
期待値を $998244353$ で割った余りを出力してください。求める値は必ず有理数になり、その分母は $998244353$ と互いに素です。 最後に改行してください。
サンプル
サンプル1
入力
4 2 1 2 3 4 1 1 1 1
出力
10
どのペアも同じチームになる確率は $(2-1)/(4-1) = 1/3$ です。全ペアの仲良し度の合計は $30$ なので、期待値は $10$ です。
サンプル2
入力
2 1 5 7 2 3
出力
29
$K = 1$ なので $2$ 人は必ず同じチームです。$5 \times 3 + 7 \times 2 = 29$ を必ず得ます。
提出するには、Twitter 、GitHub、 Googleもしくは右上の雲マークをクリックしてアカウントを作成してください。