leetcode 762. 二進位制表示中質數個計算置位(python)
阿新 • • 發佈:2018-11-09
給定兩個整數 L
和 R
,找到閉區間 [L, R]
範圍內,計算置位位數為質數的整數個數。
(注意,計算置位代表二進位制表示中1的個數。例如 21
的二進位制表示 10101
有 3 個計算置位。還有,1 不是質數。)
示例 1:
輸入: L = 6, R = 10 輸出: 4 解釋: 6 -> 110 (2 個計算置位,2 是質數) 7 -> 111 (3 個計算置位,3 是質數) 9 -> 1001 (2 個計算置位,2 是質數) 10-> 1010 (2 個計算置位,2 是質數)
示例 2:
輸入: L = 10, R = 15 輸出: 5 解釋: 10 -> 1010 (2 個計算置位, 2 是質數) 11 -> 1011 (3 個計算置位, 3 是質數) 12 -> 1100 (2 個計算置位, 2 是質數) 13 -> 1101 (3 個計算置位, 3 是質數) 14 -> 1110 (3 個計算置位, 3 是質數) 15 -> 1111 (4 個計算置位, 4 不是質數)
注意:
L, R
是L <= R
且在[1, 10^6]
中的整數。R - L
的最大值為 10000。
思路一:
因為L,R的數值在[1,1000000]之間,換算一下,其實二進位制位數也就是在20位以下。這樣的話可以用一個列表把1-20裡邊所有的質數表示出來。然後在依次判斷在區間的每一個數就行。
class Solution(object): def countPrimeSetBits(self, L, R): """ :type L: int :type R: int :rtype: int """ set_bits = [2,3,5,7,11,13,17,19] num = 0 for i in range(L,R+1): res = bin(i).count('1') if res in set_bits: num += 1 return num
思路二:和思路其實差不多,唯一的亮點是用0,1列表代替了質數列表。
class Solution(object):
def countPrimeSetBits(self, L, R):
"""
:type L: int
:type R: int
:rtype: int
"""
# L,R 是在 [1,10^6] 中的整數,因此置位的個數最多為 19
#建立列表 p,0-20 中質數為置 1 ,非質數位置 0
p = (0, 0, 1, 1, 0, 1, 0, 1, 0, 0, 0, 1, 0, 1, 0, 0, 0, 1, 0, 1)
re = 0
for n in range(L, R + 1):
re += p[bin(n).count('1')]
return re