P1637 三元上升子序列

链接:Miku

----------------------

对于这个题,我们对于每一个数i,分别求出所有比它小的,在它前面的和

比它大的,在它后面的,然后把这两个乘起来,然后再把这些积加起来就可以了

-----------------------

然而这样直接做复杂度太高了,我们要优化,仿照树状数组求逆序对的方法,我们就可以在可以接受的时间内求出并且解决问题了

------------------------

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n;
int a[30005];
int tree[30005];
int b[30005];
long long minn[30005];
long long ans;
bool cmp(const int &x,const int &y){//要考虑数字相同的情况 
    if(a[x]!=a[y])
    return a[x]<a[y];
    else
    return x>y;
}
int lowbit(int x){
    return x&-x;
}
void add(int x,int k){
    while(x<=n){
        tree[x]+=k;
        x+=lowbit(x);
    }
    return ;
}
int sum(int x){
    int ans=0;
    while(x){
        ans+=tree[x];
        x-=lowbit(x);
    }
    return ans;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;++i){
        scanf("%d",&a[i]);
        b[i]=i;
    }
    sort(b+1,b+1+n,cmp);
    for(int i=1;i<=n;++i){//求比它小的 
        add(b[i],1);
        minn[b[i]]=sum(b[i]-1);
    }
    memset(tree,0,sizeof(tree));
    for(int i=n;i>=1;--i){//倒着求,就是大的 
        add(b[i],1);
        minn[b[i]]*=sum(n)-sum(b[i]);//直接乘骑来 
    }
    for(int i=1;i<=n;++i){
        ans+=minn[i];
    }
    printf("%lld",ans);
    return 0;
}
Ac

 

posted @ 2020-02-15 19:05  Simex  阅读(107)  评论(0编辑  收藏  举报