补题 nefu1494

链接·

对于此题，求有多少对i，j那么现在我们有两个变量，很简单的想法，定一动一。那么我们对于一个串，我们首先处理前缀和，那么用a[n]-a[i]等于k的方式，就可以找到有多少个j，那么我们再从头找i，从头遍历i就可，对于i找他之后有多少个j直接加在ans上即可，那么我们就可以知道，共有多少对不同的i，j。

对于i后面有多少个j，可以用树状数组的形式，或者第一步找的时候直接处理在数组中，这样就减少了寻找有多少个j的时间复杂度，

代码：

#include<algorithm>
#include<cstdio>
#include<cstring>

using namespace std;
const int M=5e5+10;
#define ll long long
//要求两个值i,j，先求出所有可能的j的位置，再从头枚举所有的i，如果i的存在，那么我=我们看在i之后有多少个j可能跟i将数列分成三个区间，j的位置用树状数组储存，总时间复杂度
//n*logn(n是便利i，logn是树状数组查询)
int trie[M],n,m;
ll sum[M],t;
void update(int x,int t)
{
    for(int i=x;i<=M;i+=i&(-i))
    {
        trie[i]+=t;
    }
}
int query(int x)
{
    int ans=0;
    for(int i=x;i;i-=i&(-i))
    {
        ans+=trie[i];
    }
    return ans;
}


int main()
{
    while(~scanf("%d",&n))
    {
        //sum[0]=0;
        memset(trie,0,sizeof(trie));
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&t);
            sum[i]=sum[i-1]+t;
        }
        if(sum[n]%3!=0||n<3)
        {
            printf("0\n");
        }
        else
        {
            ll k=sum[n]/3;
            ll ans=0;
             for(int i=1;i<n;i++)
            {
                if(sum[n]-sum[i]==k)
                {
                    update(i,1);
                }
            }
            int s=query(n);
            for(int i=1;i<=n;i++)
            {
                if(sum[i]==k)
                ans+=s-query(i);
            }
            printf("%lld\n",ans);
        }
    }
   return 0;
}