48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
from __future__ import print_function
|
||
|
|
import random
|
||
|
|
|
||
|
|
DIFFICULTY_TARGET = 120
|
||
|
|
DIFFICULTY_WINDOW = 720
|
||
|
|
DIFFICULTY_LAG = 15
|
||
|
|
DIFFICULTY_CUT = 60
|
||
|
|
|
||
|
|
def difficulty():
|
||
|
|
times = []
|
||
|
|
diffs = []
|
||
|
|
while True:
|
||
|
|
if len(times) <= 1:
|
||
|
|
diff = 1
|
||
|
|
else:
|
||
|
|
begin = max(len(times) - DIFFICULTY_WINDOW - DIFFICULTY_LAG, 0)
|
||
|
|
end = min(begin + DIFFICULTY_WINDOW, len(times))
|
||
|
|
length = end - begin
|
||
|
|
assert length >= 2
|
||
|
|
if length <= DIFFICULTY_WINDOW - 2 * DIFFICULTY_CUT:
|
||
|
|
cut_begin = 0
|
||
|
|
cut_end = length
|
||
|
|
else:
|
||
|
|
excess = length - (DIFFICULTY_WINDOW - 2 * DIFFICULTY_CUT)
|
||
|
|
cut_begin = (excess + 1) // 2
|
||
|
|
cut_end = length - excess // 2
|
||
|
|
assert cut_begin + 2 <= cut_end
|
||
|
|
wnd = times[begin:end]
|
||
|
|
wnd.sort()
|
||
|
|
dtime = wnd[cut_end - 1] - wnd[cut_begin]
|
||
|
|
dtime = max(dtime, 1)
|
||
|
|
ddiff = sum(diffs[begin + cut_begin + 1:begin + cut_end])
|
||
|
|
diff = (ddiff * DIFFICULTY_TARGET + dtime - 1) // dtime
|
||
|
|
times.append((yield diff))
|
||
|
|
diffs.append(diff)
|
||
|
|
|
||
|
|
random.seed(1)
|
||
|
|
time = 1000
|
||
|
|
gen = difficulty()
|
||
|
|
diff = next(gen)
|
||
|
|
for i in range(100000):
|
||
|
|
power = 100 if i < 10000 else 100000000 if i < 500 else 1000000000000 if i < 1000 else 1000000000000000 if i < 2000 else 10000000000000000000 if i < 4000 else 1000000000000000000000000
|
||
|
|
time += random.randint(-diff // power - 10, 3 * diff // power + 10)
|
||
|
|
print(time, diff)
|
||
|
|
diff = gen.send(time)
|