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

  • + 0 comments

    This works in Python3

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

    for _ in range(N): command = input().split()

    if command[0] == 'pop':
        s.pop()
    elif command[0] == 'discard':
        s.discard(int(command[1]))
    else:
        s.remove(int(command[1]))
    

    print(sum(s))