結果

問題 No.303 割れません
ユーザー rpy3cpp
提出日時 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
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other WA * 2 RE * 12
権限があれば一括ダウンロードができます

ソースコード

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