码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2015-03-29 15:06:45      阅读:157      评论:0      收藏:0      [点我收藏+]

标签:带插入区间k大   整体二分   

模板题目: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|1
using 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为-MAXINT
struct 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]);
}

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

标签:带插入区间k大   整体二分   

原文地址:http://blog.csdn.net/creationaugust/article/details/44727217

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!