区间和为任意数x的个数

描述

有 $n$ 个数的数组 $a$ 下标从 $1$ 开始，数组中每个元素不超过 $1e9$ ，
设 $f(l,r,x)=[({\sum }_{i=l}^{r}a[i])==x]$ ($[a==b]$)表示 $0/1$ 。
求${\sum }_{l=1}^n{\sum }_{r=l}^{n}f(l,r,x)$的值

分析

设 $p^x={\sum }_{i=1}^{x}a[i]$ ,
所以假设$p^r-p^{l-1}=x$ => $p^r-x=p^{l-1}$
只需要每次将$ans+贡献(p^r-x)$并使$贡献p^{l-1}$++即可

#include <bits/stdc++.h>
using namespace std;
using ll = long long ;
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    vector<int>a(n+1);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    unordered_map<ll,ll>mp;
    mp[0]=1;
    ll ans=0,sum=0;
    for(int i=1;i<=n;i++){
        sum+=a[i];//前缀和
        ans+=mp[sum-m];//相当于找 $p^r-s==p^{l-1}对应p^{l-1}的个数$
        mp[sum]++;//此时相当于左侧的 $p^{l-1}$ 了
    }
    printf("%lld\n",ans);
}

练习

HDU7108
CF1398C