zoukankan      html  css  js  c++  java
  • AcWing 786. 第k个数(快排应用)

    题目链接:https://www.acwing.com/problem/content/description/788/

    解法1:

    递归性质,设计一个查找函数,不断缩小搜索范围,最终找到第k个数字。符合递归函数设计原则。

    解法2:

    二分的思想,每次只对第k个数字所在部分进行排序,最终保证,第k个数处在正确的位置。

    AC代码:

    import java.util.*;
    
    public class Main {
        
        public static int N = (int) 1e5 + 10;
        public static int[] arr = new int[N]; 
        
        public static void main(String[] args) {
            Scanner sc = new Scanner(System.in);
            
            int n = sc.nextInt(), k = sc.nextInt();
            
            for (int i = 0; i < n; i ++) arr[i] = sc.nextInt();
            
            int res = quickchoose(arr, 0, n - 1, k);
            
            System.out.println(res);
        }
        
        // 解法1
        public static int quickchoose(int[] arr, int l, int r, int k) {
            if (l >= r) return arr[l];
            
            int pivot = arr[l + r >> 1];
            
            int i = l - 1, j = r + 1;
            
            while (i < j) {
                do i ++; while(arr[i] < pivot);
                do j --; while(arr[j] > pivot);
                
                if (i < j) {
                    int t = arr[i];
                    arr[i] = arr[j];
                    arr[j] = t;
                }
            }
            
            
            if (k <= (j - l + 1)) return quickchoose(arr, l, j, k);
            else return quickchoose(arr, j + 1, r, k - (j - l + 1));
        }
        
        // 解法2
        public static int quickchoose2(int[] arr, int l, int r, int k) {
            if (l >= r) return arr[k - 1];
            
            int pivot = arr[l + r >> 1];
            
            int i = l - 1, j = r + 1;
            
            while (i < j) {
                do i ++; while(arr[i] < pivot);
                do j --; while(arr[j] > pivot);
                
                if (i < j) {
                    int t = arr[i];
                    arr[i] = arr[j];
                    arr[j] = t;
                }
            }
            
            
            if (k <= j + 1) return quickchoose2(arr, l, j, k);
            else return quickchoose2(arr, j + 1, r, k);
        }
    }
  • 相关阅读:
    洛谷P2466 [SDOI2008]Sue的小球 题解 区间DP+费用提前计算
    中国国家集训队论文集目录(1999-2008)
    洛谷P1726 上白泽慧音 题解 强连通分量
    洛谷P1410 子序列 题解 动态规划
    树堆(Treap)学习笔记 2020.8.12
    伸展树(Splay)学习笔记
    git操作
    yii 缓存探究
    yii之srbac详解
    一个PDO类
  • 原文地址:https://www.cnblogs.com/doubest/p/15314139.html
Copyright © 2011-2022 走看看