[Hackerrank] Climbing the Leaderboard.py

#!/bin/python3

import math
import os
import random
import re
import sys

# Complete the climbingLeaderboard function below.
def climbingLeaderboard(scores, alice):
    queue = sorted(set(scores), reverse = True)
    idx = len(queue) - 1
    answer = []
    
    for alice_score in alice:
        while queue[idx] <= alice_score and idx >= 0:
            idx -= 1
        if idx < 0 :
            answer.append(1)
            continue
        # print(queue , queue[idx], idx, alice_score)
        answer.append(idx + 2)
        
    return answer
    


if __name__ == '__main__':
    fptr = open(os.environ['OUTPUT_PATH'], 'w')

    scores_count = int(input())

    scores = list(map(int, input().rstrip().split()))

    alice_count = int(input())

    alice = list(map(int, input().rstrip().split()))

    result = climbingLeaderboard(scores, alice)

    fptr.write('\n'.join(map(str, result)))
    fptr.write('\n')

    fptr.close()