TypeDB Forces 2023 (Div. 1 + Div. 2, Rated, Prizes!)
D
Game on Axis
Codeforces
1900
原题
← CE →
翻译中…

There are $n$ points $1,2,\ldots,n$, each point $i$ has a number $a_i$ on it. You're playing a game on them. Initially, you are at point $1$. When you are at point $i$, take following steps:

翻译中…
  • If $1\le i\le n$, go to $i+a_i$,
  • Otherwise, the game ends.
翻译中…

Before the game begins, you can choose two integers $x$ and $y$ satisfying $1\le x\le n$, $-n \le y \le n$ and replace $a_x$ with $y$ (set $a_x := y$). Find the number of distinct pairs $(x,y)$ such that the game that you start after making the change ends in a finite number of steps.

翻译中…

Notice that you do not have to satisfy $a_x\not=y$.

翻译中…

Input

翻译中…

Each test contains multiple test cases. The first line contains an integer $t$ ($1\le t\le 10^4)$ — the number of test cases.

翻译中…

The first line of each test case contains one integer $n$ ($1\le n\le 2\cdot 10^5$) — the number of points.

翻译中…

The second line contains $n$ integers $a_1,a_2,\ldots,a_n$ ($-n \le a_i \le n$) — the numbers on the axis.

翻译中…

It's guaranteed that the sum of $n$ does not exceed $2\cdot 10^5$.

翻译中…

Output

翻译中…

For each test case, print a line containing a single integer — the number of distinct pairs $(x,y)$ with which the game ends.

翻译中…

Note

翻译中…

In the first test case, the pairs $(x,y)$ with which the game ends are $(1,-1)$ and $(1,1)$, corresponding to the routes $1\rightarrow 0$ and $1\rightarrow 2$. Note that $(1,2)$ is invalid since when $n=1$, $y=2$ violates $-n\le y\le n$. $(1,0)$ is also invalid since you will go from $1$ to $1$ forever.

翻译中…

In the second test case, the pairs are $(1,-2),(1,-1),(1,2),(2,-2),(2,-1),(2,0),(2,1),(2,2)$.

翻译中…

In the fourth test case, the pairs are $(1,-2),(1,-1),(1,1),(1,2),(2,-2),(2,1),(2,2)$.

翻译中…

样例

Input #1
9
1
0
2
-1 0
2
1 -1
2
1 1
3
-1 -2 -1
3
1 -2 -1
4
-1 4 -2 1
5
1 1 1 1 -4
5
1 1 1 1 1
Output #1
2
8
6
7
20
17
34
30
40
InputCopy OutputCopy
题目信息
题号
1787D
时间限制
1 s
内存限制
256 MB
标签
组合数学
DFS 及类似
并查集
图论
模拟

题面与样例版权归 Codeforces 及出题人所有;中文翻译由 AI 生成(CI 管线),如有出入请以原文为准。