uva11990 动态逆序对

时间:2022-08-08 20:53:13

这题说的是给了一个数组,按照他给的顺序依次删除数,在删除之前输出此时的逆序对个数

我们用Fenwick树 维护这整个数列, C[i]是一个 treap的头, 管理了在树状数组中 能影响他的点,然后我们用名次树去计算 在C[i]下小于等于要删除的那个数的个数就ok了。

#include <iostream>
#include <algorithm>
#include <string.h>
#include <vector>
#include <cstdio>
#include <stdlib.h>
using namespace std;
const int maxn=;
struct Node
{
Node *ch[];
int r;
int s;
int v;
int vloc;
int cmp(int x)const
{
if(x==v)return -;
return x<v?:;
}
void maintain()
{
s=ch[]->s+ch[]->s+vloc;
}
};
Node *null=new Node();
void rotate(Node *&o, int d)
{
Node *k=o->ch[d^];o->ch[d^]=k->ch[d];k->ch[d]=o;
o->maintain();k->maintain(); o=k;
}
void insert(Node *&o,int x)
{
if(o==null){
o=new Node();
o->ch[]=o->ch[]=null; o->v=x; o->r=rand();o->vloc=;o->s=;
}else{
int d=o->cmp(x);
if(d==-){
o->vloc+=;
}else{
insert(o->ch[d],x);if( o->ch[d]->r > o->r) rotate(o,d^);
}
}
o->maintain();
}
void remove(Node *&o, int x)
{
int d=o->cmp(x);
if(d==-){
if(o->vloc > ){
o->vloc-=;
}else if(o->ch[]==null)o=o->ch[];
else if(o->ch[]==null) o=o->ch[];
else{
int d2 = o->ch[]->r > o->ch[]->r ? : ;
rotate(o,d2); remove(o->ch[d2],x);
}
}else remove(o->ch[d],x);
if(o!=null)
o->maintain();
}
int find(Node *o,int x)
{
int num=;
while(o!=null)
{
int d=o->cmp(x);
if(d==-){
num += o->ch[]->s + o->vloc ;
break;
}else if(d==){
num+=o->ch[]->s+o->vloc;
o=o->ch[];
}else{
o=o->ch[];
}
}
return num;
}
Node *C[maxn];
int cc[maxn],A[maxn];
int lowbit(int x)
{
return (-x)&x;
}
void init(int n)
{
for(int i=; i<=n; i++)
C[i]=null,cc[i]=;
}
int N,M;
void add1(int x, int v)
{
while(x<=N){
cc[x]+=v;
x+=lowbit(x);
}
}
int sum1(int x){
int ans=;
while(x>){
ans+=cc[x];
x-=lowbit(x);
}
return ans;
}
void delet(int x,int v)
{
while(x<=N)
{
remove(C[x],v);
x+=lowbit(x);
}
}
int summ(int x,int v)
{
int ans=;
while(x>)
{
ans+=find(C[x],v);
x-=lowbit(x);
}
return ans;
}
int main()
{
null->s=; while(scanf("%d%d",&N,&M)==)
{
init(N);
long long ans=;
for(int i=; i<=N; i++)
{
int d;
scanf("%d",&d);
A[d]=i;
ans+=sum1(N)-sum1(d);
add1(d,);
int loc=i;
while(loc<=N)
{
insert(C[loc],d); loc+=lowbit(loc);
}
}
for(int i=; i<=M; i++)
{
int d;
printf("%lld\n",ans);
scanf("%d",&d);
int s1xiaoyu=summ(A[d],d);
int ss1=sum1(A[d]);
ans-=(ss1-s1xiaoyu);
int s2xiaoyu=summ(N,d);
ans-=s2xiaoyu-s1xiaoyu;
add1(A[d],-);
delet(A[d],d);
}
}
return ;
}
/*
5 4
1
5
3
4
2
5
1
4
2
*/