【BZOJ4889】不勤劳的图书管理员（树套树）

【BZOJ4889】不勤劳的图书管理员（树套树）

题面

又是权限题，烦死了
洛谷真好

题解

分开考虑每一次交换产生的贡献。
假设交换\((x,y)\)
检查\(x\)与\(y\)对于区间\([x+1,y-1]\)产生的贡献
再考虑一下\(x,y\)之间的贡献就好啦。
区间的贡献显然是区间内比他小的数的个数乘上当前位置的权值
以及区间内比他小的数的权值和，树套树实现。
具体的再说清楚这题的意思，因为我一开始理解错了。
逆序对的含义是关于\(a_i\)的逆序对
交换是交换第\(i\)个数和第\(j\)个数。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MOD 1000000007
#define MAX 55555
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
void add(int &x,int y){x+=y;if(x>=MOD)x-=MOD;}
int lb(int x){return x&(-x);}
int n,m,V[MAX],ans,p[MAX],a[MAX];
int rt[MAX],tot;
struct Node{int ls,rs,v,s;}t[MAX*200];
void Modify(int &x,int l,int r,int p,int w1,int w2)
{
    if(!x)x=++tot;add(t[x].v,w1);add(t[x].s,w2);
    if(l==r)return;int mid=(l+r)>>1;
    if(p<=mid)Modify(t[x].ls,l,mid,p,w1,w2);
    else Modify(t[x].rs,mid+1,r,p,w1,w2);
}
void Query(int &x,int l,int r,int L,int R,int &s1,int &s2)
{
    if(!x)return;
    if(L<=l&&r<=R){add(s1,t[x].v);add(s2,t[x].s);return;}
    int mid=(l+r)>>1;
    if(L<=mid)Query(t[x].ls,l,mid,L,R,s1,s2);
    if(R>mid)Query(t[x].rs,mid+1,r,L,R,s1,s2);
}
void Modify(int x,int p,int w1,int w2)
{
    for(int i=x;i<=n;i+=lb(i))
        Modify(rt[i],1,n,p,w1,w2);
}
void Query(int l,int r,int L,int R,int &s1,int &s2)
{
    if(L>R)return;int m1=0,m2=0;
    for(int i=r;i;i-=lb(i))Query(rt[i],1,n,L,R,s1,s2);
    for(int i=l-1;i;i-=lb(i))Query(rt[i],1,n,L,R,m1,m2);
    s1=(s1+MOD-m1)%MOD;s2=(s2+MOD-m2)%MOD;
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;++i)p[i]=read(),V[i]=read();
    for(int i=1;i<=n;++i)
    {
        int s1=0,s2=0;
        Modify(i,p[i],V[i],1);
        Query(1,i-1,p[i]+1,n,s1,s2);
        add(ans,s1);add(ans,1ll*s2*V[i]%MOD);
    }
    while(m--)
    {
        int x=read(),y=read(),s1,s2,m1,m2;
        if(x>y)swap(x,y);int a=p[x],b=p[y];
        s1=s2=m1=m2=0;
        Query(x+1,y-1,a+1,n,s1,s2);Query(x+1,y-1,1,a-1,m1,m2);
        add(s1,MOD-m1);add(s2,MOD-m2);
        add(ans,s1);add(ans,1ll*s2*V[x]%MOD);
        s1=s2=m1=m2=0;
        Query(x+1,y-1,1,b-1,s1,s2);Query(x+1,y-1,b+1,n,m1,m2);
        add(s1,MOD-m1);add(s2,MOD-m2);
        add(ans,s1);add(ans,1ll*s2*V[y]%MOD);
        if(a<b)add(ans,V[x]+V[y]);
        if(a>b)add(ans,MOD-(V[x]+V[y]));
        Modify(x,a,MOD-V[x],MOD-1);Modify(y,a,V[x],1);
        Modify(y,b,MOD-V[y],MOD-1);Modify(x,b,V[y],1);
        swap(p[x],p[y]);swap(V[x],V[y]);printf("%d\n",ans);
    }
}