CodeForces - 776C（前缀和+思维）

最新推荐文章于 2024-06-23 15:46:07 发布

转载最新推荐文章于 2024-06-23 15:46:07 发布 · 451 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/chenquanwei/p/9477822.html

文章标签：

#数据结构与算法

本文介绍了一道CodeForces上的题目776C的解题思路及代码实现，通过求前缀和并标记每个和的个数，遍历到1e5记录答案，特别针对k为1、-1的情况进行了特殊处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

链接：CodeForces - 776C

题意：给出数组 a[n] ，问有多少个区间和等于 k^x（x >= 0）。

题解：求前缀和，标记每个和的个数。对每一个数都遍历到1e5，记录到答案。

#include <bits/stdc++.h>
using namespace std;

long long n, k;
map<long long, long long> mp;

int main()
{
    scanf("%lld%lld", &n, &k);

    mp[0] = 1;
    long long ans = 0, s = 0;
    for(long long i = 0, a; i < n; i++){
        scanf("%lld", &a);
        s += a;
        if(k == 1) ans += mp[s-1];
        else if(k == -1) ans += mp[s-1] + mp[s+1];
        else{
            long long t = 1;
            while(abs(t) <= 1e15){
                ans += mp[s-t];
                t *= k;
            }
        }
        mp[s]++;
    }

    printf("%lld\n", ans);
    return 0;
}