文巾解题 1711. 大餐计数

Posted 2021-08-04 刘文巾

1 题目描述

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))