好數對的數目
- 題目及要求
- 暴力算法
- 哈希算法
- 在main內使用
題目及要求
給你一個整數數組 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 開始
示例 2:
輸入:nums = [1,1,1,1]
輸出:6
解釋:數組中的每組數字都是好數對
示例 3:
輸入:nums = [1,2,3]
輸出:0
提示:
1 <= nums.length <= 100
1 <= nums[i] <= 100
暴力算法
思路:雙循環每個數字都遍歷過去,符合要求ans自增,最后返回ans即好數對的數目
class Solution {
public:int numIdenticalPairs(vector<int>& nums) {int n=nums.size();int ans=0;for(int i=0;i<n;++i){for(int j=i+1;j<n;++j){if(nums[i]==nums[j]&&i<j)++ans;}}return ans;}
};
哈希算法
思路:將出現的次數存入哈希然后去判斷并不斷更新goodpairs的值,最后返回即可
class Solution {
public:int numIdenticalPairs(vector<int>& nums) {unordered_map<int, int> hash;int goodPairs = 0;for (int num : nums) {// 如果 countMap 中已經有了當前數字,則更新好數對的數量if (hash.find(num) != hash.end()) {goodPairs += hash[num];}// 更新當前數字的出現次數hash[num]++;}return goodPairs;}
};
在main內使用
int main() {Solution solution;vector<int> nums = {1, 2, 3, 1, 1, 2, 2};int result = solution.numIdenticalPairs(nums);cout << "好數對: " << result <<endl;return 0;
}