【力扣时间】【1995】【简单】统计特殊四元组

最新推荐文章于 2023-04-03 19:22:15 发布

社畜菌

最新推荐文章于 2023-04-03 19:22:15 发布

阅读量325

点赞数

分类专栏：我的算法历程文章标签： leetcode 算法

本文链接：https://blog.csdn.net/qq_19917031/article/details/122220181

版权

我的算法历程专栏收录该内容

41 篇文章 0 订阅

订阅专栏

统计特殊四元组

今天的简单题花样很多，值得瞅瞅

1、看题

这儿

题目很简单，解决方案也很好想。
而且，从最朴素的到复杂的思路因有尽有。

2、审题

额……
唯一需要注意的大概只有：初始数组nums[]是无序。
而由于四元组有下标大小的限制，所以不能使用排序。

当然也无需使用排序就是了。

3、思路

好了，由于题目规定4 <= nums.length <= 50，所以最无脑最暴力的四重循环在这题是可用的。
~~不然怎么叫简单嘛~~

不过当然，就算是我也不屑于就这么使用暴力解法。
所以我选择换个法子用暴力。

由于题目规定了四个数的下标必须满足a < b < c < d，所以很明显，各个指针之间有很严格的边界限制。

再加上规定nums[a] + nums[b] + nums[c] == nums[d]，所以当d确定后，题目就变成了在nums[0~d]中选取三个数，来保证nums[a] + nums[b] + nums[c] == nums[d]。
然后进一步，当我们在nums[0~d]中确定了c后，题目就变成了在nums[0~c]中选取两个数，来保证nums[a] + nums[b] == nums[d] - nums[c]。

以此类推，思路就成形了。

4、开工

public int countQuadruplets(int[] nums) {
        int count = 0;
        for (int i = 3; i < nums.length; i++) {
            //查询剩余0~i-1范围中能否有3个数相加，结果为nums[i]
            count += find(nums, i - 1, nums[i], 2);
        }

        return count;
    }

    public int find(int[] nums, int end, int target, int deep) {
        int count = 0;
        for (int i = end; i >= deep; i--) {
            if (deep > 0) {
                //当深度不为0时，减去当前数字，并在剩余数组中继续查询
                if (nums[i] >= target) {
                    continue;
                }

                //查询子数组中，满足情况的数量
                count += find(nums, i - 1, target - nums[i], deep - 1);
            } else {
                //当深度为0时，完全匹配target
                if (nums[i] == target) {
                    count++;
                }
            }
        }

        return count;
    }