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