-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCF_776C.java
More file actions
112 lines (86 loc) · 2.43 KB
/
CF_776C.java
File metadata and controls
112 lines (86 loc) · 2.43 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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import java.io.BufferedReader;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.StringTokenizer;
import java.util.TreeMap;
public class CF_776C {
public static void main(String[] args) throws IOException {
Scanner sc = new Scanner(System.in);
PrintWriter out = new PrintWriter(System.out);
int n = sc.nextInt(), k = sc.nextInt();
long powers[];
if(k == 0) {
powers = new long[1];
powers[0] = 0;
}
else if(k == 1) {
powers = new long[1];
powers[0] = 1;
}
else if(k == -1) {
powers = new long[2];
powers[0] = 1;
powers[1] = -1;
}
else {
int sz = (int) (Math.log(1e17) / Math.log(Math.abs(k)));
powers = new long[sz + 1];
powers[0] = 1;
long a = k;
for(int i = 1; i < powers.length; ++i) {
powers[i] = a;
a *= k;
}
}
int arr[] = new int[n];
for(int i = 0; i < n; ++i) arr[i] = sc.nextInt();
long sums[] = new long[n];
TreeMap<Long, Long> map = new TreeMap<Long, Long>();
sums[0] = arr[0];
long res = 0;
for(int i = 0; i < powers.length; ++i) {
if(sums[0] == powers[i]) {
res++;
break;
}
}
map.put(sums[0], 1L);
for(int i = 1; i < n; ++i) {
sums[i] = sums[i - 1] + arr[i];
for(int j = 0; j < powers.length; ++j) {
if(sums[i] == powers[j]) res++;
if(map.containsKey(sums[i] - powers[j])) {
res += map.get(sums[i] - powers[j]);
}
}
if(map.containsKey(sums[i]))
map.put(sums[i], map.get(sums[i]) + 1);
else
map.put(sums[i], 1L);
}
out.println(res);
out.flush();
out.close();
}
static class Scanner{
StringTokenizer st;
BufferedReader br;
public Scanner(InputStream s){ br = new BufferedReader(new InputStreamReader(s));}
public Scanner(FileReader r){ br = new BufferedReader(r);}
public String next() throws IOException
{
while (st == null || !st.hasMoreTokens())
st = new StringTokenizer(br.readLine());
return st.nextToken();
}
public int nextInt() throws IOException {return Integer.parseInt(next());}
public long nextLong() throws IOException {return Long.parseLong(next());}
public String nextLine() throws IOException {return br.readLine();}
public double nextDouble() throws IOException { return Double.parseDouble(next()); }
public boolean ready() throws IOException {return br.ready();}
}
}