BZOJ4889 & 洛谷3759:[TJOI2017]不勤劳的图书管理员——题解

https://www.lydsy.com/JudgeOnline/problem.php?id=4889

https://www.luogu.org/problemnew/show/P3759

加里敦大学有个帝国图书馆,小豆是图书馆阅览室的一个书籍管理员。他的任务是把书排成有序的,所以无序的书让他产生厌烦,两本乱序的书会让小豆产生这两本书页数的和的厌烦度。现在有n本被打乱顺序的书,在接下来m天中每天都会因为读者的阅览导致书籍顺序改变位置。因为小豆被要求在接下来的m天中至少要整理一次图书。小豆想知道,如果他前i天不去整理,第i天他的厌烦度是多少,这样他好选择厌烦度最小的那天去整理。

最开始以为看的题解很naive,后来发现是自己太naive了。

看起来就像是查逆序对,对于l和r的交换,其贡献的变更只与[l+1,r-1]这些区间有关,可以变成查询这段区间内有多少数满足与l和r构成逆序对。

这个明显可以主席树做。

然而之后你需要交换这两个数……emmm待修改主席树呗。

其实就是树状数组套一个动态开点线段树(因为本质都差不多就都叫成主席树好了。)

剩下的就是码码码了。

(注意取模对常数的影响。)

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=5e4+5;
const int p=1e9+7;
inline int read(){
    int X=0,w=0;char ch=0;
    while(ch<'0'||ch>'9'){w|=ch=='-';ch=getchar();}
    while(ch>='0'&&ch<='9')X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
struct tree{
    int l,r;
    ll v,c;
}tr[N*200];
int rt[N],pool,n,m;
ll ans,a[N],v[N],b[N],cnt[N];
inline int lowbit(int x){return x&(-x);}
inline void add(int x,int y){
    for(int i=x;i<=n;i+=lowbit(i))b[i]+=y,cnt[i]++;
}
inline ll qry_val(int x){
    ll res=0;
    for(int i=x;i;i-=lowbit(i))res+=b[i];
    return res;
}
inline ll qry_cnt(int x){
    ll res=0;
    for(int i=x;i;i-=lowbit(i))res+=cnt[i];
    return res;
}
inline void insert(int &x,int l,int r,int pos,int v,int w){
    if(!x)x=++pool;
    tr[x].v+=v;tr[x].c+=w;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(pos<=mid)insert(tr[x].l,l,mid,pos,v,w);
    else insert(tr[x].r,mid+1,r,pos,v,w);
}
inline void ins(int i,int x,int y,int w){
    for(;i<=n;i+=lowbit(i))insert(rt[i],1,n,x,y,w);
}
inline ll sqv(int x,int l,int r,int l1,int r1){
    if(r<l1||r1<l)return 0;
    if(l1<=l&&r<=r1)return tr[x].v;
    int mid=(l+r)>>1;
    return sqv(tr[x].l,l,mid,l1,r1)+sqv(tr[x].r,mid+1,r,l1,r1);
}
inline ll sqc(int x,int l,int r,int l1,int r1){
    if(r<l1||r1<l)return 0;
    if(l1<=l&&r<=r1)return tr[x].c;
    int mid=(l+r)>>1;
    return sqc(tr[x].l,l,mid,l1,r1)+sqc(tr[x].r,mid+1,r,l1,r1);
}
inline ll query_val(int l,int r,int l1,int r1){
    if(l>r||l1>r1)return 0;
    l--;ll res=0;
    for(int i=r;i;i-=lowbit(i))res=(res+sqv(rt[i],1,n,l1,r1))%p;
    for(int i=l;i;i-=lowbit(i))res=(res-sqv(rt[i],1,n,l1,r1))%p;
    return (res+p)%p;
}
inline ll query_cnt(int l,int r,int l1,int r1){
    if(l>r||l1>r1)return 0;
    l--;ll res=0;
    for(int i=r;i;i-=lowbit(i))res=(res+sqc(rt[i],1,n,l1,r1))%p;
    for(int i=l;i;i-=lowbit(i))res=(res-sqc(rt[i],1,n,l1,r1))%p;
    return (res+p)%p;
}
inline void check(ll &x){
    while(x>=p)x-=p;
    while(x<0)x+=p;
}
int main(){
    n=read(),m=read();
    for(int i=1;i<=n;i++){
        a[i]=read(),v[i]=read();
        ins(i,a[i],v[i],1);
    }
    for(int i=n;i>=1;i--){
        add(a[i],v[i]);
        ans=(ans+qry_val(a[i]-1)+qry_cnt(a[i]-1)*v[i])%p;
    }
    for(int i=1;i<=m;i++){
        int l=read(),r=read();
        if(l==r){printf("%lld\n",ans);continue;}
        if(l>r)swap(l,r);
        ans=ans+query_val(l+1,r-1,1,a[r]-1);check(ans);
        ans=ans+query_cnt(l+1,r-1,1,a[r]-1)*v[r];check(ans);
        ans=ans-query_val(l+1,r-1,a[r]+1,n);check(ans);
        ans=ans-query_cnt(l+1,r-1,a[r]+1,n)*v[r];check(ans);
        ans=ans+query_val(l+1,r-1,a[l]+1,n);check(ans);
        ans=ans+query_cnt(l+1,r-1,a[l]+1,n)*v[l];check(ans);
        ans=ans-query_val(l+1,r-1,1,a[l]-1);check(ans);
        ans=ans-query_cnt(l+1,r-1,1,a[l]-1)*v[l];check(ans);
        if(a[l]>a[r])ans-=v[l]+v[r];
        else ans+=v[l]+v[r];
        check(ans);
        ins(l,a[l],-v[l],-1);ins(r,a[r],-v[r],-1);
        swap(a[l],a[r]);swap(v[l],v[r]);
        ins(l,a[l],v[l],1);ins(r,a[r],v[r],1);
        printf("%lld\n",ans);
    }
    return 0;
}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者:luyouqi233。               +

+欢迎访问我的博客:http://www.cnblogs.com/luyouqi233/ +

+++++++++++++++++++++++++++++++++++++++++++

posted @ 2018-04-30 14:47  luyouqi233  阅读(330)  评论(0编辑  收藏  举报