2020-2021 ICPC Southeastern European Regional E. Divisible by 3(前缀和,优化暴力)

最新推荐文章于 2021-10-19 22:22:48 发布

issue是fw

最新推荐文章于 2021-10-19 22:22:48 发布

阅读量503

点赞数

分类专栏： gym 神仙思维题文章标签： c++ 算法数据结构

本文链接：https://blog.csdn.net/jziwjxjd/article/details/120347301

版权

神仙思维题同时被 2 个专栏收录

104 篇文章 14 订阅

订阅专栏

gym

70 篇文章 0 订阅

订阅专栏

LINK

区间 $[l, r]$ 的权值计算方式是( $sum=\sum\limits_{i=l}^r a_i$ )

$weight[l,r]=\sum\limits_{i=l}^r a_i*(sum-a_i)=sum*\sum\limits_{i=l}^r a_i-\sum\limits_{i=1}^r a_i^2$

令 $a_i$ 的前缀和数组为 $p_1$ , $a_i^2$ 的前缀和数组为 $p_2$

则 $weight[l,r]=(p_1[r]-p_1[l-1])^2-(p_2[r]-p_2[l-1])$

则 $p_1[r]^2+p_1[l-1]^2-2*p_1[r]*p_1[l-1]\ )-p_2[r]+p_2[l-1]=3*f$ (其中 $f$ 为任意整数)

考虑枚举 $r$

$3*f-p_1[r]^2+p_2[r]=p_1[l-1]*(p_1[l-1]-2*p_1[r])+p_2[l-1]$

注意到运算实在模 $3$ 意义下,所以我们完全可以处理一个这样的数组 $m [i] [j] [k]$ 表示

在 $p_1[r]$ 模 $3$ 为 $j$ 的情况下,且 $l-1\in[1,i]$ 时,存在多少个下标满足

$p_1[l-1]*(p_1[l-1]-2*p_1[r])+p_2[l-1]=k$

这样一来,转移就是 $O (1)$ 的了,我们枚举 $r$ ,符合条件的 $l - 1$ 索引的个数就是 $m [r - 1] [j] [k]$

其中 $j=p_1[r]\%3$ 而 $k=-p_1[r]^2+p_2[r]$

时间复杂度为 $O (n)$

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 5e5+10;
const int mod = 3;
void upd(int &x,int y){ x = (( x+y )%mod+mod)%mod; }
int n,m[4][4],p1[maxn],p2[maxn];
signed main()
{
	cin >> n;
	int ans = 0;
	for(int i=0;i<=n;i++)
	{
		if( i>0 )
		{
			int x; cin >> x;
			upd( p1[i],p1[i-1]+x );
			upd( p2[i],p2[i-1]+x*x );
			int j = p1[i]%mod, k = (( -p1[i]*p1[i]+p2[i] )%mod+mod )%mod;
			ans += m[j][k];
		}
		for(int j=0;j<=2;j++)
		{
			int k = p1[i]*( p1[i]-2*j )+p2[i];
			m[j][(k%mod+mod)%mod]++;
		}
	}
	cout << ans;
}