LeetCode简单题之差的绝对值为 K 的数对数目
生活随笔
收集整理的這篇文章主要介紹了
LeetCode简单题之差的绝对值为 K 的数对数目
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
題目
給你一個整數(shù)數(shù)組 nums 和一個整數(shù) k ,請你返回數(shù)對 (i, j) 的數(shù)目,滿足 i < j 且 |nums[i] - nums[j]| == k 。
|x| 的值定義為:
如果 x >= 0 ,那么值為 x 。
如果 x < 0 ,那么值為 -x 。
示例 1:
輸入:nums = [1,2,2,1], k = 1
輸出:4
解釋:差的絕對值為 1 的數(shù)對為:
- [1,2,2,1]
- [1,2,2,1]
- [1,2,2,1]
- [1,2,2,1]
示例 2:
輸入:nums = [1,3], k = 3
輸出:0
解釋:沒有任何數(shù)對差的絕對值為 3 。
示例 3:
輸入:nums = [3,2,1,5,4], k = 2
輸出:3
解釋:差的絕對值為 2 的數(shù)對為: - [3,2,1,5,4]
- [3,2,1,5,4]
- [3,2,1,5,4]
提示:
1 <= nums.length <= 200
1 <= nums[i] <= 100
1 <= k <= 99
來源:力扣(LeetCode)
解題思路
??根據(jù)題目的意思,我們需要尋找兩數(shù)之差絕對值等于k的值,故兩個數(shù)是對稱對等的關(guān)系。按照示例1的做法,每次訪問當前元素并試圖查找符合條件的另一個元素時,都是往后尋找的例如第一對,1和2,找完1,當遍歷到2的時候便不再計算2和1是一對了,所有1和2,或者2和1是對稱等價的,只需要計算一對即可。那么,為了能夠加速計算,我們利用哈希表計算2和1這一對。這是一個非常巧妙地思想,在前面的兩數(shù)之和也出現(xiàn)過。
class Solution:def countKDifference(self, nums: List[int], k: int) -> int:d={}ans=0for i in nums:ans+=d.get(i+k,0)+d.get(i-k,0)d[i]=d.get(i,0)+1return ans
總結(jié)
以上是生活随笔為你收集整理的LeetCode简单题之差的绝对值为 K 的数对数目的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: LeetCode简单题之检查是否区域内所
- 下一篇: LeetCode简单题之判断矩阵经轮转后