2018SD省队集训R2 D7

T1

题解

我们可以用权值线段树来弄这个。

具体来说，我们对于读入的每一个正数建立一棵权值线段树，对于读入的0进行线段树合并，对于读入的负数进行线段树乘法

那么怎么统计逆序对数量呢？线段树合并的时候用第二棵左子树size*第一棵右子树size就好了，问题是在线段树乘法的时候怎么计算逆序对数量呢。

复制的话，逆序对的个数会变成：原来的逆序对数 * 倍数+所有的左子树 * 右子树 * 倍数 * （倍数-1）/2

（可以考虑成1+…+n-1，对于一个右子树的节点来说，后面的n-1个左子树都要产生贡献，那么对于n-1的右子树节点，后面n-2个左子树都要产生贡献，以此类推）

这个所有的左子树 * 右子树 怎么计算呢？就是$(\sum_{i!=j}cnt[i]*cnt[j])/2$，除以2是因为(i,j)，(j,i)算了两次

然后$(\sum_{i!=j}cnt[i]*cnt[j])/2=((\sum cnt[i])^2-\sum cnt[i]^2)/2$，那么我们只需要在权值线段树上维护平方就好了。

代码

#include <cstdio>
#include <cstring>
#include <iostream>
#define LL long long 
using namespace std;
const int N=300000;
const int M=N*20;
const int mod=1e9+7;
int ls[M],rs[M],sz,num,root[N],tmp,he[M],pfh[M],deltac[M],ans[N];
LL ksm(LL a,LL k)
{
    LL ans=1;
    for (;k;k>>=1,a=a*a%mod)
      if (k&1) ans=ans*a%mod;
    return ans;
}
void updata(int now)
{
    he[now]=(he[ls[now]]+he[rs[now]])%mod;
    pfh[now]=(pfh[ls[now]]+pfh[rs[now]])%mod;
}
void cf(int k,int x)
{
    he[k]=(LL)he[k]*x%mod;
    pfh[k]=(LL)pfh[k]*x%mod*x%mod;
    deltac[k]=(LL)deltac[k]*x%mod;
}
void pushdown(int now)
{
    if (deltac[now]!=1)
    {
        if (ls[now]) cf(ls[now],deltac[now]);
        if (rs[now]) cf(rs[now],deltac[now]);
        deltac[now]=1;
    }
}
void insert(int &now,int l,int r,int x)
{
    if (!now) now=++sz,deltac[now]=1;
    if (l==r) {he[now]=pfh[now]=1;return;}
    int mid=(l+r)>>1;
    if (x<=mid) insert(ls[now],l,mid,x);
    else insert(rs[now],mid+1,r,x);
    updata(now);
}
int merge(int x,int y,int l,int r)
{
    if (!x || !y) return x+y;
    if (l==r)
    {
        he[x]=(he[x]+he[y])%mod;
        pfh[x]=(LL)he[x]*he[x]%mod;
        return x;
    }
    pushdown(x); pushdown(y);
    ans[num-1]=(ans[num-1]+(LL)he[rs[x]]*he[ls[y]]%mod)%mod;
    int mid=(l+r)>>1;
    ls[x]=merge(ls[x],ls[y],l,mid);
    rs[x]=merge(rs[x],rs[y],mid+1,r);
    updata(x);
    return x;
}
int main()
{
    freopen("bubble.in","r",stdin);
    freopen("bubble.out","w",stdout);
    int n;scanf("%d",&n);LL inv2=ksm(2,mod-2);
    for (int i=1;i<=n;i++)
    {
        int x;scanf("%d",&x);
        if (x>0)
        {
            root[++num]=ans[num]=0; 
            insert(root[num],1,N,x);
        }else if (x==0)
        {
            tmp=0;
            root[num-1]=merge(root[num-1],root[num],1,N);
            ans[num-1]=(ans[num-1]+ans[num])%mod;num--;
        }
        else
        {
            x=-x;int k=root[num];
            ans[num]=((LL)ans[num]*x%mod+((LL)he[k]*he[k]-pfh[k]+mod)%mod*(mod+1)%mod*inv2%mod*((LL)x*(x-1)%mod*inv2%mod))%mod;
            cf(k,x);
        }
    }
    printf("%d",ans[root[1]]);
}