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

  • + 0 comments
    n = int(input())
    s = set(map(int, input().split()))
    N = int(input())
    for _ in range(N):
        inp = input()
        try:
            if inp == 'pop':
                s.pop()
            else:
                command, number = inp.split()
                if command == 'remove':
                    s.remove(int(number))
                if command == 'discard':
                    s.discard(int(number))
        except KeyError:
            continue
    
    print(sum(s))