leetcode每日一题—1711.大餐计数

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

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

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

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

解答：

class Solution:
    def countPairs(self, deliciousness: List[int]) -> int:
        MAX = 10**9 + 7
        c = collections.Counter(deliciousness)

        m = 2*max(deliciousness)
        power = set()
        for i in range(m+2):
            if 2**i > m:
                break
            else:
                power.add(2**i)
              
        d = set(c.keys()) # 存放几种菜品的美味度
        res = 0
        seen = set()
        for i in d:
            for j in power:
                #(1,7)等价于(7,1)。若j-i在seen中，则（i,j-i)已被统计过
                if j-i in seen:
                    continue
                if j-i in d:
                    if j-i == i:
                        res = (res + (c[i]*(c[i]-1))//2) % MAX 
                    else:
                        res = (res + c[i]*c[j-i]) % MAX
            seen.add(i)
                    
        return res