結果

問題 No.195 フィボナッチ数列の理解(2)
ユーザー maspy
提出日時 2020-03-25 23:12:13
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 41 ms / 5,000 ms
コード長 1,294 bytes
コンパイル時間 275 ms
コンパイル使用メモリ 12,160 KB
実行使用メモリ 10,496 KB
最終ジャッジ日時 2025-01-02 00:17:28
合計ジャッジ時間 2,213 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

#!/usr/bin/ python3.8
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
import itertools

X, Y, Z = map(int, read().split())

F1 = [1, 0]
F2 = [0, 1]

for n in range(100):
    F1.append(F1[-1] + F1[-2])
    F2.append(F2[-1] + F2[-2])


def solve(X, Y, Z):
    if X == Y == Z:
        if X == 1:
            yield 1, 1
            return
        for f1, f2 in zip(F1[1:], F2[1:]):
            if f1 >= X:
                break
            if (X - f1) % f2 == 0:
                yield 1, (X - f1) // f2
        return
    X, Y, Z = sorted([X, Y, Z])
    for i, k in itertools.product(range(100), range(100)):
        if i == k:
            continue
        num = F2[k] * X - F2[i] * Z
        den = F2[k] * F1[i] - F2[i] * F1[k]
        if num % den != 0:
            continue
        A = num // den
        num = X + Z - (F1[i] + F1[k]) * A
        den = F2[i] + F2[k]
        if num % den != 0:
            continue
        B = num // den
        if A <= 0 or B <= 0:
            continue
        seq = [A, B]
        for n in range(100):
            seq.append(seq[-1] + seq[-2])
        if Y in seq:
            yield A, B


answer = min(solve(X, Y, Z), default=None)
if answer == None:
    print(-1)
else:
    print(*answer)
0