HackerRank The captain’s Room solution in python

In this The Captain’s Room problem there is a Mr. Anant Asankhya is the manager at the INFINITE hotel. The hotel has an infinite amount of rooms. One fine day, a finite number of tourists come to stay at the hotel. The tourists consist of A Captain and An unknown group of families consisting of K members per group where K ≠ 1.

The Captain was given a separate room, and the rest were given one room per group. Mr. Anant has an unordered list of randomly arranged room entries. The list consists of the room numbers for all of the tourists. The room numbers will appear K times per group except for the Captain’s room

Mr. Anant needs you to help him find the Captain’s room number. The total number of tourists or the total number of groups of families is not known to you. You only know the value of K and the room number list.

Problem solution in Python 2 programming.

n = int(raw_input());
dic = {} 
for x in raw_input().split():
  dic[x] = dic.get(x,0) + 1
for key in dic:
  if dic[key] != n: 
    print key
    break

Problem solution in Python 3 programming.

# Enter your code here. Read input from STDIN. Print output to STDOUT
k,arr = int(input()),list(map(int, input().split()))

myset = set(arr)

print(((sum(myset)*k)-(sum(arr)))//(k-1))

Problem solution in pypy programming.

# Enter your code here. Read input from STDIN. Print output to STDOUT
k,arr = input(), list(map(int, raw_input().split()))
s = set(arr)
print ((sum(s)*k)-(sum(arr)))//(k-1)

Problem solution in pypy3 programming.

counts = dict()
n = input()
number = input().split()
for i in number:
    if i not in counts:
        counts[i] = 1
    else:
        counts[i] += 1
for i in counts:
    if counts[i] == 1:
        print (i)