【BZOJ】3295 [Cqoi2011]动态逆序对 树状数组+线段树

来源:互联网 发布:爱淘宝首页登录 编辑:程序博客网 时间:2024/05/14 21:47

题目传送门
这题不是主席树!这题不是主席树!这题不是主席树!(重要的事情说三遍!!!)

因为这题并不需要用到历史版本,所以我们不用主席树。

这题的解题思路是树状数组套权值线段树+动态开点,并不是百度上大多数人说的主席树,还是hzwer大佬说的最正确了。

回到这题,考虑删除一个数a[i],当前答案要减少1~i-1中比a[i]大的数的个数和i+1~n中比a[i]小的数的个数。

显然这个可以用权值线段树轻松解决。

因为有修改,所以在权值线段树外面套一个树状数组来快速维护前缀和。

又因为如果把所有的节点全部建出来会爆内存,所以我们要动态开点。

附上AC代码:

#include <cstdio>#include <cctype>#define lowbit(x) ((x)&(-x))#define mid (l+r>>1)using namespace std;const int N=1e5+10;struct note{    int lt,rt,sum;}t[10000010];int n,m,a[N],x,root[N],size,wz;long long ans,s[N];inline char nc(void){    static char ch[100010],*p1=ch,*p2=ch;    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;}inline void read(int &a){    static char c=nc();int f=1;    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());    a*=f;return;}inline void updata(int x){for (int i=x; i<=n; i+=lowbit(i)) ++s[i]; return;}inline int ask(int x){int sum=0; for (int i=x; i; i-=lowbit(i)) sum+=s[i]; return sum;}//----------树状数组求逆序对---------- inline void change(int &k,int l,int r,int wz,int w){    t[!k?k=++size:k].sum+=w;    if (l==r) return;    if (wz<=mid) change(t[k].lt,l,mid,wz,w);    else change(t[k].rt,mid+1,r,wz,w);    return;}inline int query(int k,int l,int r,int wz){    if (l==r) return t[k].sum;    if (wz<=mid) return query(t[k].lt,l,mid,wz);    else return t[t[k].lt].sum+query(t[k].rt,mid+1,r,wz);}//----------权值线段树----------inline void ist(int x,int wz,int w){for (int i=x; i<=n; i+=lowbit(i)) change(root[i],1,n,wz,w);}inline int get(int x,int wz){int sum=0;for (int i=x; i; i-=lowbit(i)) sum+=query(root[i],1,n,wz); return sum;}inline int pre(int x,int wz){return get(wz,n)-get(wz,x);}inline int suc(int x,int wz){return get(n,x)-get(wz,x);}//----------树状数组----------int main(void){    read(n),read(m);    for (int i=1; i<=n; ++i) read(x),a[x]=i,ans+=ask(n)-ask(x),updata(x),ist(i,x,1);    while (m--) printf("%lld\n",ans),read(x),wz=a[x],ans-=pre(x,wz)+suc(x,wz),ist(wz,x,-1);    return 0;}
阅读全文
0 0
原创粉丝点击