饭后小甜点leetcode——分治算法

归并

寻找两个有序数组的中位数

public class Solution {
    public double FindMedianSortedArrays(int[] nums1, int[] nums2) {
        var m = nums1.Length;
        var n = nums2.Length;
        var half = (m + n) / 2;
        if(m==0&&n==0){
            return 0;
        }
        var num = new int[m+n];
        int i=0,j=0,k=0;
        while(k<=half){
            if(j<n && (i>=m || nums1[i]>nums2[j])){//j好好的
                num[k++] = nums2[j++];
            }
            else if(i<m && (j>=n || nums1[i]<=nums2[j])){//i好好的
                num[k++] = nums1[i++];
            }
        }
        double ans = (m+n)%2==0?(num[k-1]+num[k-2])/2.0:num[k-1];
        
        return ans;
    }
}

合并K个排序链表

public class Solution {
    public ListNode MergeKLists(ListNode[] lists) {
        return MergeKLists(lists, 0, lists.Length-1);
    }
    
    public ListNode MergeKLists(ListNode[] lists, int left, int right) {
        if(left<right){
            var mid = (left+right)/2;
            var l1 = MergeKLists(lists, left, mid);
            var l2 = MergeKLists(lists, mid+1, right);
            return MergeTwoLists(l1, l2);
        }
        else if(left==right){
            return lists[left];
        }
        else{
            return null;
        }
    }
    
    public ListNode MergeTwoLists(ListNode l1, ListNode l2) {
        if(l1==null) return l2;
        if(l2==null) return l1;
        if(l1.val<l2.val){ 
            l1.next = MergeTwoLists(l1.next, l2);
            return l1;
        }
        else{ 
            l2.next = MergeTwoLists(l2.next, l1);
            return l2;
        }
    }
}

快排

数组中的第K个最大元素

在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
说明:

你可以假设 k 总是有效的，且 1 ≤ k ≤ 数组的长度。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/kth-largest-element-in-an-array

public class Solution {
    public int FindKthLargest(int[] nums, int k) {
        var lo = 0; var hi = nums.Length-1;
        return Find(nums, k, lo, hi);
    }
    
    public int Find(int[] nums, int k, int lo, int hi){
        var pivot = Partition(nums, lo, hi);
        if(pivot+1==k){
            return nums[k-1];
        }
        else if(pivot+1<k){
            return Find(nums, k, pivot+1, hi);
        }
        else{
            return Find(nums, k, lo, pivot-1);
        }
    }
    
    public int Partition(int[] nums, int lo, int hi){
        var mid = nums[lo];
        while(lo<hi){
            while(lo<hi&&nums[hi]<mid){
                hi--;
            }
            nums[lo] = nums[hi];
            while(lo<hi&&nums[lo]>=mid){
                lo++;
            }
            nums[hi] = nums[lo];
        }
        nums[lo] = mid;
        return lo;
    }
}

最接近原点的 K 个点

我们有一个由平面上的点组成的列表 points。需要从中找出 K 个距离原点 (0, 0) 最近的点。

（这里，平面上两点之间的距离是欧几里德距离。）

你可以按任何顺序返回答案。除了点坐标的顺序之外，答案确保是唯一的。

示例 1：

输入：points = [[1,3],[-2,2]], K = 1
输出：[[-2,2]]
解释：
(1, 3) 和原点之间的距离为 sqrt(10)，
(-2, 2) 和原点之间的距离为 sqrt(8)，
由于 sqrt(8) < sqrt(10)，(-2, 2) 离原点更近。
我们只需要距离原点最近的 K = 1 个点，所以答案就是 [[-2,2]]。
示例 2：

输入：points = [[3,3],[5,-1],[-2,4]], K = 2
输出：[[3,3],[-2,4]]
（答案 [[-2,4],[3,3]] 也会被接受。）

提示：

1 <= K <= points.length <= 10000
-10000 < points[i][0] < 10000
-10000 < points[i][1] < 10000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/k-closest-points-to-origin

public class Solution {
    public int[][] KClosest(int[][] points, int K) {
        int l=0,r=points.Length-1;
        while(l<=r) {
            int mid=quickpivot(points,l,r);
            if(mid==K)
                 break;
            if(mid<K)
                l=mid+1;
            else
                r=mid-1;
        }
        int[][] res=new int[K][];
        for(int i=0;i<K;i++)
            res[i]=(points[i]);
        return res;
    }
    
    public int quickpivot(int[][] points,int l, int r)
    {
        int[] pivot=points[l];
        while(l<r){
            while(l<r && compare(points[r],pivot)>=0) r--;
            points[l]=points[r];
            while(l<r && compare(pivot,points[l])>=0) l++;
            points[r]=points[l];
        }
        points[l]=pivot;
        return l;
    }
    
    public int compare(int[] a, int[] b)
    {
        return (a[0]*a[0]+a[1]*a[1])-(b[0]*b[0] + b[1]*b[1]);
    }
}