hdu 2665 Kth number || poj 2761 Feed the dogs || poj 2104 K-th Number

都是划分树加线段树。



划分树定义为,它的每一个节点保存区间[lft,rht]所有元素,元素顺序与原数组(输入)相同,但是,两个子树的元素为该节点所有元素排序后(rht-lft+1)/2个进入左子树,其余的到右子树,同时维护一个num域,num[i]表示lft->i这个点有多少个进入了左子树。

这个输入的数据不可以重复。。

三个代码基本一样,就只有输入不同,第二个和第三个一样:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100010
int sorted[N];
int anode[40][N];
int line[40][N];
void bulid(int l,int r,int n){
    if(l==r) return;
    int mid=(l+r)>>1;
    int s=mid-l+1;
    for(int i=l;i<=r;i++)
        if(anode[n][i]<sorted[mid])
        s--;
        int left=l;
        int right=mid+1;
    for(int i=l;i<=r;i++){
        if(anode[n][i]<sorted[mid])
         anode[n+1][left++]=anode[n][i];
         else if(anode[n][i]==sorted[mid]&&s>0){
                anode[n+1][left++]=anode[n][i];
                s--;
                }
        else 
        anode[n+1][right++]=anode[n][i];
        line[n][i]=line[n][l-1]+left-l;         
        }
        bulid(l,mid,n+1);
        bulid(mid+1,r,n+1);
    }
int query(int L,int R,int l,int r,int n,int k){
    if(l==r) return anode[n][l];
    int mid=(L+R)>>1;
    int num=line[n][r]-line[n][l-1];
    if(num>=k){
        int nn=L+line[n][l-1]-line[n][L-1];
        int mm=nn+num-1;
        return query(L,mid,nn,mm,n+1,k);
        }
    else{
        int nn=r+line[n][R]-line[n][r];
        int mm=nn-(r-l-num);
        return query(mid+1,R,mm,nn,n+1,k-num);
        }    
    }    
int main(){
    int t,i,n,m,a,b,c;
    scanf("%d",&t);
    while(t--){
        scanf("%d %d",&n,&m);
        memset(anode,0,sizeof(anode)); 
        for(i=1;i<=n;i++){
            scanf("%d",&anode[0][i]);
            sorted[i]=anode[0][i];
            }
        sort(sorted+1,sorted+1+n);
        bulid(1,n,0);
        for(i=0;i<m;i++){
            scanf("%d %d %d",&a,&b,&c);
            printf("%d\n",query(1,n,a,b,0,c));
            }    
        }
        return 0;
    }

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 100010
int sorted[N];
int anode[40][N];
int line[40][N];
void bulid(int l,int r,int n){
    if(l==r) return;
    int mid=(l+r)>>1;
    int s=mid-l+1;
    for(int i=l;i<=r;i++)
        if(anode[n][i]<sorted[mid])
        s--;
        int left=l;
        int right=mid+1;
    for(int i=l;i<=r;i++){
        if(anode[n][i]<sorted[mid])
         anode[n+1][left++]=anode[n][i];
         else if(anode[n][i]==sorted[mid]&&s>0){
                anode[n+1][left++]=anode[n][i];
                s--;
                }
        else 
        anode[n+1][right++]=anode[n][i];
        line[n][i]=line[n][l-1]+left-l;         
        }
        bulid(l,mid,n+1);
        bulid(mid+1,r,n+1);
    }
int query(int L,int R,int l,int r,int n,int k){
    if(l==r) return anode[n][l];
    int mid=(L+R)>>1;
    int num=line[n][r]-line[n][l-1];
    if(num>=k){
        int nn=L+line[n][l-1]-line[n][L-1];
        int mm=nn+num-1;
        return query(L,mid,nn,mm,n+1,k);
        }
    else{
        int nn=r+line[n][R]-line[n][r];
        int mm=nn-(r-l-num);
        return query(mid+1,R,mm,nn,n+1,k-num);
        }    
    }    
int main(){
    int t,i,n,m,a,b,c;
    //scanf("%d",&t);
    while(~scanf("%d %d",&n,&m)){
      //  scanf("%d %d",&n,&m);
        memset(anode,0,sizeof(anode)); 
        for(i=1;i<=n;i++){
            scanf("%d",&anode[0][i]);
            sorted[i]=anode[0][i];
            }
        sort(sorted+1,sorted+1+n);
        bulid(1,n,0);
        for(i=0;i<m;i++){
            scanf("%d %d %d",&a,&b,&c);
            printf("%d\n",query(1,n,a,b,0,c));
            }    
        }
        return 0;
    }


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值