-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsubadiv.java
More file actions
84 lines (73 loc) · 1.7 KB
/
subadiv.java
File metadata and controls
84 lines (73 loc) · 1.7 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/*
ID: grifync1
LANG: JAVA
PROG: subasum2
*/
//lil lil peezy
import java.util.*;
import java.io.*;
public class subadiv {
public static int conv(String s) {
return Integer.parseInt(s);
}
public static int max(int a, int b) {
return Math.max(a, b);
}
public static int min(int a, int b) {
return Math.min(a, b);
}
public static void print(int num) {
System.out.println(num);
}
public static void prints(String s) {
System.out.println(s);
}
public static void printa(int[] a) {
for (int i = 0; i < a.length; i++) {
if (i == 0) {
System.out.print(a[i]);
} else {
System.out.print(" " + a[i]);
}
}
}
public static int[] sort(int[] nums) {
Arrays.sort(nums);
return nums;
}
public static void main(String[] args) throws IOException {
//BufferedReader in = new BufferedReader(new FileReader("subasum2.in"));
//PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter("subasum2.out")));
BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
PrintWriter out = new PrintWriter(System.out);
//StringTokenizer st = new StringTokenizer(in.readLine());
HashMap<Long, Integer> map = new HashMap<Long, Integer>();
int n = conv(in.readLine());
StringTokenizer st = new StringTokenizer(in.readLine());
int[] nums = new int[n];
for(int i=0; i<n; i++) {
nums[i]=conv(st.nextToken());
}
int rp = 0;
long ans = 0;
long sum = 0;
map.put(0L,1);
while(rp<n) {
sum += nums[rp];
long rem = sum%n;
if(rem<0) {
rem+=n;
}
if(map.containsKey(rem)) {
ans+=map.get(rem);
}
int count = map.getOrDefault(rem,0);
count++;
map.put(rem,count);
rp++;
}
out.println(ans);
in.close();
out.close();
}
}