CF1748C Zero-Sum Prefixes

题目描述

一个数组 $v_1,v_2,\ldots,v_n$ 的得分定义为满足 $v_1+v_2+\ldots+v_i=0$ 的下标 $i$($1 \le i \le n$)的个数。 给定一个长度为 $n$ 的数组 $a_1,a_2,\ldots,a_n$。你可以多次进行如下操作: - 选择一个下标 $i$($1 \le i \le n$),使得 $a_i=0$; - 然后将 $a_i$ 替换为任意整数。 通过一系列这样的操作,数组 $a$ 能获得的最大得分是多少?

输入格式

每个测试点包含多个测试用例。第一行包含一个整数 $t$($1 \le t \le 10^4$),表示测试用例的数量。 每个测试用例的第一行包含一个整数 $n$($1 \le n \le 2 \cdot 10^5$),表示数组 $a$ 的长度。 每个测试用例的第二行包含 $n$ 个整数 $a_1,a_2,\ldots,a_n$($-10^9 \le a_i \le 10^9$),表示数组 $a$。 保证所有测试用例中 $n$ 的总和不超过 $2 \cdot 10^5$。

输出格式

对于每个测试用例,输出通过一系列操作后数组 $a$ 能获得的最大得分。

说明/提示

在第一个测试用例中,最优做法是将 $a_2$ 的值改为 $-2$。 此时数组 $a$ 变为 $[2,-2,1,-1,0]$,得分为 $3$: - $a_1+a_2=2-2=0$; - $a_1+a_2+a_3+a_4=2-2+1-1=0$; - $a_1+a_2+a_3+a_4+a_5=2-2+1-1+0=0$。 在第二个测试用例中,最优做法是将 $a_3$ 的值改为 $-2\,000\,000\,000$,此时数组得分为 $1$。 在第三个测试用例中,不需要进行任何操作。 由 ChatGPT 4.1 翻译