大餐计数- countPairs

题目描述

大餐 是指 恰好包含两道不同餐品 的一餐，其美味程度之和等于 2 的幂。

你可以搭配 任意 两道餐品做一顿大餐。

给你一个整数数组 deliciousness ，其中 deliciousness[i] 是第 i​​​​​​​​​​​​​​ 道餐品的美味程度，返回你可以用数组中的餐品做出的不同 大餐 的数量。结果需要对 109 + 7 取余。

注意，只要餐品下标不同，就可以认为是不同的餐品，即便它们的美味程度相同。

示例 1：

输入：deliciousness = [1,3,5,7,9]
输出：4
解释：大餐的美味程度组合为 (1,3) 、(1,7) 、(3,5) 和 (7,9) 。
它们各自的美味程度之和分别为 4 、8 、8 和 16 ，都是 2 的幂。
示例 2：

输入：deliciousness = [1,1,1,3,3,3,7]
输出：15
解释：大餐的美味程度组合为 3 种 (1,1) ，9 种 (1,3) ，和 3 种 (1,7) 。

解题思路

1.循环遍历，将味道相同，以及对应菜品数量放入map中
2.遍历map，map中味道相同的菜品数量大于1且两两组合符合题意，则计数
3.map中味道放入key中，两次循环，两两组合符合题意则计数

代码演示

class Solution {
    public int countPairs(int[] deliciousness) {
        int nums = 0;
        final int MOD = 1000000007;
        Map<Integer,Integer> numsMap = new HashMap<>();
        int n = deliciousness.length;
        for (int i = 0; i < n; i++) {
            int value = numsMap.getOrDefault(deliciousness[i],0)+1;
            numsMap.put(deliciousness[i],value);
        }
        int size = numsMap.size();
        for (int key:numsMap.keySet()
             ) {
            if (numsMap.get(key)>1)
            {
                double res = Math.log(key * 2) / Math.log(2);
                if(res==(int)res)
                {
                    nums += numsMap.get(key)*(numsMap.get(key)-1)/2;
                    nums %= MOD;
                }

            }
        }
        Set<Integer> keySet = numsMap.keySet();
        Integer[] keys = new Integer[size];
        keySet.toArray(keys);
        for (int i = 0; i < size; i++) {
            for (int j = i+1; j < size ; j++) {
                double res = Math.log(keys[i]+keys[j])/Math.log(2);
                if (res == (int)res)
                {
                    nums += numsMap.get(keys[i])*numsMap.get(keys[j]);
                    nums %= MOD;
                }

            }
        }
        return nums;
    }
}

效果

63 / 70 个通过测试用例
状态：超出时间限制
暴力解答，虽有些地方优化了，意料之中还是超时！