力扣每日一题20220209

脆皮的炸鸡

已于 2022-02-09 10:08:12 修改

阅读量202

点赞数 1

分类专栏：每日一题文章标签： leetcode 算法动态规划

于 2022-02-09 09:57:53 首次发布

本文链接：https://blog.csdn.net/m0_46168517/article/details/122835997

版权

每日一题专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目：

给你一个整数数组 nums 和一个整数 k ，请你返回数对 (i, j) 的数目，满足 i < j 且 |nums[i] - nums[j]| == k 。
|x| 的值定义为：
如果 x >= 0 ，那么值为 x 。
如果 x < 0 ，那么值为 -x 。
1 <= nums.length <= 200
1 <= nums[i] <= 100
1 <= k <= 99

从题目可以看到，由于本题的范围定的较小，因此只是一道简单的枚举算法就可以解决。

class Solution {
public:
    int countKDifference(vector<int>& nums, int k) {
        int i, j;
        int ans = 0;
        for (i = 0; i < nums.size(); i++)
        {
            for (j = 0; j < i; j++)
            {
                if (abs(nums[i] - nums[j]) == k)
                    ans++;
            }
        }
        return ans;
    }
};

最终运行结果：
在这里插入图片描述
当然除了O（n^2）的暴力算法，下面还有一种简单的采用哈希表的O（n）算法，极大的减少了时间复杂度。

class Solution {
public:
    int countKDifference(vector<int>& nums, int k) {
        int i, j;
        int ans = 0;
        unordered_map<int, int> cnt;
        for (i = 0; i < nums.size(); i++)
        {
            ans += (cnt.count(nums[i] + k) ? cnt[nums[i] + k] : 0);
            ans += (cnt.count(nums[i] - k) ? cnt[nums[i] - k] : 0);
            cnt[nums[i]]++;
        }
        return ans;
    }
};