-
Notifications
You must be signed in to change notification settings - Fork 0
/
C.cpp
62 lines (46 loc) · 1.04 KB
/
C.cpp
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
#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
typedef long double ld;
typedef pair<ll,ll> pll;
#define all(v) (v).begin(),(v).end()
#define rall(v) (v).rbegin(),(v).rend()
#define endl "\n"
#define fast ios_base::sync_with_stdio(false);cin.tie(NULL);
ll const inf = 1e15;
ll const maxn = 1e6+5;
ll const mod = 1e9+7;
// code begins here
int main(){
fast;
ll n,k;
cin>>n>>k;
vector<ll> arr(n);
for(auto &u: arr)
cin>>u;
for(ll i=1;i<n;i++)
arr[i] += arr[i-1];
map<ll, ll> mp;
set<ll> s;
if(abs(k)==1){
s.insert(k);
s.insert(1);
} else {
ll x = (ld)log2((ld)inf)/(ld)log2((ld)abs(k));
ll i= 1;
while(x--){
s.insert(i);
i *= k;
}
}
ll ans = 0;
for(ll i=0;i<n;i++){
if(s.count(arr[i]))
ans++;
for(auto u: s)
ans += mp[arr[i]-u];
mp[arr[i]]++;
}
cout<<ans<<endl;
return 0;
}