【模板整合】整体二分模板

来源:互联网 发布:ps淘宝调色 编辑:程序博客网 时间:2024/06/05 05:32

模板题目:ZJOI2013 K大数查询原题是带插入区间第K大
注释足够看→_←

#include<iostream>#include<cstdio>#include<cstring>#include<cmath>#include<algorithm>#define MAXN 50010#define MAXINT 0x7fffffff#define lchild rt<<1,l,mid#define rchild rt<<1|1,mid+1,r#define ln rt<<1#define rn rt<<1|1using namespace std;int op,a,b,c;int n,m;struct query{    int x;    int op,a,b,c;    int fin;//操作是否得到解决     bool operator < (const query& a)const{        if (fin!=a.fin) return fin<a.fin;        return x<a.x;    }}ques[MAXN];//询问序列int ans[MAXN];//答案序列,如果对应询问是插入,ans为-MAXINTstruct seg{    int l,r;    int mem;//清空标记     int flag;//操作标记    int add;//加量标记     int sum;}tree[MAXN<<4];bool comp(query a,query b){    if (a.fin<b.fin) return 1;    if (a.fin==b.fin) a.x<b.x;    return 0;}void build(int rt=1,int l=1,int r=n){    int mid=(l+r)>>1;    tree[rt].l=l;tree[rt].r=r;    if (l==r) return;    build(lchild);build(rchild);}void push_up(int rt){    tree[rt].sum=tree[ln].sum+tree[rn].sum;}void push_down(int rt){    if (tree[rt].mem)    {        tree[ln].add=tree[rn].add=tree[ln].sum=tree[rn].sum=0;        tree[ln].mem=tree[rn].mem=1;        tree[rt].mem=0;    }    if (tree[rt].add)    {        tree[ln].add+=tree[rt].add;tree[rn].add+=tree[rt].add;        tree[ln].sum+=(tree[ln].r-tree[ln].l+1)*tree[rt].add;        tree[rn].sum+=(tree[rn].r-tree[rn].l+1)*tree[rt].add;        tree[rt].add=0;    }}void modify(int rt,int l,int r){    push_down(rt);    int L=tree[rt].l,R=tree[rt].r;    if (L==l&&R==r)    {        tree[rt].add++;        tree[rt].sum+=(R-L+1);        return;    }    int mid=(L+R)>>1;    if (r<=mid) modify(ln,l,r);    else    if (l>mid) modify(rn,l,r);    else modify(ln,l,mid),modify(rn,mid+1,r);    push_up(rt);}int query(int rt,int l,int r){    push_down(rt);    int L=tree[rt].l,R=tree[rt].r;    if (L==l&&R==r) return tree[rt].sum;    int mid=(L+R)>>1;    if (r<=mid) return query(ln,l,r);    else    if (l>mid) return query(rn,l,r);    else return query(ln,l,mid)+query(rn,mid+1,r);}void solve(int L,int R,int l,int r)//L,R是数据序列,l,r是询问序列{    if (l>r) return;    if (L==R)//答案get     {        for (int i=l;i<=r;i++)            if (ques[i].op==2) ans[ques[i].x]=L;        return;    }    tree[1].add=tree[1].sum=0;tree[1].mem=1;    int Mid=l-1;//用于确定下一次划分询问区间的端点    int mid=(L+R)>>1;     for (int i=l;i<=r;i++)    {        if (ques[i].op==1)        {            if (ques[i].c>mid)                modify(1,ques[i].a,ques[i].b),ques[i].fin=1;            else                ques[i].fin=0,Mid++;        }        else        {            int temp=query(1,ques[i].a,ques[i].b);            if (ques[i].c<=temp) ques[i].fin=1;            else ques[i].fin=0,Mid++,ques[i].c-=temp;        }    }    sort(ques+l,ques+r+1);    solve(L,mid,l,Mid);    solve(mid+1,R,Mid+1,r);} int main(){    scanf("%d%d",&n,&m);    build();    for (int i=1;i<=MAXN;i++) ans[i]=-MAXINT;    for (int i=1;i<=m;i++)    {        scanf("%d%d%d%d",&op,&a,&b,&c);        ques[i].x=i;ques[i].op=op;ques[i].a=a;ques[i].b=b;ques[i].c=c;    }    solve(0,n,1,m);    for (int i=1;i<=m;i++)        if (ans[i]!=-MAXINT) printf("%d\n",ans[i]);}
2 1
原创粉丝点击