《C语言入门100例》(第5例)给定 a，求 a 的绝对值 | 系统数学函数的应用

最新推荐文章于 2024-07-09 23:27:44 发布

leapold_Z

最新推荐文章于 2024-07-09 23:27:44 发布

阅读量225

点赞数

分类专栏： Leetcode《C语言100例》文章标签： leetcode 算法 c++

本文链接：https://blog.csdn.net/leapold_Z/article/details/124439435

版权

Leetcode《C语言100例》专栏收录该内容

16 篇文章 6 订阅

订阅专栏

【第05题】给定 a，求 a 的绝对值 | 系统数学函数的应用

文章目录

主要知识点

嗯，不要看了，确实没有，直接进入习题环节 😃

习题

1. Leetcode 2006. 差的绝对值为 K 的数对数目

题目描述

给你一个整数数组 $n u m s$ 和一个整数 $k$ ，请你返回数对 $(i, j)$ 的数目，满足 $i < j$ 且 $∣ n u m s [i] - n u m s [j] ∣ = = k$ 。
$∣ x ∣$ 的值定义为：
如果 $x > = 0$ ，那么值为 $x$ 。
如果 $x < 0$ ，那么值为 $- x$ 。
提示：
1. $1 < = n u m s . l e n g t h < = 200$
2. $1 < = n u m s [i] < = 100$
3. $1 < = k < = 99$

初见

读完题，是不是就有感觉了！没错，让我们先试一下暴力搜索：

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

编码，提交，超时…等等，过啦！！！
嗯，为什么没有超时呢？不科学啊…让我们再看一眼提示。

提示：
1. $1 < = n u m s . l e n g t h < = 200$
2. $1 < = n u m s [i] < = 100$
3. $1 < = k < = 99$

思路

没错， $89\%$ 的提交比你更快的秘密就在提示中。可以看到数组的值是有固定范围的，且只要求返回满足条件的 $(i, j)$ 的个数，那么可以用基数排序把数组中的数按顺序放入顺序数组 numMap 中，再遍历 numMap 完成搜索。

代码

int countKDifference(vector<int>& nums, int k) {
    int result(0);
    int numMap[101]{0};
    for (int i = 0; i < nums.size(); ++i) { // 将每个数放入相应位置
        numMap[nums[i]]++;
    }
    for (int i = 0; i < 101; ++i) { // 搜索满足条件的数对
        int idx = i + k;
        if (numMap[i] != 0 && idx < 101) {
            if (numMap[idx] != 0) { // 数对 (i, j) 的数目直接乘即可
                result += numMap[i] * numMap[idx];
            }
        }
        if (idx > 100) {
            break;
        }
    }
    return result;
}