JZO5951. 【NOIP2018模拟11.04】锋芒毕露

题解

可以发现求相交的圆的对数并不好求，
于是就应该考虑求不相交圆的个数。
首先先算出所有圆的个数，那么也就可以知道圆的对数。
先减去相同颜色相交的圆，这个非常简单。
考虑AABB这种不包含的情况，
从前往后扫一遍，记录每个点前面有多少个圆，也就每个点前面与它相同颜色的点的个数。
再考虑被包含的情况ABBA
先设一个阈值K，然后按照K对颜色进行分类，
第一类颜色L,出现次数大于K，颜色数量较多，种类较少
第一类颜色P,出现次数小于等于K，颜色数量较少，种类较少。
首先，LPPL这种情况，
因为L种类比较少，就枚举L的颜色，
对于一对P的方案数就是左边的P左边L的个数，右边P右边P的个数。
用前缀和维护一下，这一部分就可以做到$O(n^2/K)$
第二种情况，?LL?这种情况，
如果知道了一对?中间L的个数，显然是可以知道方案是的，
但是方案数是二次的，所以就维护一个一次的前缀和，一个二次的前缀和。
这一部分的复杂度也是$O(n^2/K)$
最后一种情况，就是P1P2P2P1，
可以知道相同P的数量很少，
所以直接枚举一对P1，用树状数组统计里面不同颜色的P对数，
所以这一部分的复杂度就是$O(nKlo{g}_n)$
现在考虑阈值K取什么，
显然是让两部分的复杂度平均，$K=\sqrt{(}n/lo{g}_n)$

code

#pragma GCC optimize(2)
#pragma G++ optimize(2)
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define ll long long
using namespace std;
const int N=1000003,mo=19260817,inv=9630409;
int col[N],t[N],n,K,ans,sum,sum1,ans1;
int lst[N],nxt[N],pre[N],g[N];
int m1,z1[N],m2,z2[N],mxcol,s[N],ss[N],s1[N],s2[N],s3[N],si[N];
char ch;
void read(int&n)
{
	for(ch=getchar();ch<'0' || ch>'9';ch=getchar());
	for(n=0;'0'<=ch && ch<='9';ch=getchar())n=(n<<1)+(n<<3)+ch-48;
}
int max(int x,int y){return x>y?x:y;}
int C(ll x){return x>0?x*(x-1)/2%mo:0;}
int x_(int x){return (-x)&x;}
int find(int x)
{
	int s=0;
	for(int i=x;i;i=i-x_(i))s=s+g[i];
	return s;
}
void ins(int x)
{
	for(int i=x;i<=n;i=i+x_(i))g[i]++;
}
ll ksm(ll x,int y)
{
	ll s=1;
	for(;y;y>>=1,x=x*x%mo)
		if(y&1)s=s*x%mo;
	return s;
}
int main()
{
	freopen("everytime.in","r",stdin);
	freopen("everytime.out","w",stdout);
	read(n);K=sqrt(n/log(n));
	for(int i=1;i<=n;i++)read(col[i]),t[col[i]]++,
		nxt[pre[i]=lst[col[i]]]=i,lst[col[i]]=i,mxcol=max(mxcol,col[i]),
			si[i]=si[pre[i]]+1,s[i]=(s[i-1]+si[i]-1)%mo;
	//AABB
	for(int i=1;i<=n;i++)
		ans=(ans+(ll)(s[i]-C(si[i]))*(t[col[i]]-si[i]))%mo;
	for(int i=1;i<=mxcol;i++)
		if(t[i])
			if(t[i]<=K)z1[++m1]=i;else z2[++m2]=i;
	
	sum=0;
	for(int i=1;i<=m1;i++)sum=(sum+C(t[z1[i]]))%mo,sum1=(sum1+C(C(t[z1[i]])));
	for(int i=1;i<=m2;i++)sum=(sum+C(t[z2[i]]))%mo,sum1=(sum1+C(C(t[z2[i]])));;
	ans1=(C(sum)-ans-sum1)%mo;ans=0;
	ans1=(ans1+mo)%mo;
	
	//LPPL ?LL?
	for(int j=1;j<=m2;j++)
	{
		s3[0]=s[0]=ss[n+1]=0;
		for(int i=1;i<=n;i++)s[i]=s[i-1]+(col[i]==z2[j]),si[i]=si[pre[i]]+1,
			s1[i]=(s1[pre[i]]+s[i])%mo,s2[i]=(s2[pre[i]]+(ll)s[i]*s[i])%mo;
		for(int i=n;i;i--)ss[i]=ss[i+1]+(col[i]==z2[j]);
		for(int i=1;i<=n;i++)
		{
			if(col[i]==z2[j])continue;
			if(t[col[i]]<=K)
			{
				ans=(ans+(ll)s3[pre[i]]*ss[i]%mo)%mo;
				s3[i]=(s3[pre[i]]+s[i])%mo;
			}
			sum=(ll)s[i]*s[i]*si[pre[i]]%mo+(ll)(1-2*s[i])*s1[pre[i]]%mo+s2[pre[i]]-(ll)si[pre[i]]*s[i];
			sum=sum%mo*inv%mo;sum=(sum+mo)%mo;ans=(ans+sum)%mo;
		}
	}
	//PPPP
	si[0]=0;
	for(int i=1;i<=n;i++)
	{
		if(t[col[i]]>K)continue;
		si[i]=si[pre[i]]+1;
		for(int j=i;pre[j];j=pre[j])ans=(ans+find(i)-find(pre[j])-C(si[i]-si[pre[j]]-1))%mo;
		for(int j=i;pre[j];j=pre[j])ins(pre[j]);
	}
	ans=(ans1-ans+mo)%mo;
	printf("%d\n",ans);
}