线段树基础 poj2104

题目链接

题目描述

有N个数，1~N，存在数列a[1~N]中（无序），输入i，j，k。求a[i ~ j] 之间第K小数。如 数列：1 5 2 6 3 7 4. 输入2 5 4 .输出：6

思路

1. 用一个结构体存储输入元素的值和当前所在位置
2. 将结构体按照输入元素值排序
3. 从1 ~ N遍历每个元素，统计位置在i，j之间的元素个数，统计到第K个，这个位置对应的元素的值就是所求答案
   

代码

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

struct Node{
  int pos,val;
}nodes[110000];

bool cmp(const Node n1, const Node n2){
    return n1.val < n2.val;
}

int main(){
    int N,M;
    scanf("%d%d",&N,&M);
    for(int i = 1; i<=N; i++){
        scanf("%d",&nodes[i].val);
        nodes[i].pos = i;
    }
    sort(nodes+1, nodes+N+1, cmp);
    while(M--){
        int a,b;
        scanf("%d%d",&a,&b);
        int cnt;
        scanf("%d",&cnt);
        int i;
        int res = 0;
        for(i = 1; i<=N; i++){
            if(a<=nodes[i].pos && nodes[i].pos<=b){
                res++;
                if(res == cnt) break;
            }
        }
        printf("%d\n",nodes[i].val);
    }
    return 0;
}