F : yeleng与新型电瓶车(1315)

题目链接：http://acm.xidian.edu.cn/problem.php?id=1315

　　前缀和 + 枚举 + map维护，map中key只有一个，不可重复，value可以重复；

区间和为sum[i] - sum[j]，题目条件是 sum[i] - sum[j] = k，即查找是否存在sum[j] = sum[i] - k，在value中存放sum[i]之前的sum[j]有多少个，若find ！= end()，则做sum[j]++，否则将新的<sum[i]，1>压map；

***注意数字范围

***1、-1、0要特殊处理

#include<stdio.h>
#include<math.h> 
#include<map>

using namespace std;

map<long long,int> S;
int main()
{
    int n;
    int m;
    int A[100005];
    long long sum[100005];
    while(scanf("%d %d",&n,&m) != EOF)
    {
        S.clear();
        int ans = 0;
        sum[0] = 0;
        S.insert(pair<long long,int> (0,1));
        for(int i = 1;i <= n;i++)
        {
            scanf("%d",&A[i]);
            sum[i] = sum[i - 1] + A[i];
            if(m == 1)
                {
                if(S.find(sum[i] - 1) != S.end())
                    ans += S[sum[i] - 1];
                }
            else if(m == 0)
            {
                if(S.find(sum[i]) != S.end())
                    ans += S[sum[i]];
            }
            else if(m == -1)
            {
                if(S.find(sum[i] - 1) != S.end() )
                    ans += S[sum[i] - 1];
                if(S.find(sum[i] + 1) != S.end() )
                    ans += S[sum[i] + 1];
            }
            else{
                long long temp = 0;
                for(int k = 0;temp < 1e14 && temp > -1e14;k++)
                {
                    temp = pow(m,k);
                    if(S.find(sum[i] - temp ) != S.end())
                        ans += S[sum[i] - temp];
                }
            }
            if(S.find(sum[i]) != S.end())
            S[sum[i]]++; 
            else
            S.insert(pair<long long,int> (sum[i],1));    
        }
        printf("%d
",ans);
    }
    return 0;
}