今天的每日一题,题目的意思也是很简单的,就是说把一个数组分成平均的三分。
1:因为数组是连续的,所以第一段一定是1~x,第二段是x+1~y,第三段是y+1~n。很显然,一个数组可能会有很多种x,y的组合,但是三个段都得相等,且三个段连续,我们很容易发现一个利用前缀和写题目的做法。
2:设长度为len,1~x是len,1~y是2*len,y+1~n是3*len,3*len = 数组的总和,所以len就一定是一个固定的值,很明显第三段是不需要考虑的,所以只需要考虑前面的俩段,len 2*len 就可以组合成一个符合题目意思的解。
3:所以题目就变成了找到每个len并且记录其后有多少个2*len,然后再把他们累加起来。
4:还有一个比较特殊的地方,就是假如题目给出来的数,都是0,那也就是说,我们只需要在n-1个位置中取俩个的问题。
然后就是代码。
#include<iostream>
using namespace std;
const int N = 1e5+5;
int a[N];
int sum[N];
int main()
{
long long n;
cin >> n ;
int f = 0;
for(int i = 1 ; i <= n ; i ++)
{
scanf("%d",&a[i]);
if(a[i] != a[i-1])
f = 1;
}
long long ans = 0 ;
long long d = 0;
for(int i = 1 ; i <= n ; i ++)
sum[i] = sum[i-1] + a[i];
if(sum[n] % 3 || n < 3)cout <<0 << endl;
else if(f == 0)
cout << (n - 1) * (n - 2) / 2<< endl ;
else
{
for(int i = n ; i >= 1; i --)
{
if(sum[i] == sum[n] - sum[n] / 3) d++;
if(sum[i] == sum[n] - 2 * sum[n] / 3) ans+= d;
}
cout << ans << endl;
}
return 0;
}