java学习笔记之小练习：数组

两数之和

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那 两个 整数，并返回它们的数组下标。你可以假设每种输入只会对应一个答案，并且你不能使用两次相同的元素。你可以按任意顺序返回答案。

示例 1：
输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

示例 2：
输入：nums = [3,2,4], target = 6
输出：[1,2]

class Solution {
    public int[] twoSum(int[] nums, int target) {

        int[] pos = new int[2];

        for (int i = 0; i < nums.length-1; i++) {
            for (int j =i+1; j < nums.length; j++) {
                if(nums[i]+nums[j]==target){
                    pos[0]=i;
                    pos[1]=j;
                    break;
                }
            }
        }
        return pos;
        
    }
}

好的解法：

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] pos = new int[2];
        HashMap<Integer,Integer> hm=new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            if(hm.containsKey(target-nums[i])){
                pos[0]=i;
                pos[1]=hm.get(target-nums[i]);
                break;
            }
            hm.put(nums[i],i); 
        }
        return pos;        
    }
}

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

给定两个大小分别为 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

简单的解法，时间复杂度 O(m+n)

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int i = 0;
        int j = 0;

        List<Integer> list = new ArrayList<>();
        while (i < nums1.length && j < nums2.length) {
            if (nums1[i] <= nums2[j]) {
                list.add(nums1[i]);
                i++;
            } else {
                list.add(nums2[j]);
                j++;
            }
        }
        while (i < nums1.length) {
            list.add(nums1[i]);
            i++;
        }
        while (j < nums2.length) {
            list.add(nums2[j]);
            j++;
        }

        int len = list.size();
        double mid = len % 2 == 0 ? (list.get(len / 2) + list.get(len / 2 - 1)) / 2.0 : list.get((len - 1) / 2);
        return mid;

    }
}