文巾解题 1711. 大餐计数

1 题目描述

文巾解题 1711. 大餐计数

2 解题思路

先设立一个哈希表,来得到不同的美味值分别有几道菜。然后对幂值和美味值进行遍历。

import math
class Solution(object):
    def countPairs(self, deliciousness):
        """
        :type deliciousness: List[int]
        :rtype: int
        """
        deliciousness.sort()
        dit={}
        for i in deliciousness:
            if(i in dit):
                dit[i]+=1
            else:
                dit[i]=1
#字典,描述 每个不同的美味程度有几道不同的菜品

        n=int(math.log(deliciousness[-1]*2,2))
#这一组菜品中,美味程度最大的两个之和,也不会超过2的n次方

        count=0
        key=dit.keys()
        key.sort()

        for i in range(n+1):
#遍历2的幂
            for j in key:
#遍历一个菜品的美味值(为了防止重复计算,我们规定第一个菜品的美味值一定小于等于第二个菜品)
                print(j)
                tmp=2**i-j
#tmp表示另一个菜品
                if(j>tmp):
                    break 
#如果第一个菜品的美味值比第二个菜品的美味值大,那么说明这一组菜品已经在之前被遍历过了。
                elif(j==tmp and dit[j]==1):
                    break
#如果这一个菜品正好等于2的幂的一半,但是只有一个,那么是不存在一组的
                elif(j==tmp and dit[j]>1):
                    count+=(dit[j]-1)*dit[j]//2
#如果这一个菜品正好等于2的幂的一半,但是有多个,那么相当于C(n,2)组美味
                else:
                    if(tmp not in dit):
                        continue
#先看另一个菜品有没有
                    count+=dit[j]*dit[tmp]
        return(count % (10**9 + 7))

上一篇:Leetcode 1711 大餐计数


下一篇:力扣刷题:1664. 生成平衡数组的方案数