洛谷 1637 三元上升子序列

【题解】

　　显然每个位置的贡献是它前面比它小的数的个数乘上它后面比它大的数的个数。

　　那么离散化之后用树状数组维护即可。

#include<cstdio>
#include<algorithm>
#include<cstring>
#define rg register
#define maxn (500010)
#define lowbit (x&-x)
using namespace std;
int n,m,opt,t[maxn],a[maxn],b[maxn],small[maxn],big[maxn];
long long ans=0;
inline int read(){
	int f=1,k=0; char c=getchar();
	while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
	while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
	return k*f;
}
inline void add(int x){for(;x<=n;x+=lowbit) t[x]++;}
inline int query(int x){int ret=0; for(;x;x-=lowbit) ret+=t[x]; return ret;}
inline void out(){
	for(int i=1;i<=n;i++) printf("%d ",small[i]); puts("");
	for(int i=1;i<=n;i++) printf("%d ",big[i]);
}
int main(){
	n=read();
	for(rg int i=1;i<=n;i++) a[i]=b[i]=read();
	sort(b+1,b+1+n); int N=unique(b+1,b+1+n)-b-1;
	for(rg int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+1+N,a[i])-b;
	for(rg int i=1;i<=n;i++){
		small[i]=query(a[i]-1);
		add(a[i]);
	}
	memset(t,0,sizeof(t));
	for(rg int i=n;i>=1;i--){
		big[i]=n-i-query(a[i]);
		add(a[i]);
	}
	for(rg int i=1;i<=n;i++) ans+=small[i]*big[i];
	printf("%lld
",ans);
	//out();
	return 0;
}