1512. 好数对的数目

Problem#

给你一个整数数组 nums。

如果一组数字 (i, j) 满足 nums[i] == nums[j] 且 i < j，就认为这是一组 好数对。

返回好数对的数目。

https://leetcode.cn/problems/number-of-good-pairs/description/

示例 1：

输入：nums = [1,2,3,1,1,3] 输出：4 解释：有 4 组好数对：(0,3), (0,4), (3,4), (2,5)

示例 2：

输入：nums = [1,1,1,1] 输出：6 解释：数组中的每组数字都是好数对

约束：

1 <= nums.length <= 100
1 <= nums[i] <= 100

思路#

双重循环统计即可。也可以先用哈希表统计每个数字出现次数，每个数字贡献 c * (c-1) / 2 对。

1
from collections import Counter
2

3

4
class Solution:
5
    def numIdenticalPairs(self, nums: list[int]) -> int:
6
        count = Counter(nums)
7
        return sum(c * (c - 1) // 2 for c in count.values())

1
import unittest
2

3

4
class TestNumIdenticalPairs(unittest.TestCase):
5
    def setUp(self):
6
        self.solution = Solution()
7

8
    def test_example1(self):
9
        self.assertEqual(self.solution.numIdenticalPairs([1, 2, 3, 1, 1, 3]), 4)
10

11
    def test_example2(self):
12
        self.assertEqual(self.solution.numIdenticalPairs([1, 1, 1, 1]), 6)
13

14
    def test_no_pairs(self):
15
        self.assertEqual(self.solution.numIdenticalPairs([1, 2, 3]), 0)
16

17
    def test_single(self):
18
        self.assertEqual(self.solution.numIdenticalPairs([1]), 0)
19

20

21
if __name__ == '__main__':
22
    unittest.main()