文章目录
- 1. 题目
- 2. 解题
1. 题目
大餐 是指 恰好包含两道不同餐品 的一餐,其美味程度之和等于 2 的幂。
你可以搭配 任意 两道餐品做一顿大餐。
给你一个整数数组 deliciousness ,其中 deliciousness[i] 是第 i 道餐品的美味程度,返回你可以用数组中的餐品做出的不同 大餐 的数量。
结果需要对 10^9 + 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 <= deliciousness.length <= 10^5
0 <= deliciousness[i] <= 2^20
https://leetcode-cn.com/problems/count-good-meals/
2. 解题
class Solution {
public:int countPairs(vector<int>& deliciousness) {map<int,long long> m;for(auto d : deliciousness)m[d]++;//计数long long ans = 0, mod = 1e9+7;for(int i = 0; i <= 22; i++){ // 枚举 2的幂int target = 1 << i;for(auto it = m.begin(); it != m.end() && it->first <= target/2; ++it){int d1 = target-it->first;//另一份餐的美味程度if(m.find(d1) == m.end())//不存在continue;if(it->first != d1)//不相等ans = (ans+it->second*m[d1])%mod;else//相等 Cn2ans = (ans+m[d1]*(m[d1]-1)/2)%mod;}}return ans;}
};
684 ms 58 MB C++
我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!