代码随想录算法训练营_day07

最新推荐文章于 2024-08-16 12:19:15 发布

kennyS_ss

最新推荐文章于 2024-08-16 12:19:15 发布

阅读量699

点赞数 15

分类专栏：代码随想录算法训练营文章标签：算法

本文链接：https://blog.csdn.net/qq_43745961/article/details/140949717

版权

代码随想录算法训练营专栏收录该内容

16 篇文章 0 订阅

订阅专栏

题目信息 454. 四数相加 II

题目链接: https://leetcode.cn/problems/4sum-ii/description/
题目描述: 给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：
0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

解题思路

本题是使用哈希法的经典题目，这道题目是四个独立的数组，只要找到A[i] + B[j] + C[k] + D[l] = 0就可以，不用考虑有重复的四个元素相加等于0的情况，所以相对于题目18. 四数之和，题目15.三数之和，还是简单了不少！

如果本题想难度升级：就是给出一个数组（而不是四个数组），在这里找出四个元素相加等于0，答案中不可以包含重复的四元组，大家可以思考一下，后续的文章我也会讲到的。

本题解题步骤：

首先定义一个unordered_map，key放a和b两数之和，value 放a和b两数之和出现的次数。
遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
定义int变量count，用来统计 a+b+c+d = 0 出现的次数。
再遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。
最后返回统计值 count 就可以了

代码实现

public boolean isAnagram(String s, String t) {  
    int[] record = new int[26];  
  
    for (int i = 0; i< s.length();i++){  
        record[s.charAt(i) - 'a']++;  
        }  
    for (int i = 0; i < t.length() ; i++){  
        record [t.charAt(i) - 'a']--;  
    }  
    for (int i = 0; i < record.length ; i++){  
        if(record[i] != 0){  
            return false;  
        }  
    }  
    return true;  
}

题目信息 383. 赎金信

题目链接: https://leetcode.cn/problems/ransom-note/
题目描述: 给你两个字符串：ransomNote 和 magazine ，判断 ransomNote 能不能由 magazine 里面的字符构成。

如果可以，返回 true ；否则返回 false 。

magazine 中的每个字符只能在 ransomNote 中使用一次。

解题思路

暴力枚举，两层for循环，不断去寻找，

代码实现

    public boolean canConstruct(String ransomNote, String magazine) {  
    List<Character> ransomNoteList = new ArrayList<>();  
    for (char c : ransomNote.toCharArray()){  
        ransomNoteList.add(c);  
    }  
        for (int i = 0; i < magazine.length() ;i++){  
            char currentChar = magazine.charAt(i);  
            for (int j = 0; j< ransomNoteList.size() ; j++){  
                if (currentChar == ransomNoteList.get(j)){  
                    ransomNoteList.remove(j);  
                    break;                }  
            }  
        }  
        return ransomNoteList.isEmpty();  
}

解题思路

因为题目说只有小写字母，那可以采用空间换取时间的哈希策略，用一个长度为26的数组来记录magazine里字母出现的次数。

然后再用ransomNote去验证这个数组是否包含了ransomNote所需要的所有字母。

依然是数组在哈希法中的应用。

一些同学可能想，用数组干啥，都用map完事了，其实在本题的情况下，使用map的空间消耗要比数组大一些的，因为map要维护红黑树或者哈希表，而且还要做哈希函数，是费时的！数据量大的话就能体现出来差别了。所以数组更加简单直接有效！

代码实现

   public boolean canConstruct(String ransomNote, String magazine) {  
        if (ransomNote.length() > magazine.length()){  
            return false;  
        }  
        int [] record = new int [26];  
        for (char c : magazine.toCharArray()){  
            record[c - 'a'] += 1;  
        }  
        for (char c : ransomNote.toCharArray()){  
            record[c - 'a'] -= 1;  
        }  
        for (int i : record){  
            if (i < 0){  
                return false;  
            }  
        }  
        return true;  
}

题目信息 15. 三数之和

题目链接: https://leetcode.cn/problems/3sum/
题目描述: 给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。

解题思路

两层for循环就可以确定 a 和b 的数值了，可以使用哈希法来确定 0-(a+b) 是否在数组里出现过，其实这个思路是正确的，但是我们有一个非常棘手的问题，就是题目中说的不可以包含重复的三元组。

把符合条件的三元组放进vector中，然后再去重，这样是非常费时的，很容易超时，也是这道题目通过率如此之低的根源所在。

去重的过程不好处理，有很多小细节，如果在面试中很难想到位。

时间复杂度可以做到O(n^2)，但还是比较费时的，因为不好做剪枝操作。**

代码实现

public List<List<Integer>> threeSum(int[] nums) {  
    List<List<Integer>> result = new ArrayList<>();  
    Arrays.sort(nums);  
  
    for (int i = 0; i< nums.length; i++){  
        if (nums[i] > 0){  
            return result;  
        }  
        if (i > 0 && nums[i] == nums[i - 1] ){  
            continue;  
        }  
        HashSet<Integer> set = new HashSet<>();  
        for (int j = i + 1;j < nums.length;j++){  
            if (j > i + 2 && nums[j] == nums[j - 1] && nums[j - 1] == nums[j - 2]){  
                continue;  
            }  
            int c = - nums[i] - nums[j];  
            if (set.contains(c)){  
                result.add(Arrays.asList(nums[i],nums[j],c));  
                set.remove(c);  
            }  
            else {  
                set.add(nums[j]);  
            }  
        }  
    }  
    return result;  
}

解题思路

其实这道题目使用哈希法并不十分合适，因为在去重的操作中有很多细节需要注意，在面试中很难直接写出没有bug的代码。

而且使用哈希法在使用两层for循环的时候，能做的剪枝操作很有限，虽然时间复杂度是O(n^2)，也是可以在leetcode上通过，但是程序的执行时间依然比较长。

接下来我来介绍另一个解法：双指针法，这道题目使用双指针法要比哈希法高效一些，那么来讲解一下具体实现的思路。

动画效果如下：

15.三数之和

拿这个nums数组来举例，首先将数组排序，然后有一层for循环，i从下标0的地方开始，同时定一个下标left 定义在i+1的位置上，定义下标right 在数组结尾的位置上。

依然还是在数组中找到 abc 使得a + b +c =0，我们这里相当于 a = nums[i]，b = nums[left]，c = nums[right]。

接下来如何移动left 和right呢，如果nums[i] + nums[left] + nums[right] > 0 就说明此时三数之和大了，因为数组是排序后了，所以right下标就应该向左移动，这样才能让三数之和小一些。

如果 nums[i] + nums[left] + nums[right] < 0 说明此时三数之和小了，left 就向右移动，才能让三数之和大一些，直到left与right相遇为止。

时间复杂度：O(n^2)。

代码实现

    public boolean isHappy(int n) {
        Set<Integer> record = new HashSet<>();
        while (n != 1 && !record.contains(n)) {
            record.add(n);
            n = getNextNumber(n);
        }
        return n == 1;
    }

    private int getNextNumber(int n) {
        int res = 0;
        while (n > 0) {
            int temp = n % 10;
            res += temp * temp;
            n = n / 10;
        }
        return res;
    }

去重逻辑的思考

#a的去重

说到去重，其实主要考虑三个数的去重。 a, b ,c, 对应的就是 nums[i]，nums[left]，nums[right]

a 如果重复了怎么办，a是nums里遍历的元素，那么应该直接跳过去。

但这里有一个问题，是判断 nums[i] 与 nums[i + 1]是否相同，还是判断 nums[i] 与 nums[i-1] 是否相同。

有同学可能想，这不都一样吗。

其实不一样！

都是和 nums[i]进行比较，是比较它的前一个，还是比较它的后一个。

如果我们的写法是这样：

if (nums[i] == nums[i + 1]) { // 去重操作
    continue;
}

1
2
3

那我们就把三元组中出现重复元素的情况直接pass掉了。例如{-1, -1 ,2} 这组数据，当遍历到第一个-1 的时候，判断下一个也是-1，那这组数据就pass了。

我们要做的是不能有重复的三元组，但三元组内的元素是可以重复的！

所以这里是有两个重复的维度。

那么应该这么写：

if (i > 0 && nums[i] == nums[i - 1]) {
    continue;
}

1
2
3

这么写就是当前使用 nums[i]，我们判断前一位是不是一样的元素，在看 {-1, -1 ,2} 这组数据，当遍历到第一个 -1 的时候，只要前一位没有-1，那么 {-1, -1 ,2} 这组数据一样可以收录到结果集里。

这是一个非常细节的思考过程。

#b与c的去重

很多同学写本题的时候，去重的逻辑多加了对right 和left 的去重：（代码中注释部分）

while (right > left) {
    if (nums[i] + nums[left] + nums[right] > 0) {
        right--;
        // 去重 right
        while (left < right && nums[right] == nums[right + 1]) right--;
    } else if (nums[i] + nums[left] + nums[right] < 0) {
        left++;
        // 去重 left
        while (left < right && nums[left] == nums[left - 1]) left++;
    } else {
    }
}

但细想一下，这种去重其实对提升程序运行效率是没有帮助的。

拿right去重为例，即使不加这个去重逻辑，依然根据 while (right > left) 和 if (nums[i] + nums[left] + nums[right] > 0) 去完成right-- 的操作。

多加了 while (left < right && nums[right] == nums[right + 1]) right--; 这一行代码，其实就是把需要执行的逻辑提前执行了，但并没有减少判断的逻辑。

最直白的思考过程，就是right还是一个数一个数的减下去的，所以在哪里减的都是一样的。

所以这种去重是可以不加的。仅仅是把去重的逻辑提前了而已。

题目信息 18. 四数之和

题目链接: https://leetcode.cn/problems/4sum/description/
题目描述: 给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

解题思路

四数之和，和15.三数之和 (opens new window)是一个思路，都是使用双指针法, 基本解法就是在15.三数之和 (opens new window)的基础上再套一层for循环。

但是有一些细节需要注意，例如：不要判断nums[k] > target 就返回了，三数之和可以通过 nums[i] > 0 就返回了，因为 0 已经是确定的数了，四数之和这道题目 target是任意值。比如：数组是[-4, -3, -2, -1]，target是-10，不能因为-4 > -10而跳过。但是我们依旧可以去做剪枝，逻辑变成nums[i] > target && (nums[i] >=0 || target >= 0)就可以了。

15.三数之和 (opens new window)的双指针解法是一层for循环num[i]为确定值，然后循环内有left和right下标作为双指针，找到nums[i] + nums[left] + nums[right] == 0。

四数之和的双指针解法是两层for循环nums[k] + nums[i]为确定值，依然是循环内有left和right下标作为双指针，找出nums[k] + nums[i] + nums[left] + nums[right] == target的情况，三数之和的时间复杂度是O(n^{2)，四数之和的时间复杂度是O(n}3) 。

那么一样的道理，五数之和、六数之和等等都采用这种解法。

对于15.三数之和 (opens new window)双指针法就是将原本暴力O(n^{3)的解法，降为O(n}2)的解法，四数之和的双指针解法就是将原本暴力O(n^{4)的解法，降为O(n}3)的解法。

之前我们讲过哈希表的经典题目：454.四数相加II (opens new window)，相对于本题简单很多，因为本题是要求在一个集合中找出四个数相加等于target，同时四元组不能重复。

而454.四数相加II (opens new window)是四个独立的数组，只要找到A[i] + B[j] + C[k] + D[l] = 0就可以，不用考虑有重复的四个元素相加等于0的情况，所以相对于本题还是简单了不少！

代码实现

public List<List<Integer>> fourSum(int[] nums, int target) {  
    List<List<Integer>> result = new ArrayList<>();  
    if (nums == null || nums.length < 4) {  
        return result;  
    }  
    Arrays.sort(nums);  
  
    for (int i = 0; i < nums.length - 3; i++) {  
        // 修改第一个限制条件  
        if (nums[i] > target && nums[i] > 0 && target > 0) {  
            break; // 使用 break 而不是 return        }  
        if (i > 0 && nums[i - 1] == nums[i]) {  
            continue;  
        }  
  
        for (int j = i + 1; j < nums.length - 2; j++) {  
            // 修改第二个限制条件  
            if (nums[i] + nums[j] > target && nums[i] + nums[j] > 0 && target > 0) {  
                break; // 使用 break 而不是 return            }  
            if (j > i + 1 && nums[j - 1] == nums[j]) {  
                continue;  
            }  
  
            int left = j + 1;  
            int right = nums.length - 1;  
  
            while (right > left) {  
                long sum = (long) nums[i] + nums[j] + nums[left] + nums[right];  
                if (sum > target) {  
                    right--;  
                } else if (sum < target) {  
                    left++;  
                } else {  
                    result.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));  
                    while (right > left && nums[right] == nums[right - 1]) right--;  
                    while (right > left && nums[left] == nums[left + 1]) left++;  
                    left++;  
                    right--;  
                }  
            }  
        }  
    }  
    return result;  
}

错误版本

public List<List<Integer>> fourSum2(int[] nums, int target) {  
    List<List<Integer>> result = new ArrayList<>();  
    Arrays.sort(nums);  
  
    for (int i = 0; i < nums.length; i++){  
        if (nums[i] > 0 && nums[i] > target){  
            return result;  
        }  
        if (i > 0 && nums[i - 1] == nums[i]){  
            continue;  
        }  
        for (int j = i + 1;j < nums.length ; j++){  
            if (nums[i] + nums[j] > 0 && nums[i] + nums[j] > target){  
                return result;  
            }  
            if (j > i + 1 && nums[j - 1] ==nums[j]){  
                continue;  
            }  
            int left = j + 1;  
            int right = nums.length - 1;  
            while (right > left){  
                long sum = (long) nums[i] + nums[j] + nums[left] + nums[right];  
                if (sum > target){  
                    right--;  
                }else if (sum < target){  
                    left++;  
                }else {  
                    result.add(Arrays.asList(nums[i],nums[j],nums[left],nums[right]));  
                    while (right > left && nums[right] == nums[right - 1]) right--;  
                    while (right > left && nums[left] == nums[left + 1]) left++;  
                    left++;  
                    right--;  
                }  
            }  
        }  
  
    }  
    return result;  
}

kennyS_ss

关注

15
点赞
踩
10

收藏

觉得还不错? 一键收藏
0
评论
代码随想录算法训练营_day07

本题是使用哈希法的经典题目，这道题目是四个独立的数组，只要找到A[i] + B[j] + C[k] + D[l] = 0就可以，不用考虑有重复的四个元素相加等于0的情况，所以相对于题目18. 四数之和，题目15.三数之和，还是简单了不少！如果本题想难度升级：就是给出一个数组（而不是四个数组），在这里找出四个元素相加等于0，答案中不可以包含重复的四元组，大家可以思考一下，后续的文章我也会讲到的。本题解题步骤：题目信息 383. 赎金信题目链接: https://leetcode.cn/problem
复制链接

扫一扫