码迷,mamicode.com
首页 > 其他好文 > 详细

好数对的数目

时间:2021-01-05 10:45:49      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:tps   col   来源   key   val   解释   http   style   遍历数组   

给你一个整数数组 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
解释:数组中的每组数字都是好数对

思路

1,定义一个map,key为元素值,val为元素下标的list集合

2,遍历数组,将相同值的元素下标都放到list中

3,计算list集合的排列数组个数,是个以1为首项,公差为1的等差数列,个数为list.size - 1

 

代码

 public int numIdenticalPairs(int[] nums) {
        Map<Integer,List<Integer>> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            List<Integer> vals = map.get(nums[i]);
            if(vals == null){
                vals = new ArrayList<>();
                map.put(nums[i],vals);
            }
            vals.add(i);
        }
        int count = 0;
        for (Map.Entry<Integer, List<Integer>> entry : map.entrySet()) {
            List<Integer> list = entry.getValue();
            int size = list.size();
            count += size * (size - 1) / 2;
        }

        return count;
    }

 

 

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-good-pairs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

好数对的数目

标签:tps   col   来源   key   val   解释   http   style   遍历数组   

原文地址:https://www.cnblogs.com/dongma/p/14219375.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!