您的位置:首页 > 其它

bzoj3289 Mato的文件管理 莫队+树状数组

2017-07-25 21:38 423 查看
求逆序对个数,莫队套树状数组

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 50005
using namespace std;
int n,m,nn,a
,c[5000005],be
,maxn;
struct Query{
int l,r,id,ans;
}qr
;
bool cmp1(Query a,Query b){
if(be[a.l]==be[b.l])
return a.r<b.r;
return be[a.l]<be[b.l];
}
bool cmp2(Query a,Query b){
return a.id<b.id;
}
int lowbit(int x){
return x&(-x);
}
void add(int x,int y){
while(x<=maxn)
{c[x]+=y;x+=lowbit(x);}
}
int query(int x){
int ans=0;
while(x)
{ans+=c[x];x-=lowbit(x);}
return ans;
}
void work(){
int l=1,r=0,tot=0;
for(int i=1;i<=m;i++){
while(l<qr[i].l){
tot-=query(a[l]-1);
add(a[l++],-1);
}
while(l>qr[i].l){
add(a[--l],1);
tot+=query(a[l]-1);
}
while(r<qr[i].r){
add(a[++r],1);
tot+=r-l+1-query(a[r]);
//printf("%d  %d\n",query(a[r]),tot);
}
while(r>qr[i].r){
tot-=r-l+1-query(a[r]);
add(a[r--],-1);
}
qr[i].ans=tot;
}
}
int main()
{
scanf("%d",&n); nn=sqrt(n);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
maxn=max(maxn,a[i]);
be[i]=(i-1)/nn+1;
}
scanf("%d",&m);
int l,r;
for(int i=1;i<=m;i++)
{
scanf("%d%d",&l,&r);
qr[i].l=l; qr[i].r=r;
qr[i].id=i;
}
sort(qr+1,qr+m+1,cmp1);
work();
sort(qr+1,qr+m+1,cmp2);
for(int i=1;i<=m;i++)
printf("%d\n",qr[i].ans);
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: