Leetcode刷题:2021-07-07每日一题 - 大餐计数

265 阅读2分钟

这是我参与新手入门的第2篇文章

一、题目描述:

大餐 是指 恰好包含两道不同餐品 的一餐,其美味程度之和等于 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) 。

二、思路分析:

实现思路

最简单的实现方式的 暴力解法, 两重 for 循环暴力去解, 但是会超时 然后感觉可以按照 两数之和的解法去做, 使用 HashMap 记录 美味程度的出现次数 对于 map 中的每一个 key。for 循环判断 2的幂的数字,判断 差值是否出现在 map 中 如果有, 需要注意 2者是否相同 最后需要对结果进行 除以 2 因为进行了重复的统计

遇到的问题

溢出问题, 使用 long 保存结果

三、AC 代码

     public int countPairs(int[] deliciousness) {

        int mod = 1000000007;
        // 记录次数
        Map<Integer, Integer> map = new HashMap<>();

        for (int num : deliciousness) {
            int count = map.getOrDefault(num, 0);
            map.put(num, ++count);
        }

        long res = 0;
        int max = (int) Math.pow(2, 21);
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            int key = entry.getKey();
            int value = entry.getValue();

            for (int i = 1; i <= max; i <<= 1) {
                int diff = i - key;

                if (map.containsKey(diff)) {
                    int diffCount = map.get(diff);
                    if (diff == key) {
                        res += (long) value * (value - 1);
                    } else {
                        res += (long) value * diffCount;
                    }
                }
            }
        }

        res /= 2;
        return (int) (res % mod);
    }