結果

問題 No.1255 ハイレーツ・オブ・ボリビアン
コンテスト
ユーザー tamato
提出日時 2020-10-09 23:12:07
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 842 bytes
コンパイル時間 186 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 167,096 KB
最終ジャッジ日時 2024-07-20 14:08:12
合計ジャッジ時間 4,657 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 1 WA * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 1000000007
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    # x**k == y % mod
    # O(mod**0.5)
    def DiscreteLog(x, y, mod):
        m = int(mod ** 0.5) + 1
        dic = {1: 0}

        # baby-step
        xi = 1
        for i in range(1, m + 1):
            xi = (xi * x) % mod
            dic[xi] = i
        if y in dic:
            if dic[y] != 0:
                return dic[y]

        # giant-step
        r = pow(xi, mod - 2, mod)
        for a in range(1, m + 1):
            y = (y * r) % mod
            if y in dic:
                return a * m + dic[y]

        return -1

    for _ in range(int(input())):
        N = int(input())
        if N == 1:
            print(1)
            continue
        ans = DiscreteLog(2, 1, 2*N-1)
        print(ans)


if __name__ == '__main__':
    main()
0