E. Special Elements Codeforces Round 640 (Div. 4)

最新推荐文章于 2024-09-26 09:13:29 发布

翰墨织梦

最新推荐文章于 2024-09-26 09:13:29 发布

阅读量578

点赞数 22

分类专栏： codeforces 文章标签： c++

本文链接：https://blog.csdn.net/halation_Z/article/details/136858231

版权

codeforces 专栏收录该内容

29 篇文章 0 订阅

订阅专栏

本文讨论了一个编程问题，要求在严格的时限内找出数组中可以表示为连续元素之和的特殊元素数量。建议使用编译型静态类型语言如C++，并指出在Python中应考虑使用PyPy。给出示例和算法实现，包括使用unordered_map优化计数过程。

摘要由CSDN通过智能技术生成

Pay attention to the non-standard memory limit in this problem.

In order to cut off efficient solutions from inefficient ones in this problem, the time limit is rather strict. Prefer to use compiled statically typed languages (e.g. C++). If you use Python, then submit solutions on PyPy. Try to write an efficient solution.

The array $$$a=[a_1, a_2, \ldots, a_n]$$$ ($$$1 \le a_i \le n$$$) is given. Its element $$$a_i$$$ is called special if there exists a pair of indices $$$l$$$ and $$$r$$$ ($$$1 \le l < r \le n$$$) such that $$$a_i = a_l + a_{l+1} + \ldots + a_r$$$. In other words, an element is called special if it can be represented as the sum of two or more consecutive elements of an array (no matter if they are special or not).

Print the number of special elements of the given array $$$a$$$.

For example, if $$$n=9$$$ and $$$a=[3,1,4,1,5,9,2,6,5]$$$, then the answer is $$$5$$$:

$$$a_3=4$$$ is a special element, since $$$a_3=4=a_1+a_2=3+1$$$;
$$$a_5=5$$$ is a special element, since $$$a_5=5=a_2+a_3=1+4$$$;
$$$a_6=9$$$ is a special element, since $$$a_6=9=a_1+a_2+a_3+a_4=3+1+4+1$$$;
$$$a_8=6$$$ is a special element, since $$$a_8=6=a_2+a_3+a_4=1+4+1$$$;
$$$a_9=5$$$ is a special element, since $$$a_9=5=a_2+a_3=1+4$$$.

Please note that some of the elements of the array $$$a$$$ may be equal — if several elements are equal and special, then all of them should be counted in the answer.

Input

The first line contains an integer $$$t$$$ ($$$1 \le t \le 1000$$$) — the number of test cases in the input. Then $$$t$$$ test cases follow.

Each test case is given in two lines. The first line contains an integer $$$n$$$ ($$$1 \le n \le 8000$$$) — the length of the array $$$a$$$. The second line contains integers $$$a_1, a_2, \dots, a_n$$$ ($$$1 \le a_i \le n$$$).

It is guaranteed that the sum of the values of $$$n$$$ for all test cases in the input does not exceed $$$8000$$$.

Output

Print $$$t$$$ numbers — the number of special elements for each of the given arrays.

Example

input

5
9
3 1 4 1 5 9 2 6 5
3
1 1 2
5
1 1 1 1 1
8
8 7 6 5 4 3 2 1
1
1

output

题目大意：

给n个数，特殊的数是其余至少两个位置连续的数的和，问有多少个特殊的数？

思路：

暴力双for，注意要用unordered_map去标记，mp会超时。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define endl "\n"

const ll N = 1e4+7;
ll n,m,k;
ll v[N];

void solve(){
	cin >> n;
	unordered_map<ll,ll>mp;
	for(ll i = 0 ; i < n ; i ++)
		cin >> v[i],mp[v[i]]++;
	ll sum=0;
	for(ll i = 0 ; i < n-1 ; i ++){
		ll cnt = v[i];
		for(ll j = i+1 ; j < n ; j ++){
			cnt+=v[j];
			if(cnt > n)break;
			sum += mp[cnt];
			mp[cnt]=0;
		}
	}
	cout << sum << endl;
	return;
}

int main(){
	ll t=1;cin >> t;
	while(t--)solve();
	return 0;
}