题目来源于知识星球—英雄算法联盟,一月算法集训专题
前言
经过漫长的休息之后,在寒假以及新年到来之际再次刷起算法,培养自己的算法素养和算法思维,加油!今天只有一道简单题。
2006.差的绝对值为K的数对数目(简单)
1.题目描述
2.解题思路
本题有两种解法,第一种为暴力解法,即使用双重for循环进行遍历,第一层循环从下标为0处开始遍历,第二层循环
从下标为 i+1 处开始遍历,计算两个数的差的绝对值,如果满足题意,则将结果变量ans加1返回即可;
第二种解法为哈希表,首先用哈希表来记录每个元素出现的次数,在遍历哈希表,判断相隔K个间隔的元素是否存在,
两个次数相乘再相加即为结果。
3.代码演示(C++)
(1)暴力解法
class Solution
{
public:
int countKDifference(vector<int>& nums, int k)
{
int ans=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)
{
ans++;
}
}
}
return ans;
}
};
(2)哈希表
class Solution
{
public:
int countKDifference(vector<int>& nums, int k)
{
int ans=0;
vector<int>map(101);
for(int i:nums)//遍历nums数组中的每个元素
{
map[i]++;//统计每个元素出现的次数
}
for(int i=1;i<=100;i++)
{
if(i+k<=100)
{
ans+=map[i]*map[i+k];
}
}
return ans;
}
};
4.题目链接
总结
学习到了C++中基于范围的for循环的基本语法,即
vector<int>vec;
for(int i:vec)
{
cout<<i<<endl;
}
上方代码的作用为实现了对 vector 类型变量 vec 内容的打印,变量 i 遍历 vector 中的每一个元素,直到 vector 结束。