[ 树状数组 ] BZOJ5055

最新推荐文章于 2019-04-18 10:55:48 发布

gjghfd

最新推荐文章于 2019-04-18 10:55:48 发布

阅读量224

点赞数

分类专栏：树状数组

本文链接：https://blog.csdn.net/gjghfd/article/details/78892914

版权

树状数组专栏收录该内容

15 篇文章 0 订阅

订阅专栏

先求出 $b_i=a_i \times \sum_{j=1}^i [a_j<a_i]~a_j$ ， $c_i=a_i \times \sum_{j=2}^i [a_j<a_i]~b_j$ 。
然后答案就是 $c_i$ 的和。离散后树状数组统计就好了。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef pair<int,int> abcd;
#define fi first
#define se second
const int N=300010;
const int M=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 void Read(int& x){
    char c=nc();
    for(;c<'0'||c>'9';c=nc());
    for(x=0;c>='0'&&c<='9';x=(x<<3)+(x<<1)+c-48,c=nc());
}
int j,k,n,m,a[N];
int w[N];
int num,Ans;
int s[2][N],b[N];
abcd c[N];
inline void Add(int& x,int y){
    x=(x+y)%M;
}
inline int Query(int x,bool d){
    int Ans=0;
    for(;x;x-=x&-x)Add(Ans,s[d][x]);
    return Ans;
}
inline void Update(int x,int y,bool d){
    for(;x<=n;x+=x&-x)Add(s[d][x],y);
}
int main(){
    Read(n);
    for(int i=1;i<=n;i++)Read(a[i]),c[i].fi=a[i],c[i].se=i;
    sort(c+1,c+n+1);
    w[c[1].se]=num=1;
    for(int i=2;i<=n;i++)
    if(c[i].fi==c[i-1].fi)w[c[i].se]=num;else w[c[i].se]=++num;
    for(int i=1;i<=n;i++){
        Add(Ans,1ll*a[i]*Query(w[i]-1,1)%M);
        Update(w[i],1ll*a[i]*Query(w[i]-1,0)%M,1);
        Update(w[i],a[i],0);
    }
    cout<<Ans<<endl;
    return 0;
}