zoukankan      html  css  js  c++  java
  • BZOJ 3110 [ZJOI2013]K大数查询 (整体二分+线段树)

    dynamic rankings这道题的思想一样

    只不过是把树状数组换成线段树区间修改,求第$K$大的而不是第$K$小的

    这道题还有负数,需要离散

     1 #include <vector>
     2 #include <cstdio>
     3 #include <cstring>
     4 #include <algorithm>
     5 #define N1 50500
     6 #define M1 100500
     7 #define ll long long
     8 #define dd double
     9 #define inf 233333333
    10 using namespace std;
    11 
    12 int gint()
    13 {
    14     int ret=0,fh=1;char c=getchar();
    15     while(c<'0'||c>'9'){if(c=='-')fh=-1;c=getchar();}
    16     while(c>='0'&&c<='9'){ret=ret*10+c-'0';c=getchar();}
    17     return ret*fh;
    18 }
    19 ll gll()
    20 {
    21     ll ret=0;int fh=1; char c=getchar();
    22     while(c<'0'||c>'9'){if(c=='-')fh=-1;c=getchar();}
    23     while(c>='0'&&c<='9'){ret=ret*10+c-'0';c=getchar();}
    24     return ret*fh;
    25 }
    26 
    27 struct SEG{
    28 ll sum[N1<<2],tag[N1<<2];int use[N1<<2],que[N1<<2],tl;
    29 void pushdown(int l,int r,int rt)
    30 {
    31     if(!tag[rt]) return;
    32     int mid=(l+r)>>1;
    33     sum[rt<<1]+=tag[rt]*(mid-l+1); tag[rt<<1]+=tag[rt];
    34     sum[rt<<1|1]+=tag[rt]*(r-mid); tag[rt<<1|1]+=tag[rt];
    35     if(!use[rt<<1]) use[rt<<1]=1,que[++tl]=rt<<1;
    36     if(!use[rt<<1|1]) use[rt<<1|1]=1,que[++tl]=rt<<1|1;
    37     tag[rt]=0;
    38 }
    39 void pushup(int rt){ sum[rt]=sum[rt<<1]+sum[rt<<1|1]; }
    40 void update(int L,int R,int l,int r,int rt,int w)
    41 {
    42     if(!use[rt]) use[rt]=1,que[++tl]=rt;
    43     if(L<=l&&r<=R){ sum[rt]+=(r-l+1)*w; tag[rt]+=w; return;}
    44     int mid=(l+r)>>1; pushdown(l,r,rt);
    45     if(L<=mid) update(L,R,l,mid,rt<<1,w);
    46     if(R>mid) update(L,R,mid+1,r,rt<<1|1,w);
    47     pushup(rt); 
    48 }
    49 ll query(int L,int R,int l,int r,int rt)
    50 {
    51     if(L<=l&&r<=R) return sum[rt];
    52     int mid=(l+r)>>1; ll ans=0; pushdown(l,r,rt);
    53     if(L<=mid) ans+=query(L,R,l,mid,rt<<1);
    54     if(R>mid) ans+=query(L,R,mid+1,r,rt<<1|1);
    55     return ans;
    56 }
    57 void clr(){ while(tl){ use[que[tl]]=0; sum[que[tl]]=0; tag[que[tl]]=0; que[tl--]=0; } }
    58 }s;
    59 struct node{int fl,l,r,ans,t;ll K;}a[N1],tmp[N1];
    60 int n,m,ma,nn;
    61 int f[N1],w[N1],use[N1];
    62 
    63 void alldic(int l,int r,int ql,int qr)
    64 {
    65     if(l>r||ql>qr) return;
    66     int mid=(l+r)>>1,i,j,S=ql,E;ll sum;
    67     for(i=ql;i<=qr;i++)
    68     {
    69         if(a[i].fl==1){
    70             if(a[i].K>=mid){ s.update(a[i].l,a[i].r,1,n,1,1); use[i]=1; }
    71             if(a[i].K>mid){ tmp[S++]=a[i]; }
    72         }else{
    73             sum=s.query(a[i].l,a[i].r,1,n,1);
    74             if(sum<a[i].K){ a[i].K-=sum; }
    75             else{ f[a[i].t]=w[mid]; tmp[S++]=a[i]; use[i]=1; }
    76         }
    77     }
    78     for(i=ql,E=S;i<=qr;i++)
    79     {
    80         if(!use[i]){ tmp[E++]=a[i]; }
    81         else{ use[i]=0; }
    82     }
    83     for(i=ql;i<=qr;i++) a[i]=tmp[i];
    84     s.clr();
    85     alldic(mid+1,r,ql,S-1); alldic(l,mid-1,S,E-1);
    86 } 
    87 
    88 int main()
    89 {
    90     scanf("%d%d",&n,&m);
    91     int i,j; memset(f,-1,sizeof(f));
    92     for(i=1;i<=m;i++){ a[i].fl=gint(); a[i].l=gint(); a[i].r=gint(); a[i].K=gll(); a[i].t=i; if(a[i].fl==1) w[++ma]=a[i].K; }
    93     sort(w+1,w+ma+1); ma=unique(w+1,w+ma+1)-(w+1); 
    94     for(i=1;i<=m;i++){ if(a[i].fl==1) a[i].K=lower_bound(w+1,w+ma+1,a[i].K)-w; }
    95     alldic(1,ma,1,m);
    96     for(i=1;i<=m;i++) if(f[i]!=-1) printf("%d
    ",f[i]);
    97     return 0;
    98 }
  • 相关阅读:
    java数据结构之队列
    java数据结构之线性表
    habase 报错 ERROR: Can't get master address from ZooKeeper; znode data == null
    java.io.IOException: Could not find status of job:job_1534233312603_0002
    windows本地调试安装hadoop(idea) : ERROR util.Shell: Failed to locate the winutils binary in the hadoop binary path
    搭建hadoop_之 创建3个虚拟机配置好网络
    java 数组
    java 中break 和continue 的非常规用法
    java运算符
    java数据类型
  • 原文地址:https://www.cnblogs.com/guapisolo/p/10219491.html
Copyright © 2011-2022 走看看