求序列第K大数 POJ2104

原创 2013年12月05日 18:41:59

在POJ2104中,由于会给定序列的区间,而且是多次查询,所以像快速排序加二分这种分治算法的时间效率就不算很高了,然后自己尝试用二叉搜索树保存数据,粗略的写了个java代码:

 

import java.util.Arrays;
import java.util.Scanner;

/**
 *
 * @author Sot_fzh
 */
public class Poj2104 {

    static class Point {

        int x;
        int left = -1;
        int right = -1;
    }
    static Point[] p;

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int num = scan.nextInt();
        int count = scan.nextInt();
        int[] a = new int[num];
        int[] b = new int[num];
        p = new Point[num];
        for (int i = 0; i < num; i++) {
            a[i] = b[i] = scan.nextInt();
        }
        Arrays.sort(b);
        for (int i = 0; i < num; i++) {
            p[i] = new Point();
            p[i].x = b[i];
        }
        int head = creTree(b, 0, num - 1);

        while (count-- != 0) {
            int low = scan.nextInt() - 1;
            int high = scan.nextInt() - 1;
            int k = scan.nextInt();
            int h = head;
            while (true) {
                int down = 0, up = 0;
                for (int i = low; i <= high; i++) {
                    if (a[i] < p[h].x) {
                        down++;
                    }
                    if (a[i] > p[h].x) {
                        up++;
                    }
                }

                if (down >= k) {            //证明在左子树
                    h = p[h].left;

                } else if (down < k) {                //证明不在左子树
                    if (down==k-1&&up+down+1==high-low+1) {        //如果大于中值点数+小于中值点的数+1==总共数
                        System.out.println(p[h].x);
                        break;
                    }

                    h = p[h].right;
                }
                                  
            }


        }

 


    }

    public static int creTree(int[] b, int low, int high) {
        int mid = (low + high) / 2;
        if (low <= mid - 1) {
            p[mid].left = creTree(b, low, mid - 1);
        }
        if (high >= mid + 1) {
            p[mid].right = creTree(b, mid + 1, high);
        }
        return mid;
    }

    public static void printP(int head) {               //先序遍历
        System.out.print(p[head].x + " ");                
        if (p[head].left != -1) {
            printP(p[head].left);
        }
        if (p[head].right != -1) {
            printP(p[head].right);
        }
    }

    public static void printM(int head) {            //中序遍历
        if (p[head].left != -1) {
            printM(p[head].left);
        }
        System.out.print(p[head].x + " ");
        if (p[head].right != -1) {
            printM(p[head].right);
        }

    }
}

 

由于自己长期怕麻烦又习惯使用java编写程序,经典的树形数据结构其实自己还很少用代码来实现,以上代码又是tle。。

 

但是无意中发现这道题目可以这样做:

设置节点,保存序列时把每个数的原始下标也保存下来,然后进行排序,之后从小到大依次检验节点的下标是否在给定区间中,若存在k--,否则下一个,当k减到0时,证明Kth已经找到。这里根据节点保存的原始下标来检验是否包含在给定区间中,这一点比较巧妙,因为用一个比较大小来代替一个for循环了。

以下代码AC

 

import java.util.Arrays;
import java.util.Scanner;

/**
 *
 * @author Sot_fzh
 */
public class My2104 {

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int len = scan.nextInt();
        int num = scan.nextInt();
        Point[] pa = new Point[len];

        for (int i = 0; i < len; i++) {
            pa[i] = new Point(i, scan.nextInt());
        }
        Arrays.sort(pa);

        while (num-- != 0) {
            int left = scan.nextInt() - 1;
            int right = scan.nextInt() - 1;
            int k = scan.nextInt();
            for (int i = 0; i < len; i++) {             //从小到达遍历排序数组
                if (pa[i].i >= left && pa[i].i <= right) {
                    k--;
                }

                if (k == 0) {
                    System.out.println(pa[i].num);
                    break;
                }

            }
        }
    }

    static class Point implements Comparable{

        int i, num;
        public Point(int i, int num) {
            this.i = i;
            this.num = num;
        }
        @Override
        public int compareTo(Object o) {
            return num-((Point)o).num;
        }
    }
}

使用STL求第K大数

项目上需要使用C++,抽空正好把C++熟悉一下。 期待接下来的这个项目能给我的C++编程能力带来提升。 先来个set版本的。 #include #include #include #i...
  • creater2009
  • creater2009
  • 2012年04月10日 19:36
  • 1273

堆的应用!--求第k大数

求一个数列中的第k大的数,将前k个数建最小堆,后面若比堆顶元素还小,则舍去,否则将堆顶置换成该元素,然后维护堆,最后输出堆顶元素~~ 若求第k小的数,则只需建最大堆即可。 #include int...
  • fengsigaoju
  • fengsigaoju
  • 2015年07月16日 19:48
  • 888

区间K大数查询(求解方法总结)

问题:在一个无序序列中,查找给定区间中的第K大的数 (这是一个很经典的问题,但是之前并没有深究,最近刷题的时候碰到了,就来总结一波……) Method 1:先排序,然后...
  • iwbfaith
  • iwbfaith
  • 2017年02月28日 12:45
  • 320

求第K大数[STL:nth_element]

#include #include #include #include #include #include #include #include #include using namespace std...
  • u012469987
  • u012469987
  • 2014年09月12日 22:24
  • 700

使用分治法求解数组的第k大数

http://blog.csdn.net/hackbuteer1/article/details/6651804 首先,需要在一个数组中,随机抽出一个元素,然后比这个元素大的放到他的右边,比这个元素...
  • miao6664659
  • miao6664659
  • 2012年10月30日 10:58
  • 1972

整体二分\cdq分治——洛谷P3332 [ZJOI2013]K大数查询

https://daniu.luogu.org/problem/show?pid=3332 第一次接触整体二分; 上课的时候小红说这用树套树做,但感觉好难啊,二维线段树都不会,怎么做树套树啊; ...
  • largecub233
  • largecub233
  • 2017年04月06日 11:04
  • 1154

Lq_ 区间k大数查询

算法训练 区间k大数查询   时间限制:1.0s   内存限制:256.0MB 锦囊1 锦囊2 锦囊3 问题描述 给定一个序列,每次询问序列中第l个数到第r个数中第K大的数是哪...
  • briup_acmer
  • briup_acmer
  • 2015年03月24日 15:46
  • 1047

蓝桥杯——区间k大数查询

问题描述 给定一个序列,每次询问序列中第l个数到第r个数中第K大的数是哪个。输入格式 第一行包含一个数n,表示序列长度。第二行包含n个正整数,表示给定的序列。第三个包含一个正整数m,表示询问个数。...
  • SYaoJun
  • SYaoJun
  • 2016年02月17日 23:08
  • 713

poj 2985 并查集+线段树 线段树求第k大数 The k-th Largest Group

题意:        有一群猫,现在给它们编号,然后组队,0 1 2 代表 1 号和 2 号的队伍合并。 1 4 询问第 4 大的队伍 Size 是多大。 解:(贴的,他写的好。表示学习)   ...
  • Julyana_Lin
  • Julyana_Lin
  • 2012年08月26日 16:48
  • 2343

bzoj 1901: Zju2112 Dynamic Rankings(带修改的区间第k大,树状数组+主席树)

果然在zju上交是要
  • u010697167
  • u010697167
  • 2014年07月30日 13:23
  • 1088
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:求序列第K大数 POJ2104
举报原因:
原因补充:

(最多只允许输入30个字)