程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> poj 2104 K-th Number

poj 2104 K-th Number

編輯:C++入門知識

題目思路:劃分樹求區間第k大數。

[cpp] 
#include<stdio.h> 
#include<stdlib.h> 
#include<string.h> 
#include<string> 
#include<queue> 
#include<algorithm> 
#include<vector> 
#include<stack> 
#include<list> 
#include<iostream> 
#include<map> 
using namespace std; 
#define inf 0x3f3f3f3f 
#define M 110000 
int max(int a,int b) 

    return a>b?a:b; 

int min(int a,int b) 

    return a<b?a:b; 

int tr[20][M],s[20][M]; 
int sorted[M]; 
void build(int d,int l,int r) 

    if(l==r) return ; 
    int i,j,k,mid; 
    mid=(l+r)>>1,j=l,k=mid+1; 
    for(i=l;i<=r;i++) 
    { 
        s[d][i]=s[d][i-1]; 
        if(tr[d][i]<=sorted[mid]) 
        { 
            s[d][i]++; 
            tr[d+1][j++]=tr[d][i]; 
        } 
        else 
        { 
            tr[d+1][k++]=tr[d][i]; 
        } 
    } 
    build(d+1,l,mid); 
    build(d+1,mid+1,r); 

int getkth(int d,int lp,int rp,int l,int r,int k) 

    int lt,rt; 
    if(lp==rp) return tr[d][lp]; 
    int mid=(lp+rp)>>1; 
    if(k<=s[d][r]-s[d][l-1]) 
    { 
        lt=lp+s[d][l-1]-s[d][lp-1]; 
        rt=lp+s[d][r]-s[d][lp-1]-1; 
        return getkth(d+1,lp,mid,lt,rt,k); 
    } 
    else 
    { 
        lt=mid+1+(l-lp)-(s[d][l-1]-s[d][lp-1]); 
        rt=mid+(r-lp+1)-(s[d][r]-s[d][lp-1]); 
        return getkth(d+1,mid+1,rp,lt,rt,k-(s[d][r]-s[d][l-1])); 
    } 

int main() 

    int i,n,m,l,r,k; 
    while(scanf("%d%d",&n,&m)!=EOF) 
    { 
        for(i=1;i<=n;i++) 
        { 
            scanf("%d",&tr[0][i]); 
            sorted[i]=tr[0][i]; 
        } 
        sort(sorted+1,sorted+n+1); 
        for(i=0;i<20;i++) 
            s[i][0]=0; 
        build(0,1,n); 
      //  for(i=1;i<=n;i++) 
     //   { 
      //      printf("i %d s %d\n",i,s[0][i]); 
      //  } 
        while(m--) 
        { www.2cto.com
            scanf("%d%d%d",&l,&r,&k); 
            printf("%d\n",getkth(0,1,n,l,r,k)); 
        } 
    } 

 

作者:Wings_of_Liberty

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved