Submission #33304231

Source Code Expand

#

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()

Submission Info

Submission Time
Task B - Better Students Are Needed!
User kazkaz
Language Python (3.8.2)
Score 200
Code Size 941 Byte
Status AC
Exec Time 28 ms
Memory 9260 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 200 / 200
Status
AC × 3
AC × 30
Set Name Test Cases
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
Case Name Status Exec Time Memory
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