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

  • + 0 comments

    For Python3 Platform

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

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

    if(cmd[0] == "pop"):
        try:
            s.pop()
        except KeyError:
            pass
    elif(cmd[0] == "remove"):
        try:
            s.remove(int(cmd[1]))
        except KeyError:
            pass
    elif(cmd[0] == "discard"):
        s.discard(int(cmd[1]))
    

    print(sum(s))