結果

問題 No.303 割れません
ユーザー rpy3cpprpy3cpp
提出日時 2017-10-12 23:41:09
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 1,092 bytes
コンパイル時間 172 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 107,520 KB
最終ジャッジ日時 2024-11-17 10:58:25
合計ジャッジ時間 6,261 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
51,712 KB
testcase_01 AC 40 ms
52,096 KB
testcase_02 AC 38 ms
52,096 KB
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 WA -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 WA -
権限があれば一括ダウンロードができます

ソースコード

diff #

def solve(L):
    if L == 1:
        return 1, 1
    elif L == 2:
        return 4, 1  # 3を買ってきて切る
    elif L & 1:
        return L, fib(L)
    else:
        return L, 2 * fibfib(L//2)


def fib(L):
    '''a1 = fib(2*k), a0 = fib(2*k-1)
    '''
    k = (L + 1) // 2
    a1, a0 = 0, 1
    p1, p0 = 1, 1
    while k:
        if k & 1:
            ac = a1 * p1
            a1 = ac + a1*p0 + a0*p1
            a0 = ac + a0*p0
        p2 = p1 * p1
        p1 = p2 + 2*p1*p0
        p0 = p2 + p0*p0
        k >>= 1
    if L & 1:
        return a0
    else:
        return a1


def fibfib(m):
    '''
    m = 2 * k + 1 or
    m = 2 * k
    a1 = fib(2*k), a0 = fib(2*k-1)
    '''
    k = m // 2
    a1, a0 = 0, 1
    p1, p0 = 1, 1
    while k:
        if k & 1:
            ac = a1 * p1
            a1 = ac + a1*p0 + a0*p1
            a0 = ac + a0*p0
        p2 = p1 * p1
        p1 = p2 + 2*p1*p0
        p0 = p2 + p0*p0
        k >>= 1
    if m & 1:
        return a1 * (a1 + a0)
    else:
        return a1 * a0
    
L = int(input())
cost, n_pat = solve(L)
print(cost)
print(n_pat)
0