代码随想录day6 Java版

刚面试完痛定思痛，开始肝哈希表第二天

1. 两数之和

梦回大一，想起自己算法从入门到放弃的拦路虎2333333

首先正常暴力需要两层循环，可以通过边遍历边查看历史记录来实现一层循环，恰好哈希表可以完美解决，因为是两个数，遍历过a时，存入target-a，再用于匹配后面的数

由于本题要返回对应的数组下标，因此使用map存储

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

454.四数相加II

上面一个题的变种，四个数相加通过分治思想转化为两个两数之和再相加

感觉这题新手可能以为n平方时间复杂度肯定过不了，但题目有提示，所以直接大胆冲

• 1 <= n <= 200

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        Map map = new HashMap<>();
        int count = 0;
        for (int i = 0; i < nums1.length; i++){
            for (int j = 0; j < nums2.length; j++) {
                if (map.containsKey(nums1[i]+nums2[j])) {
                    map.put(nums1[i]+nums2[j],map.get(nums1[i]+nums2[j])+1);
                }
                else {
                    map.put(nums1[i]+nums2[j],1);
                }
            }
        }

        for (int i = 0; i < nums3.length; i++){
            for (int j = 0; j < nums4.length; j++) {
                if (map.containsKey(-nums3[i]-nums4[j])) {
                    count += map.get(-nums3[i]-nums4[j]);
                }
            }
        }
        return count;
    }
}

看题解用的HashMap的方法，可以直接省去判断的过程，代码简化很多

map.getOrDefault

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int res = 0;
        Map map = new HashMap();
        //统计两个数组中的元素之和，同时统计出现的次数，放入map
        for (int i : nums1) {
            for (int j : nums2) {
                int sum = i + j;
                map.put(sum, map.getOrDefault(sum, 0) + 1);
            }
        }
        //统计剩余的两个元素的和，在map中找是否存在相加为0的情况，同时记录次数
        for (int i : nums3) {
            for (int j : nums4) {
                res += map.getOrDefault(0 - i - j, 0);
            }
        }
        return res;
    }
}

383. 赎金信

跟第一天的很像，也可以用数组来实现哈希表，这里练习下刚学的getOrDefault

class Solution {
    public boolean canConstruct(String ransomNote, String magazine) {
        Map map = new HashMap<>();
        for (char c : magazine.toCharArray()) {
            map.put(c,map.getOrDefault(c,0)+1);
        }
        for (char c : ransomNote.toCharArray()) {
            map.put(c,map.getOrDefault(c,0)-1);
        }
        for (int i : map.values()) {
            if (i < 0) return false;
        }
        return true;
    }
}