Предложение
pkpnd находится на правильном пути, по сути, обрабатываете по одной цифре за раз, и если это 1
, подсчитайте количество вариантов, которые существуют под ним, с помощью стандартной комбинаторики.
nCr()
можно заменить на предварительное вычисление таблицы, требующее O(n^2)
хранения / времени.Может быть другое свойство, которое вы можете использовать, чтобы уменьшить количество nCr
, которое вам нужно сохранить, используя свойство абсорбции вместе со стандартной рекурсивной формулой .
Даже с тысячами битов эта таблица не должна быть слишком большой.Хранение ответа также не должно быть слишком плохим, так как 2 ^ 1000 - это ~ 300 цифр.Если бы вы имели в виду сотни тысяч , тогда это был бы другой вопрос.:)
import math
def nCr(n,r):
return math.factorial(n) // math.factorial(r) // math.factorial(n-r)
def get_index(value):
b = len(value)
k = sum(c == '1' for c in value)
count = 0
for digit in value:
b -= 1
if digit == '1':
if b >= k:
count += nCr(b, k)
k -= 1
return count
print(get_index('0011')) # 0
print(get_index('0101')) # 1
print(get_index('0110')) # 2
print(get_index('1001')) # 3
print(get_index('1010')) # 4
print(get_index('1100')) # 5
Хороший вопрос, кстати.