本文主要是介绍【Leetcode 1995 】 统计特殊四元组 —— 最全的注解,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
给你一个 下标从 0 开始 的整数数组 nums
,返回满足下述条件的 不同 四元组 (a, b, c, d)
的 数目 :
nums[a] + nums[b] + nums[c] == nums[d]
,且a < b < c < d
示例 1:
输入:nums = [1,2,3,6] 输出:1 解释:满足要求的唯一一个四元组是 (0, 1, 2, 3) 因为 1 + 2 + 3 == 6 。
示例 2:
输入:nums = [3,3,6,4,5] 输出:0 解释:[3,3,6,4,5] 中不存在满足要求的四元组。
示例 3:
输入:nums = [1,1,1,3,5] 输出:4 解释:满足要求的 4 个四元组如下: - (0, 1, 2, 3): 1 + 1 + 1 == 3 - (0, 1, 3, 4): 1 + 1 + 3 == 5 - (0, 2, 3, 4): 1 + 1 + 3 == 5 - (1, 2, 3, 4): 1 + 1 + 3 == 5
最全的注解
function countQuadruplets(nums: number[]): number {const cnt = new Map<number, number>();let ans = 0;const len = nums.length;// a,b,c,d//由于 nums[a] + nums[b] + nums[c] = nums[d]//所以 nums[a] + nums[b] = nums[d] - nums[c]for (let b = len - 3; b >= 1; b--) {for (let a = b - 1; a >= 0; a--) {const sum = nums[b] + nums[a];cnt.set(sum, (cnt.get(sum) || 0) + 1);}// 找出 d,其索引最低要比b大2,因为 b < c < dfor (let d = b + 2, c = b + 1; d < len; d++) {// b + 1 代替为c// 之所以 c 被固定为 b+1,而不是和d一样向后依次枚举//是因为 随着b从后向前枚举时,c同时也从后向前枚举了,//而在每次b移动时,d都要把后面的枚举一遍,相当于也和不同的c进行了匹配const diff = nums[d] - nums[c];if (cnt.has(diff)) {ans += cnt.get(diff)!;}}}return ans;
}
这篇关于【Leetcode 1995 】 统计特殊四元组 —— 最全的注解的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!