楼兰图腾
题目链接:https://www.acwing.com/problem/content/243/
思路:我们可以用树状数组倒叙扫描数组a,利用树状数组求出每个a[i]后边每个a[i]后边有几个数比它大。
正序扫描数组a,求出每个a[i]前边有几个数比它大;
然后再依次枚举每个点作为中间的点。
#include<iostream> #include<algorithm> #include<string> #include<cstring> #include<cstdio> #include<cmath> #include<map> #include<queue> using namespace std; long long int a[200005],l[200005],r[200005],ll[200005],rr[200005],st[200005]; long long int n; void add(long long int x) { for(;x<=n;x+=x&-x) st[x]++; } long long int ask(long long int x) { long long int ans=0; for(;x;x-=x&-x) ans+=st[x]; return ans; } int main() { scanf("%lld",&n); for(int i=1;i<=n;i++) scanf("%lld",&a[i]); for(int i=1;i<=n;i++) { add(a[i]); l[i]=ask(a[i]-1); ll[i]=ask(n)-ask(a[i]); } memset(st,0,sizeof(st)); for(int i=n;i>=1;i--) { add(a[i]); r[i]=ask(a[i]-1); rr[i]=ask(n)-ask(a[i]); } long long int ans=0,ans1=0; for(int i=1;i<=n;i++) { ans=ans+l[i]*r[i]; ans1=ans1+ll[i]*rr[i]; } printf("%lld %lld\n",ans1,ans); return 0; }