提出 #69276247

---

ソースコード 拡げる

import sys
from collections import defaultdict, deque

def read_ints_until_k(total_needed):
    vals = []
    while len(vals) < total_needed:
        line = sys.stdin.readline()
        if not line:
            break
        vals.extend(map(int, line.strip().split()))
    return vals[:total_needed]

def main():
    # ---- 1) N M L U ----
    header = read_ints_until_k(4)
    if len(header) < 4:
        return
    N, M, L, U = header

    # ---- 2) 前半: A 構成 ----
    def clip(x):  # Ai ∈ [1, U]
        return max(1, min(U, x))

    base = []
    base += [clip(L)] * 50
    span = U - L
    for k in range(1, 19):
        v = span // (2 ** k)   # +L しない版（あなたの仕様）
        v = clip(v)
        base += [v] * 25

    if N <= len(base):
        A = base[:N]
    else:
        A = base + [clip(L)] * (N - len(base))

    minA = min(A) if A else 1  # 許容オーバー量

    # ---- 3) A を 1 行で出力 & flush ----
    sys.stdout.write(" ".join(map(str, A)) + "\n")
    sys.stdout.flush()

    # ---- 4) B を受け取る ----
    B = read_ints_until_k(M)
    if len(B) < M:
        return

    # ---- 5) 値グループごとの割り当て ----
    X = [0] * N
    S = [0] * M  # 各山の現在合計

    value_to_indices = defaultdict(deque)
    for idx, val in enumerate(A):
        value_to_indices[val].append(idx)
    unique_values_desc = sorted(value_to_indices.keys(), reverse=True)

    for v in unique_values_desc:
        dq = value_to_indices[v]
        if not dq:
            continue
        for j in range(M):
            if not dq:
                break
            # 既に達成・超過している山には追加しない
            if S[j] >= B[j]:
                continue
            # 追加後は B[j] + minA まで許容
            upper = B[j] + minA
            remain_cap = upper - S[j]
            if remain_cap <= 0:
                continue
            # この値 v を何枚まで入れられるか（上限以内の最大枚数）
            can_take = min(len(dq), remain_cap // v)
            if can_take <= 0:
                continue
            for _ in range(can_take):
                idx = dq.popleft()
                X[idx] = j + 1
            S[j] += v * can_take
        # この値グループは一巡で終了（グループ内での再訪なし）

    # ---- 6) X を 1 行で出力 & flush ----
    sys.stdout.write(" ".join(map(str, X)) + "\n")
    sys.stdout.flush()

if __name__ == "__main__":
    main()

提出情報

ジャッジ結果