問題一覧 > 通常問題

No.3698 くじ引きでチーム分け

レベル : / 実行時間制限 : 1ケース 2.000秒 / メモリ制限 : 512 MB / 標準ジャッジ問題
タグ : (AC するまで非表示) / 解いたユーザー数 20
作問者 : yuki2006
お気に入りにしたユーザー ProblemId : 8301 / 自分の提出
問題文最終更新日: 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もしくは右上の雲マークをクリックしてアカウントを作成してください。