1711. 大餐计数

题目来源:1711. 大餐计数

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

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

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

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

 

/**
 * @param {number[]} deliciousness
 * @return {number}
 */
var countPairs = function(deliciousness) {
  let n = deliciousness.length;
  let MOD = 1000000007;
  let maxVal = Math.max(...deliciousness);
  let sumMax = maxVal * 2;

  let res = 0;
  let map = new Map();
  for(let i = 0;i < n; i++){
    let val = deliciousness[i];
    for(let sum = 1;sum <= sumMax; sum<<=1){
      let count = map.get(sum - val) || 0;
      res = (res + count) % MOD;
    }
    map.set(val, (map.get(val)||0)+1)
  }
  return res;
};

示例 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 <= deliciousness.length <= 105
  • 0 <= deliciousness[i] <= 220
posted @ 2021-07-07 09:45  尖子  阅读(33)  评论(0编辑  收藏  举报