4307: 水晶迷宫

时间限制:2000 ms 内存限制:256 MB
上传者:
提交:3 通过:2

题目描述

lbq 被困在一个具有 $n$ 个点的迷宫里。

初始时,lbq 将被传送到 $1$ 号点,每个点 $i$ 上有一个整数 $a_i$,当 lbq 位于点 $i$ 时,若 $1\le i+a_i\le n$,lbq 会被传送到点 $i+a_i$;反之,lbq 会逃出迷宫。

传送开始前,lbq 有一次使用魔法的机会,具体来说,lbq 可以选择一个点 $i$$(1\le i\le n)$ 并将 $a_i$ 改为任意整数 $j$$(-n\le j\le n)$。

lbq 想知道,有多少组 $(i,j)$ 可以使得 lbq 能够在有限步之内逃出迷宫。

输入格式

第 $1$ 行为一个整数 $t$,代表数据组数。

对于每组数据,第 $1$ 行有一个整数 $n$,代表迷宫中点的个数;第 $2$ 行有 $n$ 个整数 $a_i$,代表初始时点 $i$ 上的整数。

输出格式

每行一个整数,即每组数据 lbq 使用魔法的可行方案数。

输入样例 复制

3
3
1 1 1
4
-1 4 -2 1
5
1 2 1 2 -3

输出样例 复制

15
34
46

数据范围与提示

分类标签