Set .add() HackerRank Solution in Python

Problem

If we want to add a single element to an existing set, we can use the .add() operation.

It adds the element to the set and returns 'None'.

Example

>>> s = set('HackerRank')
>>> s.add('H')
>>> print s
set(['a', 'c', 'e', 'H', 'k', 'n', 'r', 'R'])
>>> print s.add('HackerRank')
None
>>> print s
set(['a', 'c', 'e', 'HackerRank', 'H', 'k', 'n', 'r', 'R'])

Task

Apply your knowledge of the .add() operation to help your friend Rupal.

Rupal has a huge collection of country stamps. She decided to count the total number of distinct country stamps in her collection. She asked for your help. You pick the stamps one by one from a stack of N country stamps.

Find the total number of distinct country stamps.

Input Format

The first line contains an integer N, the total number of country stamps.

The next N lines contain the name of the country where the stamp is from.

Output Format

Output the total number of distinct country stamps on a single line.

Sample Input

7

UK

China

USA

France

New Zealand

UK

France 

Sample Output

5

Explanation

UK and France repeat twice. Hence, the total number of distinct country stamps is  5(five).

Program

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

# Set .add() HackerRank Solution in Python

# take a input of testcases
N = int(input())

# initialize the country set
country = set()

for i in range(N):
    country.add(input())

    # print the length of country set
print(len(country))