QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#670762#9225. Fibonacci Fusion27568TL 835ms20988kbPython31.3kb2024-10-24 00:47:352024-10-24 00:47:37

Judging History

你现在查看的是最新测评结果

  • [2024-10-24 00:47:37]
  • 评测
  • 测评结果:TL
  • 用时:835ms
  • 内存:20988kb
  • [2024-10-24 00:47:35]
  • 提交

answer

import sys
sys.set_int_max_str_digits(0)
import math

sq5 = math.sqrt(5)

P = 1000000000000002097
fio = [0, 1]
for i in range(50):
    fio.append((fio[-1] + fio[-2]) % P)

def mul(m1, m2):
    m3 = [
        [(m1[0][0] * m2[0][0] + m1[0][1] * m2[1][0]) % P, (m1[0][0] * m2[0][1] + m1[0][1] * m2[1][1]) % P],
        [(m1[1][0] * m2[0][0] + m1[1][1] * m2[1][0]) % P, (m1[1][0] * m2[0][1] + m1[1][1] * m2[1][1]) % P]
    ]
    return m3

def Fio(n):
    if n <= 20:
        return fio[n]
    y = [[1, 0], [0, 1]]
    x = [[0, 1], [1, 1]]
    while n > 0:
        if n % 2 == 1:
            y = mul(y, x)
        x = mul(x, x)
        n //= 2
    return y[0][0]

def get(x):
    if x <= 20:
        y = 1
        while True:
            if fio[y] >= x:
                return y
            y += 1
    return math.ceil((math.log10(sq5) + math.log10(x)) / math.log10((1 + sq5) / 2))

n = int(input())
a = [int(input()) for _ in range(n)]
a = sorted(a)

ans = 0
mp = {-1:0}

for j in range(0, n):
    i = get(a[j])
    a[j] %= P
    ans += mp.get((Fio(i) - a[j] + P) % P, 0)
    ans += mp.get((Fio(i + 1) - a[j] + P) % P, 0)
    ans += mp.get((Fio(i + 2) - a[j] + P) % P, 0)
    ans += mp.get((Fio(i + 3) - a[j] + P) % P, 0)

    if mp.get(a[j], 0) == 0:
        mp.update({a[j]:1})
    else:
        mp[a[j]] += 1


print(ans)

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 12ms
memory: 10860kb

input:

6
50
8
8
5
72
354224848179261915070

output:

4

result:

ok 1 number(s): "4"

Test #2:

score: 0
Accepted
time: 835ms
memory: 20988kb

input:

28
200878223506436882933619847964496455022155117513398820563747455993172799881403389571477889821109288771413214004090719097929400406252135763028179112130390003528046316900603668569910008417315162907579003880220844686222148696041857432602133894827753998572080650383305777912447151917272483538029469449...

output:

27

result:

ok 1 number(s): "27"

Test #3:

score: 0
Accepted
time: 343ms
memory: 13484kb

input:

5187
2640352926124261912741724778991366987330659389621881876017670644497364093930668042530271338851702874394631009332660937266680740235862107353443518003194307853104942996827176097428402408674756368623972812842571069642405111849826172879369776309763468485788964245903781380419155348915131587410703749...

output:

6073

result:

ok 1 number(s): "6073"

Test #4:

score: 0
Accepted
time: 359ms
memory: 14164kb

input:

200000
2
2
2
2
1
2
1
1
2
2
1
1
1
2
2
1
1
2
1
1
2
2
1
2
2
2
1
1
1
1
2
2
1
2
1
2
1
1
2
2
1
1
1
2
1
1
2
1
2
2
2
2
1
2
2
1
1
1
2
1
1
1
1
1
2
1
2
2
1
1
1
2
2
2
1
1
2
1
1
2
1
2
1
1
1
2
2
2
1
1
1
1
2
1
2
1
1
2
2
1
1
2
1
1
2
1
2
2
1
2
1
2
2
1
1
2
1
1
1
2
2
2
1
2
2
1
1
2
2
2
2
1
2
1
1
2
1
2
2
1
1
1
1
2
2
2
2...

output:

15003749259

result:

ok 1 number(s): "15003749259"

Test #5:

score: -100
Time Limit Exceeded

input:

200000
944176313232170622314
2590599414036674999101
753315073608896000424
9299685298577430049245
9361800333778142620806
8988699166328904060999
9606920674025578304023
4203331868598952026136
5183047027116137697788
3968714342776915029801
8130984095583566992354
3206443643596048048798
6248561214283254355...

output:


result: