BZOJ 3745: [Coci2015]Norma cdq分治

3745: [Coci2015]Norma

Time Limit: 20 Sec  Memory Limit: 64 MB
Submit: 403  Solved: 198
[Submit][Status][Discuss]

Description

Input

第1行，一个整数N；

第2~n+1行，每行一个整数表示序列a。

Output

输出答案对10^9取模后的结果。

Sample Input

4
2
4
1
4

Sample Output

109
【数据范围】
N <= 500000
1 <= a_i <= 10^8

当你试图用 int 卡常时

你不仅会因为 mod 失去优美的代码复杂度

而且 你其实跑的比 long long 慢

泪泪泪

题解看这里

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=500100,mod=int(1e9);

int n,a[N];

ll sum[3][N];
ll mul[3][N];

ll ans;

void cdq(int l,int r)
{
	if(l==r){(ans+=1ll*a[l]*a[l]%mod)%=mod;return ;}
	int mid((l+r)>>1);
	cdq(l,mid);cdq(mid+1,r);
	sum[0][mid]=sum[1][mid]=sum[2][mid]=0;
	mul[0][mid]=mul[1][mid]=mul[2][mid]=0;
	
	for(int i(mid+1),mx(0),mn(mod);i<=r;++i)
	{
		mx=max(mx,a[i]);
		mn=min(mn,a[i]);
		sum[0][i]=(sum[0][i-1]+mx)%mod;
		sum[1][i]=(sum[1][i-1]+mn)%mod;
		sum[2][i]=(sum[2][i-1]+1ll*mx*mn%mod)%mod;
		mul[0][i]=(mul[0][i-1]+1ll*mx*i%mod)%mod;
		mul[1][i]=(mul[1][i-1]+1ll*mn*i%mod)%mod;
		mul[2][i]=(mul[2][i-1]+1ll*mx*mn%mod*i%mod)%mod;
	}
	
	for(int i(mid),j(mid),k(mid),mx(0),mn(mod),x,y;i>=l;i--)
	{
		mx=max(mx,a[i]);
		mn=min(mn,a[i]);
		while(j<r&&a[j]>=mn&&a[j+1]>=mn)j++;
		while(k<r&&a[k]<=mx&&a[k+1]<=mx)k++;
		x=min(j,k);y=max(j,k);
		
		(ans+=(1ll*mx*mn%mod*(1ll*(x+mid-i-i+3)*(x-mid)/2%mod)%mod)%mod)%=mod;
		(ans+=(1ll*mul[2][r]-mul[2][y]-1ll*(i-1)*(sum[2][r]-sum[2][y])%mod)%mod)%=mod;
		if(k>j)
			(ans+=1ll*mx*((mul[1][y]-mul[1][x]-1ll*(i-1)*(sum[1][y]-sum[1][x])%mod)%mod+mod)%mod)%=mod;
		else
			(ans+=1ll*mn*((mul[0][y]-mul[0][x]-1ll*(i-1)*(sum[0][y]-sum[0][x])%mod)%mod+mod)%mod)%=mod;
		ans=(ans%mod+mod)%mod;
	}
}

int main()
{
	n=read();
	register int i;
	for(i=1;i<=n;++i)a[i]=read();
	cdq(1,n);
	cout<<ans<<endl;
	return 0;
}
/*
4
2 4 1 4

109
*/