結果

問題 No.2142 Segment Zero
コンテスト
ユーザー LyricalMaestro
提出日時 2024-01-22 02:49:30
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 352 ms / 2,000 ms
コード長 1,361 bytes
コンパイル時間 269 ms
コンパイル使用メモリ 82,580 KB
実行使用メモリ 75,560 KB
最終ジャッジ日時 2024-09-28 06:19:00
合計ジャッジ時間 5,278 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

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


def main():
    N, K = map(int, input().split())

    # 三角数をベースにやる方針
    upper_n = -1
    for n in range(1, N + 1):
        ans = (n * (n + 1)) // 2
        if ans >= K:
            upper_n = n
            break

    answer1 = 0
    if upper_n < N:
        answer1 += 1
    if ans > K:
        answer1 += 1

    # 真ん中の数を切り取る方針
    answer2 = -1
    for n in reversed(range(2, N + 1)):
        base = (N * (N + 1)) // 2
        base -= ((n - 1) * n) // 2

        if K < base:
            break
        elif K == base:
            answer2 = 1
            break

        low = 1
        high = n - 1
        while high - low > 1:
            mid = (low + high) // 2
            value = (mid * (mid + 1)) // 2
            if value == K - base:
                answer2 = 1
                break
            elif value < K - base:
                low = mid
            else:
                high = mid
        value = (low * (low + 1)) // 2
        if value == K - base:
            answer2 = 1
            break
        value = (high * (high + 1)) // 2
        if value == K - base:
            answer2 = 1
            break
    
    if answer2 == -1:
        print(answer1)
    else:
        print(min(answer2, answer1))
        







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