Set remove, discard, pop problem | hackerrank

remove_discard_pop.py

n = int(input())
s = set(map(int, input().split()))
o = int(input())

for i in range(o):
    op = input().split()
    if(op[0] == "pop"):
        if len(s)>0:
            s.pop()
    elif(op[0] == "remove"):
        if int(op[1]) in s:
            s.remove(int(op[1]))
    elif(op[0] == "discard"):
        s.discard(int(op[1]))
    else:
        assert False
print(sum(s))