[BZOJ2674]Attack（整体二分+树状数组套线段树）

题目描述

传送门
数据有坑：
给出的不一定是左下角和右上角，需要自己判断交换。

题解

没见过整体二分里面还有个树套树的…
首先把坐标离散
感觉是K大数查询和矩阵乘法两道题的结合版，就是动态求矩阵的k大值
用bit套线段树的原因是没法写二维bit，于是线段树动态开点
把初始的矩阵和SWAP操作都看做修改，QUERY看成查询
二分第k大的数是mid，做权值小于等于mid的修改，查询的时候如果总数小于k就递归(mid+1,r)，否则(l,mid)
但是这道题和之前的查询K大数不同的是有修改，而不是添加，所以要将每一个修改拆成两个，先把原来的减掉再把新的加上
交换的话要重新记录一下，因为有可能换过的还要继续换

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<map>
using namespace std;
#define N 60005

char s[10];
int n,m,maxn,cnt,lsh[N*10],LSH,sz,ch[N*10],val[N*10];
struct hp{int ord,x,y,p,q,val,k,id,num,ans,f;}opt[N*10];
int root[N*10],sum[N*100],ls[N*100],rs[N*100];
int ans[N];
bool flag[N];

int read()
{
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
int find(int x)
{
    if (!x) return 0;
    int l=1,r=LSH,mid,ans;
    while (l<=r)
    {
        mid=(l+r)>>1;
        if (lsh[mid]>=x) ans=mid,r=mid-1;
        else l=mid+1;
    }
    return ans;
}
void change(int &now,int l,int r,int x,int v)
{
    int mid=(l+r)>>1;
    if (!now) now=++sz;
    sum[now]+=v;
    if (l==r) return;
    if (x<=mid) change(ls[now],l,mid,x,v);
    else change(rs[now],mid+1,r,x,v);
}
int query(int now,int l,int r,int lr,int rr)
{
    if (!now) return 0;
    int mid=(l+r)>>1,ans=0;
    if (lr<=l&&r<=rr) return sum[now];
    if (lr<=mid&&ls[now]) ans+=query(ls[now],l,mid,lr,rr);
    if (mid+1<=rr&&rs[now]) ans+=query(rs[now],mid+1,r,lr,rr);
    return ans;
}
void CHANGE(int x,int y,int z)
{
    for (int i=x;i<=LSH;i+=i&(-i))
        change(root[i],1,LSH,y,z);
}
int QUERY(int x,int y,int z)
{
    int ans=0;
    for (int i=x;i>=1;i-=i&(-i))
        ans+=query(root[i],1,LSH,y,z);
    return ans;
}
void clear(int x)
{
    if (ls[x]) clear(ls[x]);
    if (rs[x]) clear(rs[x]);
    sum[x]=ls[x]=rs[x]=0;
}
void CLEAR(int x)
{
    for (int i=x;i<=LSH;i+=i&(-i))
        if (root[i]) clear(root[i]),root[i]=0;
}
int cmp(hp a,hp b)
{
    return a.num<b.num;
}
void dvd(int l,int r,int x,int y)
{
    if (x>y) return;
    int mid=(l+r)>>1;
    int px=0,py=y-x+1;
    int tot=0;
    for (int i=x;i<=y;++i)
        if (opt[i].ord==1)
        {
            if (opt[i].val<=mid)
            {
                CHANGE(opt[i].x,opt[i].y,opt[i].f);
                ch[++tot]=opt[i].x;
                opt[i].num=++px;
            }
            else opt[i].num=++py;
        }
        else
        {
            int t1=QUERY(opt[i].p,opt[i].y,opt[i].q);
            int t2=QUERY(opt[i].x-1,opt[i].y,opt[i].q);
            opt[i].ans=t1-t2;
            if (opt[i].ans<opt[i].k)
            {
                opt[i].k-=opt[i].ans;
                opt[i].ans=0;
                opt[i].num=++py;
            }
            else opt[i].num=++px;
        }
    for (int i=1;i<=tot;++i)
        CLEAR(ch[i]);
    sz=0;
    if (l==r)
    {
        for (int i=x;i<=y;++i)
            if (opt[i].ord==2)
            {
                if (opt[i].ans>=opt[i].k) ans[opt[i].id]=l;
                else ans[opt[i].id]=-1;
            }
        return;
    }
    sort(opt+x,opt+y+1,cmp);
    dvd(l,mid,x,x+px-1);
    dvd(mid+1,r,x+px,y);
}
int main()
{
    n=read();m=read();
    for (int i=1;i<=n;++i)
    {
        opt[++cnt].x=read()+1;opt[cnt].y=read()+1;val[i]=opt[cnt].val=read();maxn=max(maxn,opt[cnt].val);opt[cnt].f=1;
        lsh[++LSH]=opt[cnt].x;lsh[++LSH]=opt[cnt].y;
        opt[cnt].ord=1;
    }
    for (int i=1;i<=m;++i)
    {
        scanf("%s",s);
        if (s[0]=='S')
        {
            int id=read()+1,jd=read()+1;
            if (id==jd||val[id]==val[jd]) continue;
            opt[++cnt].x=opt[id].x;opt[cnt].y=opt[id].y;opt[cnt].val=val[id];opt[cnt].ord=1;opt[cnt].id=i;opt[cnt].f=-1;
            opt[++cnt].x=opt[id].x;opt[cnt].y=opt[id].y;opt[cnt].val=val[jd];opt[cnt].ord=1;opt[cnt].id=i;opt[cnt].f=1;
            opt[++cnt].x=opt[jd].x;opt[cnt].y=opt[jd].y;opt[cnt].val=val[jd];opt[cnt].ord=1;opt[cnt].id=i;opt[cnt].f=-1;
            opt[++cnt].x=opt[jd].x;opt[cnt].y=opt[jd].y;opt[cnt].val=val[id];opt[cnt].ord=1;opt[cnt].id=i;opt[cnt].f=1;
            swap(val[id],val[jd]);
        }
        else
        {
            flag[i]=1;
            opt[++cnt].x=read()+1;opt[cnt].y=read()+1;opt[cnt].p=read()+1;opt[cnt].q=read()+1;opt[cnt].k=read();
            if (opt[cnt].x>opt[cnt].p) swap(opt[cnt].x,opt[cnt].p);
            if (opt[cnt].y>opt[cnt].q) swap(opt[cnt].y,opt[cnt].q);
            lsh[++LSH]=opt[cnt].x;lsh[++LSH]=opt[cnt].y;lsh[++LSH]=opt[cnt].p;lsh[++LSH]=opt[cnt].q;
            opt[cnt].ord=2;opt[cnt].id=i;
        }
    }
    sort(lsh+1,lsh+LSH+1);LSH=unique(lsh+1,lsh+LSH+1)-lsh-1;
    for (int i=1;i<=cnt;++i) opt[i].x=find(opt[i].x),opt[i].y=find(opt[i].y),opt[i].p=find(opt[i].p),opt[i].q=find(opt[i].q);
    dvd(0,maxn,1,cnt);
    for (int i=1;i<=m;++i)
        if (flag[i])
        {
            if (ans[i]==-1) puts("It doesn't exist.");
            else printf("%d\n",ans[i]);
        }
}