一、题目
给你一个整数数组 nums
。
如果一组数字 (i,j)
满足 nums[i]
== nums[j]
且 i
< j
,就可以认为这是一组 好数对 。
返回好数对的数目。
示例 1:
输入:nums = [1,2,3,1,1,3] 输出:4 解释:有 4 组好数对,分别是 (0,3), (0,4), (3,4), (2,5) ,下标从 0 开始
二、个人思路
思路:该题实际就是计算符合好数对的数量,实际不难,直接双重循环暴力解决。
code1:
class Solution(object):
def numIdenticalPairs(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
result = 0
for i in range(0,len(nums)):
for j in range(i,len(nums)):
if(nums[i]==nums[j]):
if(i<j):
result+=1;
return result
结果:结果是通过了,有基础的同学应当都会在第一个if中采用且运算符,不知道为什么我使用且运算符一直报错,明天我装一下Python环境找找原因。
三、拓展
咱们来看看,高手的理解吧。
一次遍历就可以了,边填充哈希表边算结果,不用再遍历一遍哈希表
class Solution {
public:
int numIdenticalPairs(vector<int>& nums) {
unordered_map<int, int> mp;
int res = 0;
for(int n : nums){
res += mp[n];
++mp[n];
}
return res;
}
};
四、总结
1、明天得试一试为啥且运算符出问题
2、多学一些数据结构,解题的思路会更加广阔,也有可能提升算法的性能。