第k小的数值对

2021-10-20 22:03:59 阅读：311 来源： 互联网

长度为N的数组arr，一定可以组成N^2个数值对。
例如arr = [3,1,2]，
数值对有(3,3) (3,1) (3,2) (1,3) (1,1) (1,2) (2,3) (2,1) (2,2)，
也就是任意两个数都有数值对，而且自己和自己也算数值对。
数值对怎么排序？规定，第一维数据从小到大，第一维数据一样的，第二维数组也从小到大。所以上面的数值对排序的结果为：
(1,1)(1,2)(1,3)(2,1)(2,2)(2,3)(3,1)(3,2)(3,3)

给定一个数组arr，和整数k，返回第k小的数值对。

public class Solution {
    public static void swap(int[] arr, int i, int j) {
        int tmp = arr[i];
        arr[i] = arr[j];
        arr[j] = tmp;
    }

    private static int[] partition(int[] nums, int L, int R) {
        int pivot = nums[(int) (Math.random() * (R - L + 1)) + L];
        int less = L - 1;
        int more = R + 1;
        int index = L;
        while (index < more) {
            if (nums[index] < pivot) {
                swap(nums, ++less, index++);
            } else if (nums[index] > pivot) {
                swap(nums, index, --more);
            } else {
                index++;
            }
        }
        return new int[]{less + 1, more - 1};
    }

    private static int getKth(int[] nums, int K) {
        int L = 0, R = nums.length - 1;
        while (L < R) {
            int[] partition = partition(nums, L, R);
            if (K >= partition[0] && K <= partition[1]) {
                return nums[K];
            } else if (K < partition[0]) {
                R = partition[0] - 1;
            } else {
                L = partition[1] + 1;
            }
        }
        return nums[L];
    }

    private static int[] solve(int[] nums, int K) {
        int n = nums.length;
        int first = getKth(nums, (K - 1) / n);
        int lessNum = 0;
        int equalNum = 0;
        for (int num : nums) {
            if (num < first) {
                lessNum++;
            } else if (num == first) {
                equalNum++;
            }
        }
        int rest = K - lessNum * n;
        return new int[]{first, getKth(nums, (rest - 1) / equalNum)};
    }
}

标签：index,arr,nums,int,partition,数值
来源： https://www.cnblogs.com/tianyiya/p/15430951.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

第k小的数值对