前缀和 CodeForces - 776C Molly's Chemicals

最新推荐文章于 2021-01-27 21:12:44 发布

PiKa__Qiu

最新推荐文章于 2021-01-27 21:12:44 发布

阅读量527

点赞数

分类专栏：其他

本文链接：https://blog.csdn.net/PiKa__Qiu/article/details/57143579

版权

其他专栏收录该内容

3 篇文章 0 订阅

订阅专栏

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

Input

4 -3
3 -6 -3 12

Output

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;
}