bzoj3192 [JLOI2013]删除物品

题目

咋眼一看，这道题就是一道模拟题，不过，应该要加高级数据结构（笑）。

首先，我们可以把两堆堆顶合成一个堆顶就方便处理了，我们来感性认识一下，就把第一堆翻转一下，第二堆接在后面。我们维护一个堆顶，每次答案加上 最大的数到堆顶中数的个数 。而且如果这样的话，移动物品也就不用处理了，一举两得，多好。

#include<bits/stdc++.h>
#define N 100005
using namespace std;
int n1,n2,n,mid,l,r;
long long cnt;
struct seq{
    int num;
    int id;
};seq A[N+1];
struct Tree{
    int l;
    int r;
    int data;
};Tree T[4*N+1];
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    int x=0,b=1;
    char c=nc();
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
}
inline void pushup(int rt)
{
    T[rt].data=T[rt*2].data+T[rt*2+1].data;
}
inline void build(int rt,int l,int r)
{
    T[rt].l=l,T[rt].r=r;
    if(l==r)
    {
        T[rt].data=1;
        return;
    }
    int mid=(l+r)/2;
    build(rt*2,l,mid),build(rt*2+1,mid+1,r);
    pushup(rt);
}
inline void modify(int rt,int pos,int val)
{
    if(T[rt].l==pos&&T[rt].r==pos)
    {
        T[rt].data=val;
        return;
    }
    int mid=(T[rt].l+T[rt].r)/2;
    if(pos<=mid)modify(rt*2,pos,val);
    else modify(rt*2+1,pos,val);
    pushup(rt);
}
inline int qurry(int rt,int l,int r)
{
    if(l>r)return 0;//注意这句
    if(T[rt].l==l&&T[rt].r==r)
        return T[rt].data;
    int mid=(T[rt].l+T[rt].r)/2;
    if(r<=mid)return qurry(rt*2,l,r);
    else if(l>mid)return qurry(rt*2+1,l,r);
    else return qurry(rt*2,l,mid)+qurry(rt*2+1,mid+1,r);
}
inline bool cmp(const seq &A,const seq &B)
{
    return A.num>B.num;
}
int main()
{
    //freopen("in.txt","r",stdin);
    n1=read(),n2=read();
    for(int i=n1;i>=1;i--)A[i].num=read(),A[i].id=i;
    for(int i=n1+1;i<=n1+n2;i++)A[i].num=read(),A[i].id=i;
    n=n1+n2;sort(A+1,A+n+1,cmp);build(1,1,n);
    mid=n1;
    for(int i=1;i<=n;i++)
    {
        l=mid;r=A[i].id;
        if(r>l)cnt+=qurry(1,l+1,r-1);
        else cnt+=qurry(1,r+1,l);
        mid=A[i].id,modify(1,mid,0);
    }
    printf("%lld",cnt);
    return 0;
}


这题树状数组好写很多，线段树我这种写法要注意每次l与r的大小关系，不然无限TTTTTTT。