java 最小的K个数

算法与数据结构 专栏收录该内容
5 篇文章 0 订阅

题目描述

输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。

package nowcoder.test1;

import java.util.ArrayList;
import java.util.Arrays;

public class GetLeastNumbers {
    public static ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
        ArrayList<Integer> res = new ArrayList<Integer>();
        if (k == 0 || k > input.length) {
            return res;
        }
        int start = 0, end = input.length - 1;
        int index = partition(input, start, end);
        while (index != k - 1) {
            if (index > k - 1) {
                // 从index前面重新找
                end = index - 1;
                index = partition(input, start, end);
            } else {
                start = index + 1;
                index = partition(input, start, end);
            }
        }
        for (int i = 0; i < k; i++) {
            res.add(input[i]);
        }

        return res;

    }

    private static int partition(int[] a, int start, int end) {
        int tmp = a[start];
        int i = start + 1, j = end;
        while (i <= j) {
            while (a[i] < tmp && i < end) {
                i++;
            }
            while (a[j] > tmp && j > start + 1) {
                j--;
            }
            if (i >= j) {
                break;
            }
            exch(a, i, j);
        }
        exch(a, start, j);

        return j;
    }

    private static void exch(int[] a, int i, int j) {
        int swap = a[i];
        a[i] = a[j];
        a[j] = swap;
    }

    public static void main(String[] args) {
        int[] a = { 9, 3, 15, 18, 22, 16, 5, 12, 7 };
        ArrayList<Integer> list = GetLeastNumbers_Solution(a, 8);
        for (Integer integer : list) {
            System.out.print(integer + " ");
        }
        System.out.println();
        Arrays.sort(a);
        show(a);
    }

    static void show(int[] a) {
        for (int i : a) {
            System.out.print(i + " ");
        }
        System.out.println();
    }
}

  • 0
    点赞
  • 0
    评论
  • 0
    收藏
  • 一键三连
    一键三连
  • 扫一扫,分享海报

©️2021 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值