You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
1.6 KiB
Python
80 lines
1.6 KiB
Python
3 years ago
|
import matplotlib.pyplot as plt
|
||
|
|
|
||
|
|
INPUT = 33100000
|
||
|
|
|
||
|
|
|
||
|
|
def get_divisors(num):
|
||
|
|
divisors = set()
|
||
|
|
divisors.add(1)
|
||
|
|
divisors.add(num)
|
||
|
|
|
||
|
|
upper_limit = num
|
||
|
|
lower_limit = 2
|
||
|
|
|
||
|
|
while lower_limit < upper_limit:
|
||
|
|
quotient, rest = divmod(num, lower_limit)
|
||
|
|
if rest == 0:
|
||
|
|
divisors.add(lower_limit)
|
||
|
|
divisors.add(quotient)
|
||
|
|
lower_limit += 1
|
||
|
|
|
||
|
|
return divisors
|
||
|
|
|
||
|
|
|
||
|
|
def present_count(house_number):
|
||
|
|
divisors = get_divisors(house_number)
|
||
|
|
presents = 0
|
||
|
|
for divisor in divisors:
|
||
|
|
presents += divisor * 10
|
||
|
|
return presents
|
||
|
|
|
||
|
|
|
||
|
|
X = []
|
||
|
|
Y = []
|
||
|
|
|
||
|
|
for x in range(1, 100):
|
||
|
|
if x % 100 == 0:
|
||
|
|
print(x)
|
||
|
|
y = present_count(x)
|
||
|
|
plt.plot(x, y, "bo")
|
||
|
|
plt.annotate(f"{x}", (x, y), xytext=(2, 0), textcoords="offset points")
|
||
|
|
|
||
|
|
plt.show()
|
||
|
|
|
||
|
|
print("Estimation...")
|
||
|
|
prev_house_number = None
|
||
|
|
house_number = 1
|
||
|
|
presents = 10
|
||
|
|
while True:
|
||
|
|
prev_house_number = house_number
|
||
|
|
house_number = house_number * 2
|
||
|
|
presents += house_number * 10
|
||
|
|
print(house_number, presents)
|
||
|
|
|
||
|
|
if presents > INPUT:
|
||
|
|
break
|
||
|
|
|
||
|
|
lower_bound = prev_house_number
|
||
|
|
upper_bound = house_number
|
||
|
|
|
||
|
|
print("Estimation lower bound:", lower_bound)
|
||
|
|
print("Estimation upper bound:", upper_bound)
|
||
|
|
print()
|
||
|
|
|
||
|
|
current = int(lower_bound + (upper_bound - lower_bound) / 2)
|
||
|
|
|
||
|
|
loop_count = 0
|
||
|
|
while True:
|
||
|
|
presents = present_count(current)
|
||
|
|
print("loop", current, lower_bound, upper_bound, presents)
|
||
|
|
if presents < INPUT:
|
||
|
|
lower_bound = current
|
||
|
|
else:
|
||
|
|
upper_bound = current
|
||
|
|
|
||
|
|
current = int(lower_bound + (upper_bound - lower_bound) / 2)
|
||
|
|
|
||
|
|
print("finished")
|
||
|
|
# 2097152 too high
|
||
|
|
# 1540098 too high
|