前缀和 CodeForces - 776C Molly's Chemicals

C. Molly's Chemicals

time limit per test

2.5 seconds

memory limit per test

512 megabytes

input

standard input

output

standard output

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 a_i.

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.

Input

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 ≤ 10⁵, 1 ≤ |k| ≤ 10).

Next line contains n integers a₁, a₂, ..., a_n ( - 10⁹ ≤ a_i ≤ 10⁹) — affection values of chemicals.

Output

Output a single integer — the number of valid segments.

Examples

Input

4 2
2 2 2 2

Output

8

Input

4 -3
3 -6 -3 12

Output

3

Note

Do keep in mind that k⁰ = 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].

题意：给n个数和数k，求这n个数里面有多少段的和sum满足sum == k^i (i=0, 1, 2...)。

思路：暴力的话公式是sum[i] - sum[j] = k^t,但是肯定超时，转化一下，变成sum[j] = sum[i] - k^t，那么可以用map类型的数组记录到每一个前缀sum[i]时满足上式，即sum[i]-k的个数。

具体见代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <map>
#include <cstdlib>

typedef long long LL;
using namespace std;
int a[100000+50];
map<LL, int> mp;

int main()
{
    int n, k;
    while(~scanf("%d%d", &n, &k)) {
        for(int i = 0; i <n; i++)
            scanf("%d", &a[i]);
        LL p = 1;
        LL ans = 0;
       while(abs(p) < 1e15) {
            mp.clear();
            mp[0] = 1;
            LL sum = 0;
            for(int i = 0; i < n; i++) {
                sum += a[i];
                ans += mp[sum-p];
                mp[sum]++;
            }
            p *= k;
            if(p == 1) break;
       }
       printf("%lld\n", ans);
    }
    return 0;
}