#ATarc140d. [ARC140D] One to One

[ARC140D] One to One

题目描述

对于所有元素都在 11NN 之间的长度为 NN 的整数序列 X=(X1,X2,,XN)X=(X_1,X_2,\dots,X_N),我们定义如下问题,并将其答案记为 f(X)f(X)

有一个包含 NN 个顶点的无向图 GG。(GG 可能包含重边和自环。)GGNN 条边,第 ii 条边连接顶点 ii 和顶点 XiX_i。请计算 GG 的连通分量个数。

给定一个长度为 NN 的整数序列 A=(A1,A2,,AN)A=(A_1,A_2,\dots,A_N),其中每个 AiA_i11NN 之间的整数,或者是 1-1

请考虑所有满足 Ai1Ai=XiA_i\neq -1 \Rightarrow A_i = X_i 的、所有元素都在 11NN 之间的长度为 NN 的整数序列 X=(X1,X2,,XN)X=(X_1,X_2,\dots,X_N)。对于所有这样的 XX,计算 f(X)f(X) 的总和,并对 998244353998244353 取模后输出。

输入格式

输入通过标准输入给出,格式如下:

NN A1A_1 A2A_2 \dots ANA_N

输出格式

输出答案。

样例 1

输入

3
-1 1 3

输出

5

样例 2

输入

1
1

输出

1

样例 3

输入

8
-1 3 -1 -1 8 -1 -1 -1

输出

433760

说明/提示

限制条件

  • 1N20001 \leq N \leq 2000
  • 每个 AiA_i11NN 之间的整数,或者是 1-1
  • 输入均为整数。

样例解释 1

满足条件的 XX 有以下 33 种情况。

  • X=(1,1,3)X=(1,1,3) 时,答案为 22
  • X=(2,1,3)X=(2,1,3) 时,答案为 22
  • X=(3,1,3)X=(3,1,3) 时,答案为 11

因此答案为 2+2+1=52+2+1=5

由 ChatGPT 4.1 翻译