CF466C Number of Ways (#树状数组)

题意翻译

题目描述 你得到了一个数组a，包含n个数字a[1],a[2],a[3],...,a[n].

现在你要找到把它分成三份的方法，使得每一份之内所有数的和相等。

简单地说，你需要找到一个数对(i,j)，使得

输入数据

第一行是一个整数n(1<=n<=500000)，表示数列中有多少个数。

接下来一行有n个数，分别表示a[1],a[2],a[3],...,a[n]。

输出数据

一行一个整数，表示满足条件的(i,j)的总数。

输入输出样例

输入样例#1： 复制

5
1 2 3 0 3

输出样例#1： 复制

2

输入样例#2： 复制

4
0 1 -1 0

输出样例#2： 复制

1

输入样例#3： 复制

2
4 1

输出样例#3： 复制

0

---

思路

有前缀和的地方，怎么能缺少树状数组呢？（雾）

题目大意是给定长为n的序列ai，将其分成3个连续子串，要求每个子串的和相同，求划分的方案数。

具体看一下代码，还是和纯前缀和的做法很像的。

#include <stdio.h>
#include <iostream>
using namespace std;
long long int a[500001],bit[500001],n,s,t,lxydl,cnt;
inline int lowbit(int x)
{
	return x&-x;
}
inline void update(int x,int k)
{
	cnt++;//记录多少个符合总和2/3的 
	while(x<=n)
	{
		bit[x]+=k;//全部加上1 
		x+=lowbit(x);
	}
}
inline int query(int x)//查询 
{
	int sum(0);
	while(x>0)
	{
		sum+=bit[x];
		x-=lowbit(x);
	}
	return sum;
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	register int i,j;
	cin>>n;
	for(i=1;i<=n;i++)
	{
		cin>>a[i];
		s+=a[i];//求序列总和 
	}
	if(s%3)//总和不被3整除肯定无解 
	{
		cout<<0<<endl;
		return 0;
	}
	for(i=1;i<=n;i++)
	{
		t+=a[i];//不断加
		if(t==s/3*2 && i!=n)//如果找到总和2/3的，就记录到树状数组，此时树状数组的意义是记录总和2/3的区间 
		{
			update(i,1);//更新 
		}
	}
	for(t=0,i=1;i<=n;i++)//在扫一遍 
	{
		t+=a[i];
		if(t==s/3)//若找到总和1/3
		{
			lxydl=lxydl+cnt-query(i);//2/3的总数减去2/3的区间 
		}
	}
	cout<<lxydl<<endl;
	return 0;
}