Hyperspace Jump (easy) (Codeforces 958D1)
Tags: Math, expression parsing

The Rebel fleet is on the run. It consists of m ships currently gathered around a single planet. Just a few seconds ago, the vastly more powerful Empire fleet has appeared in the same solar system, and the Rebels will need to escape into hyperspace. In order to spread the fleet, the captain of each ship has independently come up with the coordinate to which that ship will jump. In the obsolete navigation system used by the Rebels, this coordinate is given as the value of an arithmetic expression of the form .To plan the future of the resistance movement, Princess Heidi needs to know, for each ship, how many ships are going to end up at the same coordinate after the jump. You are her only hope!

Input: The first line of the input contains a single integer m (1 ≤ m ≤ 200 000) – the number of ships. The next m lines describe one jump coordinate each, given as an arithmetic expression. An expression has the form (a+b)/c. Namely, it consists of: an opening parenthesis (, a positive integer a of up to two decimal digits, a plus sign +, a positive integer b of up to two decimal digits, a closing parenthesis ), a slash /, and a positive integer c of up to two decimal digits.

Output: Print a single line consisting of m space-separated integers. The i-th integer should be equal to the number of ships whose coordinate is equal to that of the i-th ship (including the i-th ship itself).

Note: In the sample testcase, the second and the third ship will both end up at the coordinate 3.Note that this problem has only two versions – easy and hard.

Sample Inputs:
4
(99+98)/97
(26+4)/10
(12+33)/15
(5+1)/7

Sample Outputs:
1 2 2 1

Solution (Python):
def main():
    return nShipsWithSameCoordinate(int(input()), {})

def nShipsWithSameCoordinate(nShips, nShipsAt):
    ships_Coordinates = list(range(nShips))
    for i in ships_Coordinates:
        (n1, remaining) = input()[1:].split('+')
        (n2, n3) = remaining.split(')/')
        coordinate = (int(n1) + int(n2)) / int(n3)
        if coordinate not in nShipsAt:
            nShipsAt[coordinate] = 1
        else:
            nShipsAt[coordinate] += 1
        ships_Coordinates[i] = coordinate
    return ' '.join(map(str, (nShipsAt[c] for c in ships_Coordinates)))
if __name__ == '__main__':
    print(main())

Solution (Python):
n = int(input())
dic = {}
for i in range(n):
    numstrings = []
    stg = input().split('(')
    stg = stg[1].split('+')
    numstrings.append(stg[0])
    stg = stg[1].split(')')
    numstrings.append(stg[0])
    stg = stg[1].split('/')
    numstrings.append(stg[1])
    numstrings = [float(x) for x in numstrings]
    x = (numstrings[0] + numstrings[1]) / numstrings[2]
    if x in dic:
        dic[x].append(i)
    else:
        dic[x] = [i]
ans = [0 for i in range(n)]
for x in dic:
    w = str(len(dic[x]))
    for j in dic[x]:
        ans[j] = w
print(' '.join(ans))

Solution (Python):
n = int(input())
a = [0] * n
for i in range(n):
    s = input()
    (x, y) = s.split('+')
    x = float(x[1:])
    (u, v) = y.split('/')
    v = float(v)
    u = float(u[:-1])
    a[i] = (x + u) / v
k = {}
for i in range(n):
    if a[i] not in k:
        k[a[i]] = 0
    k[a[i]] += 1
o = [0] * n
for i in range(n):
    o[i] = str(k[a[i]])
print(' '.join(o))