結果

問題 No.1918 Simple Math ?
ユーザー LyricalMaestroLyricalMaestro
提出日時 2024-09-09 01:59:29
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,522 bytes
コンパイル時間 243 ms
コンパイル使用メモリ 82,728 KB
実行使用メモリ 82,364 KB
最終ジャッジ日時 2024-09-09 01:59:38
合計ジャッジ時間 9,431 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 93 ms
82,364 KB
testcase_01 AC 57 ms
65,932 KB
testcase_02 AC 62 ms
69,752 KB
testcase_03 AC 60 ms
68,372 KB
testcase_04 AC 58 ms
65,964 KB
testcase_05 AC 57 ms
65,988 KB
testcase_06 AC 1,787 ms
75,584 KB
testcase_07 AC 1,604 ms
75,528 KB
testcase_08 AC 916 ms
68,940 KB
testcase_09 AC 372 ms
68,228 KB
testcase_10 TLE -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

## https://yukicoder.me/problems/no/1918

MOD = 10 ** 9 + 7

def solve(a, N):
    
    n1 = ((a + 1) // 2) + 1
    n2 = ((a + 9) // 6) + 1
    answer = 0
    low = 1
    for i in range(1, min(n2, N) + 1):
        x = a * i
        high = 10 ** 9
        while high - low > 1:
            mid = (high + low) // 2
            if mid ** 2 <= x:
                low = mid
            else:
                high = mid
        if high ** 2 <= x:
            v = high
        else:
            v = low
        
        answer += v
        answer %= MOD
        low = v

    for i in range(min(n2, N) + 1, min(n1, N) + 1):
        x = a * i
        while (v + 1) ** 2 <= x:
            v += 1
        answer += v
        answer %= MOD
    
    # 最上位から
    low = 0
    high = 10 ** 9
    while high - low > 1:
        mid = (high + low) // 2
        if mid ** 2 <= a * N:
            low = mid
        else:
            high = mid
    if high ** 2 <= a * N:
        v = high
    else:
        v = low

    n0 = N
    while n1 < n0:
        x = (v ** 2) // a
        if x * a < v ** 2:
            x += 1
        x = max(n1 + 1, x)

        y = (n0 - x + 1) % MOD
        answer += (y * v) % MOD
        answer %= MOD
        v -= 1
        n0 = x - 1
    return answer


def main():
    T = int(input())
    answers = []
    for _ in range(T):
        a, N = map(int, input().split())
        ans = solve(a, N)
        answers.append(ans)

    for ans in answers:
        print(ans)

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