寻找两个正序数组的中位数
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n)) 。
示例 1:
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
示例 2:
输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
示例 3:
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000
示例 4:
输入:nums1 = [], nums2 = [1]
输出:1.00000
示例 5:
输入:nums1 = [2], nums2 = []
输出:2.00000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int n1=nums1.length;
int n2=nums2.length;
int []nums;
nums=new int[n1+n2];
if(n1==0){
if(n2%2==0){
return (nums2[n2/2-1]+nums2[n2/2])/2.0;
}else{
return nums2[n2/2];
}
}
if(n2==0){
if(n1%2==0){
return (nums1[n1/2-1]+nums1[n1/2])/2.0;
}else{
return nums1[n1/2];
}
}
int count=0;
int i=0,j=0;
while(count!=(n1+n2)){
if(nums1[i]<nums2[j]){
nums[count++]=nums1[i++];
}else{
nums[count++]=nums2[j++];
}
if(i==n1){
while(j!=n2){
nums[count++]=nums2[j++];
}
break;
}
if(j==n2){
while(i!=n1){
nums[count++]=nums1[i++];
}
break;
}
}
//此处用(n1+n2)和count都一样
if((n1+n2)%2==0){
return((nums[count/2-1]+nums[count/2])/2.0);
}else{
return nums[(count/2)];
}
}
}
举例:
[1,3,5]
[2,4]
上面为什么用(n1+n2)和count都一样?
//不要觉着count在nums[count++]数组里,所以实际的数组长度应该是count+1,这种想法是错误的,因为在nums[count++]后,最终可能是nums[4],而count已经等于5了。
今天在做查找中位数题目中的
给两个正序数组排序给我的思考:
1:在给数组分配空间后(Java),会自动给数组赋初值,如int类型初值为0
2:count++是先使用,后++;
++count是先++后使用
但不论先加还是后加最终的count值相等。
注意⚠️:若用nums[++count]也可以,最终的count值依然是5,但是nums数组要注意,此时数组长度增加了1,即增加了nums[5], 而最初的nums[0]=0是分配空间后,自动赋值的。
此时nums[++count]:nums[0]=0,nums[1]=1,nums[2]=2,nums[3]=3,nums[4]=4,nums[5]=5;
数组长度为6,最终count=5
原来(nums[count++]):
nums[0]=1,nums[1]=2,nums[2]=3,nums[3]=4,nums[4]=5
数组长度为5,最终count=5
合并两个正序数组并排序:
package November;
import java.util.Scanner;
public class aa {
public static void fundMedian(int[] nums1,int[] nums2) {
int n1=nums1.length;
int n2=nums2.length;
int []nums;
nums=new int[n1+n2];
int count=0;
int i=0,j=0;
while(count!=(n1+n2)){
if(i==n1){
while(j!=n2){
nums[count++]=nums2[j++];
}
break;
}
if(j==n2){
while(i!=n1){
nums[count++]=nums1[i++];
}
break;
}
if(nums1[i]<nums2[j]){
nums[count++]=nums1[i++];
}else{
nums[count++]=nums2[j++];
}
}
// System.out.println(count);
// for(int p=0;p<=(count);p++) {
// System.out.println(nums[p]);
// }
}
public static void main(String[] args) {
int []a=new int[3];
int []b=new int[2];
Scanner inScanner=new Scanner(System.in);
for(int i=0;i<3;i++) {
a[i]=inScanner.nextInt();
}
for(int j=0;j<2;j++) {
b[j]=inScanner.nextInt();
}
aa.fundMedian(a, b);
}
}