AT_arc187_b [ARC187B] Sum of CC
Description
[problemUrl]: https://atcoder.jp/contests/arc187/tasks/arc187_b
長さ $ N $ の数列 $ A=(A_1,\ldots,A_N) $ に対し,$ f(A) $ を以下で定義します.
- 頂点に $ 1 $ から $ N $ の番号がついた $ N $ 頂点 $ 0 $ 辺のグラフを用意する.$ 1\leq\ i\
Input Format
入力は以下の形式で標準入力から与えられる.
> $ N $ $ M $ $ B_1 $ $ \ldots $ $ B_N $
Output Format
答えを出力せよ.
Explanation/Hint
### 制約
- 入力される数値は全て整数
- $ 2\ \leq\ N\ \leq\ 2000 $
- $ 1\leq\ M\leq\ 2000 $
- $ B_i $ は $ -1 $ または $ 1 $ 以上 $ M $ 以下の整数
### Sample Explanation 1
$ B' $ として考えられるのは $ (2,1,1),(2,2,1),(2,3,1) $ の $ 3 $ 個です. $ B'=(2,1,1) $ のとき,頂点 $ 2 $ と頂点 $ 3 $ の間にのみ辺が張られるので,連結成分数は $ 2 $ です.よって $ f(B')=2 $ です. 同様に,$ B'=(2,2,1) $ のとき $ f(B')=2 $, $ B'=(2,3,1) $ のとき $ f(B')=2 $ なので答えは $ 2+2+2=6 $ となります.
### Sample Explanation 3
総和を $ 998244353 $ で割ったあまりを求めることに注意してください.