树状数组优化算法-CSDN博客

本文链接：https://blog.csdn.net/wanherun/article/details/78108110

先枚举每一个数，看它前面有几个比它小，算一下和为sum1，后面有几个比它大，算一下和为sum2，对答案的贡献为A[i]sum1sum2。

离散化后，树状数组就可以了。

#include<bits/stdc++.h>
#define N 300000
using namespace std;
int n,A[N+5];
int data[N+5],siz;
long long tmp,ans;
long long c[N+5],Mn[N+5],Mx[N+5];
const long long mod=19260817;
inline char nc()
{
	static char buf[100000],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
	int x=0,b=1;
	char c=nc();
	for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
	for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
	return x*b;
}
inline void write(int x)
{
	if(x==0)putchar('0');
	else
	{
		char buf[15];
		int len=0;
		if(x<0)putchar('-'),x=-x;
		while(x)buf[++len]=x%10+'0',x/=10;
		for(int i=len;i>=1;i--)putchar(buf[i]);
	}
	putchar(' ');
}
inline int lowbit(int x)
{
	return x&-x;
}
inline void add(int pos,int val)
{
	for(int i=pos;i<=n;i+=lowbit(i))
		c[i]+=val;
}
inline long long qurry(int pos)
{
	long long sum=0;
	for(int i=pos;i>=1;i-=lowbit(i))
		sum+=c[i];
	return sum;
}
int main()
{
	freopen("in.txt","r",stdin);
	n=read();
	for(int i=1;i<=n;i++)A[i]=read();
	for(int i=1;i<=n;i++)data[i]=A[i];
	sort(data+1,data+n+1);
	siz=unique(data+1,data+n+1)-data-1;
	for(int i=1;i<=n;i++)A[i]=lower_bound(data+1,data+siz+1,A[i])-data;
	for(int i=1;i<=n;i++)
	{
		Mn[i]=qurry(A[i]-1)%mod;
		add(A[i],data[A[i]]);
	}
	memset(c,0,sizeof(c));
	for(int i=n;i>=1;i--)
	{
		Mx[i]=(qurry(siz)-qurry(A[i]))%mod;
		add(A[i],data[A[i]]);
	}
	for(int i=1;i<=n;i++)
	{
		tmp=Mn[i]*Mx[i]%mod;
		ans=(ans+tmp*data[A[i]])%mod;
	}
	cout<<ans;
	return 0;
}

好慢呀，怕是被____走了（逃