結果

問題 No.2844 Birthday Party Decoration
ユーザー akasia_midoriakasia_midori
提出日時 2024-08-23 22:24:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 225 ms / 2,000 ms
コード長 1,659 bytes
コンパイル時間 159 ms
コンパイル使用メモリ 82,252 KB
実行使用メモリ 77,812 KB
最終ジャッジ日時 2024-08-23 22:24:25
合計ジャッジ時間 1,574 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
54,292 KB
testcase_01 AC 225 ms
77,756 KB
testcase_02 AC 217 ms
77,484 KB
testcase_03 AC 224 ms
77,812 KB
testcase_04 AC 210 ms
77,548 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

def oi(): return int(input())
def os(): return input().rstrip()
def mi(): return list(map(int, input().split()))

# import sys
# input = sys.stdin.readline
# import sys
# sys.setrecursionlimit(10**8)
# import pypyjit
# pypyjit.set_param('max_unroll_recursion=-1')
input_count = 0
T = oi()
def solve(N,X,C):

    # 左右で各bitが埋まる最短の移動距離を調べる

    left = [0] * N
    right = [float("inf")] * N

    dictC = {}
    maxC = 0
    for i, c in enumerate(C):
        dictC[c] = i
        maxC = max(maxC, c)
    maxC += 1
    sums = 0
    for i, b in enumerate(bin(X)[2:].rjust(maxC, "0")[::-1]):
        if i in dictC:
            if b=="0":
                # これまでの和にどれだけ足せばいい？
                left[dictC[i]] = (pow(2,i) - sums)
            else:
                #今が1bit立っているということは0で達成できる
                left[dictC[i]] = 0
        sums += pow(2, i) * int(b)
        # print(sums)

    sums = 0
    for i, b in enumerate(bin(X)[2:].rjust(maxC, "0")[::-1]):
        if i in dictC:
            if b=="0":
                if X >= pow(2,i):
                    # これまでの和からどれだけ引けばいい？
                    right[dictC[i]] = sums+1
            else:
                right[dictC[i]] = 0
        sums += pow(2, i) * int(b)

    id_left = [(l,i) for i,l in enumerate(left)]
    out = float("inf")
    nexts = 0
    for a in sorted(id_left, reverse=True):
        out = min(out, (nexts+a[0])*2)
        nexts = max(nexts, right[a[1]])
    print(out)
        # print(sums)
for _ in range(T):
    N, X = mi()
    C = mi()

    solve(N,X,C)
0