Set .discard(), .remove() & .pop()

  • + 0 comments
    n = int(input())
    s = set(map(int, input().split()))
    for _ in range(int(input())):
        cmd=input().split()
        if cmd[0]=='pop':
            s.pop()
        elif cmd[0] == 'remove':
            s.remove(int(cmd[-1]))
        elif cmd[0] == "discard":
            s.discard(int(cmd[-1]))
    print(sum(s))