CRAN02 - Roommate Agreement

Leonard was always sickened by how Sheldon considered himself better than him. To decide once and for all who is better among them they decided to ask each other a puzzle. Sheldon pointed out that according to Roommate Agreement Sheldon will ask first. Leonard seeing an opportunity decided that the winner will get to rewrite the Roommate Agreement.

Sheldon thought for a moment then agreed to the terms thinking that Leonard will never be able to answer right. For Leonard, Sheldon thought of a puzzle which is as follows. He gave Leonard n numbers, which can be both positive and negative. Leonard had to find the number of continuous sequence of numbers such that their sum is zero.

For example if the sequence is- 5, 2, -2, 5, -5, 9

There are 3 such sequences

2, -2

5, -5

2, -2, 5, -5

Since this is a golden opportunity for Leonard to rewrite the Roommate Agreement and get rid of Sheldon's ridiculous clauses, he can't afford to lose. So he turns to you for help. Don't let him down.


First line contains T - number of test cases

Second line contains n - the number of elements in a particular test case.

Next line contain n elements, aiĀ (1 <= i <= n) separated by spaces.


The number of such sequences whose sum if zero.


1 <= t <= 5

1 <= n <= 10^6

-10 <= ai <= 10


0 1 -1 0
5 2 -2 5 -5 9

hide comments
prakash: 2016-12-31 17:50:07

good problem

Shubham Gupta: 2016-06-23 17:01:55

Be sure to declare variables in long/ long long. [final answer and intermediate variables. ] Costed me 3 WAs.
P.s. took 1.25s, anyone who got it in less ?

=(Francky)=> 20 psolvers under 0.42s, top psolvers at 0.10s.

Last edit: 2016-06-23 17:09:52
jack_reacher: 2016-05-20 17:37:12

use long long for final answer

Reem Obaid: 2016-04-27 23:19:25

printf("%I64d\n",ans); gives WA on test 8. all i did was change it to cout <<ans <<endl ; and got AC

utkarsh538: 2016-04-15 11:01:20

learnt a new logic :)

Shashank Tiwari: 2015-12-20 08:42:10

Last edit: 2015-12-20 08:42:21
Divyaanand Sinha: 2015-12-15 11:48:24

getting wrong answer for case 8..

Shashank Tiwari: 2015-12-13 13:38:35

Make sure that final ans can be in long. So , check for integer overflows.

Last edit: 2015-12-20 08:41:21
Abhilash: 2015-07-18 18:40:06

simple one

Ankit Jain: 2013-12-17 21:24:29

nice problem..:)...beautiful to solve..

Added by:CSI
Time limit:3s
Source limit:50000B
Memory limit:1536MB
Cluster: Cube (Intel G860)
Languages:All except: ASM64