Submission #33305130

---

Source Code Expand

from operator import itemgetter as ig

def main(N, X, Y, Z, A, B):
    ans = []
    
    *data, = zip(range(1, N+1), A, B)
    
    # 数学
    data.sort(key=ig(1), reverse=True)
    n, *_ = zip(*data)
    ans.extend(n[:X])
    
    data = data[X:]
    
    # 英語
    data.sort(key=ig(0))
    data.sort(key=ig(2), reverse=True)
    n, *_ = zip(*data)
    ans.extend(n[:Y])
    
    data = data[Y:]
    
    # 合計
    data = [(n, a+b) for n, a, b in data]
    data.sort(key=ig(0))
    data.sort(key=ig(1), reverse=True)
    n, *_ = zip(*data)
    ans.extend(n[:Z])
    

    return "\n".join(map(str, sorted(ans)))

N, X, Y, Z = map(int, input().split())

*A, = map(int, input().split())
*B, = map(int, input().split())

print(main(N, X, Y, Z, A, B))

Submission Info

Judge Result