Molly Hooper has n different kinds of chemicals arranged in a line. Each of the chemicals has an affection value, The i-th of them has affection value ai.
Molly wants Sherlock to fall in love with her. She intends to do this by mixing a contiguous segment of chemicals together to make a love potion with total affection value as a non-negative integer power of k. Total affection value of a continuous segment of chemicals is the sum of affection values of each chemical in that segment.
Help her to do so in finding the total number of such segments.
The first line of input contains two integers, n and k, the number of chemicals and the number, such that the total affection value is a non-negative power of this number k. (1 ≤ n ≤ 105, 1 ≤ |k| ≤ 10).
Next line contains n integers a1, a2, ..., an ( - 109 ≤ ai ≤ 109) — affection values of chemicals.
Output a single integer — the number of valid segments.
4 2 2 2 2 2
8
4 -3 3 -6 -3 12
3
Do keep in mind that k0 = 1.
In the first sample, Molly can get following different affection values:
- 2: segments [1, 1], [2, 2], [3, 3], [4, 4];
- 4: segments [1, 2], [2, 3], [3, 4];
- 6: segments [1, 3], [2, 4];
- 8: segments [1, 4].
Out of these, 2, 4 and 8 are powers of k = 2. Therefore, the answer is 8.
In the second sample, Molly can choose segments [1, 2], [3, 3], [3, 4].
思路:先用前缀和处理子序列的和,如果枚举每个子序列的和即sum[i]-sum[j]=k^x,肯定会超时,那么可以做一个变换即sum[i]-k^x=sum[j],这样的姿势就能过了,对于k, 用map存储sum[i]有几种,如果mp[sum[i+1]-k]存在,那么ans+=mp[sum[i+1]-k]。 然后k*=k
注意:这种变换要求是连续的。而且枚举每个k的时候,都要从头前缀,在前缀的过程中记录ans,如果一开始前缀好了后面的值会影响前面的,然而他们可能并不是连续的
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
#include <cmath>
using namespace std;
typedef long long ll;
const int maxn = 1e5 + 5;
const ll INF = 1e14 + 5;
ll sum;
int a[maxn];
map<ll, int> mp;
int main()
{
int n, k;
scanf("%d%d", &n, &k);
for(int i = 1; i <= n; i++)
scanf("%d", &a[i]);
ll pow = 1, ans = 0;
while(abs(pow) < INF)
{
mp.clear();
mp[0] = 1;
sum = 0;
for(int i = 1; i <= n; i++)
{
sum = sum + a[i];
ans += mp[sum-pow];
mp[sum]++;
}
pow = pow * k;
if(pow == 1) break;
}
cout << ans << endl;
return 0;
}