[Code+#1]Yazid 的新生舞会

XXIV.[Code+#1]Yazid 的新生舞会

关于众数，我们通常是对于每个数求出它作为众数的区间数。

考虑某个数 $x$。我们可以发现，如果令 $a_i=x$ 的所有位置有 $b_i=1$，其余位置有 $b_i=-1$，则如果某个区间 $[l,r]$ 关于 $b$ 的区间和 $>0$ 的话，则有 $x$ 是 $[l,r]$ 的绝对众数。

考虑区间和可以被拆成前缀和之差。于是我们考虑求出 $b$ 的前缀和 $s$ 数组。则，对于位置 $i$，我们需要找出有多少个 $j$，满足 $j<i\land s_j<s_i$。

这明显是二维数点问题，常规做法是线段树。但是，我们不能对所有颜色全部从头到尾做一遍二维数点，不然复杂度就是 $O(n^2\log n)$ 的。考虑 $s_i$ 数组有何性质。

稍加观察可以发现，$s$ 数组在大部分位置都是有 $s_i=s_{i-1}-1$ 的，只有在 $a_i=x$ 处才会出现例外；于是我们就可以考虑相邻的两个 $a_j=a_i=x$。它实际在线段树上的贡献，是 $[s_j,s_i)$ 区间 $+1$，可以直接完成。

然后考虑一个单独的 $a_i$，我们要做的是在线段树上求前缀和；现在既然是 $[j,i)$ 的区间了，那么我们就对 $[s_j,s_i)$ 中的所有东西都求前缀和。实际上就是对线段树上每个位置乘上了一个系数，直接在线段树上同时维护系数即可。

时间复杂度 $O(n\log n)$。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,a[500100];
vector<int>v[500100];
ll res;
#define lson x<<1
#define rson x<<1|1
#define mid ((l+r)>=0?(l+r)>>1:(l+r-1)>>1)
struct Segtree{
	int tag,sum;
	ll mus;
}seg[4001000];
void ADD(int x,int l,int r,int y=1){seg[x].tag+=y,seg[x].sum+=(r-l+1)*y,seg[x].mus+=1ll*((n-l)+(n-r))*(r-l+1)*y/2;}
void pushdown(int x,int l,int r){ADD(lson,l,mid,seg[x].tag),ADD(rson,mid+1,r,seg[x].tag),seg[x].tag=0;}
void pushup(int x){seg[x].sum=seg[lson].sum+seg[rson].sum,seg[x].mus=seg[lson].mus+seg[rson].mus;}
void reset(int x,int l,int r){
	seg[x].tag=seg[x].sum=seg[x].mus=0;
	if(l==r)return;
	if(seg[lson].sum)reset(lson,l,mid);
	if(seg[rson].sum)reset(rson,mid+1,r);
}
void modify(int x,int l,int r,int L,int R){
	if(l>R||r<L)return;
	if(L<=l&&r<=R)ADD(x,l,r);
	else pushdown(x,l,r),modify(lson,l,mid,L,R),modify(rson,mid+1,r,L,R),pushup(x);
}
ll querytri(int x,int l,int r,int L,int R){
	if(l>R||r<L)return 0;
	if(L<=l&&r<=R)return seg[x].mus-1ll*seg[x].sum*(n-R-1);
	pushdown(x,l,r);
	return querytri(lson,l,mid,L,R)+querytri(rson,mid+1,r,L,R); 
}
ll querypla(int x,int l,int r,int L,int R){
	if(l>=L)return 0;
	if(r<L)return 1ll*seg[x].sum*(R-L+1);
	pushdown(x,l,r);
	return querypla(lson,l,mid,L,R)+querypla(rson,mid+1,r,L,R);
}
int main(){
	scanf("%d%d",&n,&a[0]);
	for(int i=0;i<n;i++)v[i].push_back(0);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]),v[a[i]].push_back(i);
	for(int i=0;i<n;i++)v[i].push_back(n+1);
	for(int i=0;i<n;i++){
		if(v[i].size()==2)continue;
		int las=0;
		for(int j=1;j+1<v[i].size();j++){
//			printf("ADD:[%d,%d]\n",las-(v[i][j]-v[i][j-1]-1),las);
			modify(1,-n,n,las-(v[i][j]-v[i][j-1]-1),las);
			las-=(v[i][j]-v[i][j-1]-1)-1;
			res+=querytri(1,-n,n,las-(v[i][j+1]-v[i][j]),las-1)+querypla(1,-n,n,las-(v[i][j+1]-v[i][j]),las-1);
//			printf("SUM:[%d,%d]:%lld\n",las-(v[i][j+1]-v[i][j]),las-1,query(1,-n,n,las-(v[i][j+1]-v[i][j]),las-1));
		}
		reset(1,-n,n);
	}
	printf("%lld\n",res);
	return 0;
}