提出 #33304231

ソースコード 拡げる

#

def main():
    N, X, Y, Z = map(int, input().split())
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))
    J = [[] for i in range(N)]
    G = [0 for i in range(N)]
    for i in range(N):
        J[i] = [i, A[i], B[i], A[i]+B[i]]
    J = sorted(J, key=lambda x:-x[1])
    #print(J)
    for x in range(X):
        G[J[x][0]] = 1
    #print(G)
    J.sort()
    J = sorted(J, key=lambda x:-x[2])
    #print(J)
    cnt = 0
    i = 0
    while cnt < Y:
        if G[J[i][0]] ==0:
            G[J[i][0]] =1
            cnt+=1
        i+=1
    #print(G)
    J.sort()
    J = sorted(J, key=lambda x:-x[3])
    #print(J)
    cnt = 0
    i = 0
    while cnt < Z:
        if G[J[i][0]] ==0:
            G[J[i][0]] =1
            cnt+=1
        i+=1
    #print(G)
    for i in range(N):
        if G[i] ==1:
            print(i+1)
    


if __name__=='__main__':
    main()

提出情報

提出日時
問題 B - Better Students Are Needed!
ユーザ kazkaz
言語 Python (3.8.2)
得点 200
コード長 941 Byte
結果 AC
実行時間 28 ms
メモリ 9260 KiB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 200 / 200
結果
AC × 3
AC × 30
セット名 テストケース
Sample sample_01.txt, sample_02.txt, sample_03.txt
All sample_01.txt, sample_02.txt, sample_03.txt, test_01.txt, test_02.txt, test_03.txt, test_04.txt, test_05.txt, test_06.txt, test_07.txt, test_08.txt, test_09.txt, test_10.txt, test_11.txt, test_12.txt, test_13.txt, test_14.txt, test_15.txt, test_16.txt, test_17.txt, test_18.txt, test_19.txt, test_20.txt, test_21.txt, test_22.txt, test_23.txt, test_24.txt, test_25.txt, test_26.txt, test_27.txt
ケース名 結果 実行時間 メモリ
sample_01.txt AC 19 ms 8928 KiB
sample_02.txt AC 21 ms 9156 KiB
sample_03.txt AC 26 ms 9056 KiB
test_01.txt AC 23 ms 8960 KiB
test_02.txt AC 21 ms 9180 KiB
test_03.txt AC 26 ms 9100 KiB
test_04.txt AC 22 ms 8952 KiB
test_05.txt AC 21 ms 9156 KiB
test_06.txt AC 21 ms 9012 KiB
test_07.txt AC 19 ms 8964 KiB
test_08.txt AC 22 ms 9260 KiB
test_09.txt AC 27 ms 9072 KiB
test_10.txt AC 20 ms 8936 KiB
test_11.txt AC 21 ms 9252 KiB
test_12.txt AC 21 ms 9008 KiB
test_13.txt AC 25 ms 9004 KiB
test_14.txt AC 23 ms 8996 KiB
test_15.txt AC 23 ms 9072 KiB
test_16.txt AC 22 ms 8924 KiB
test_17.txt AC 22 ms 9064 KiB
test_18.txt AC 21 ms 9108 KiB
test_19.txt AC 19 ms 8936 KiB
test_20.txt AC 25 ms 9040 KiB
test_21.txt AC 20 ms 9068 KiB
test_22.txt AC 28 ms 9008 KiB
test_23.txt AC 22 ms 9068 KiB
test_24.txt AC 19 ms 9048 KiB
test_25.txt AC 20 ms 9060 KiB
test_26.txt AC 24 ms 9104 KiB
test_27.txt AC 22 ms 9148 KiB