LeetCode - 1711. 大餐计数——哈希表、双指针

1711. 大餐计数

题目描述

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

解题思路

class Solution_1711 {
    
    
    public int countPairs(int[] deliciousness) {
    
    
        Map<Integer,Integer> map = new HashMap<>();
        long ans = 0;
        for (int j = 0;j < deliciousness.length;++j) {
    
    
            for (int i = 0; i < 22; i++) {
    
    
                int target = (int) Math.pow(2,i);
                if (target - deliciousness[j] < 0) continue;
                if (map.containsKey(target - deliciousness[j])){
    
    
                    ans += map.get(target - deliciousness[j]);
                    System.out.println(ans);
                }
            }
            map.put(deliciousness[j],map.getOrDefault(deliciousness[j],0) + 1);
            System.out.println(map.toString());
        }
        ans %= 1000000007;
        return (int)ans;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_35655602/article/details/115103974