파이썬 참고용 코드

from queue import PriorityQueue
import random
 
class Human:
    age = 0
    weight = 0
    height = 0
 
    def __init__(self, age = 20, weight= 70, height=180):
        self.age = age
        self.weight = weight
        self.height = height
 
    def __lt__(self, other):
        if self.age != other.age:
            return self.age < other.age
        elif self.weight != other.weight :
            return self.weight < other.weight
        else :
            return self.height < other.height
 
    def __print__(self):
        print("age : " + str(self.age) + ", weight : " + str(self.weight) + ", height : " + str(self.height))
        # print("age : " + str(self.age))
 
# 우선순위 큐
pq = PriorityQueue()
list = []
 
for i in range(0,20):
    pq.put(Human(10 + random.randrange(0,20), 60 + random.randrange(0,10) , 160 + random.randrange(0,20)))
    list.append(Human(10 + random.randrange(0,20), 60 + random.randrange(0,10) , 160 + random.randrange(0,20)));
 
while not pq.empty():
    pq.get().__print__()
 
list.sort()
print(list)
 
# 스택
stack = []
for i in range(0, 10):
    stack.append(i)
 
while len(stack) != 0:
    print(stack.pop())
 
 
 

 

 

 
 
 
 
def solution(table, languages, preference):
    ans = ""
    dictionary = []
    n = len(languages)
 
    for i in range(0, 5):
        dictionary.append({})
        for j in range(0, 5):
            #dictionary[i].append({table[i][j] : n - j})
            dictionary[i][table[i][j]] = 5 - j
 
    high_languages = []
    high_score = -1
    for i in range(0, 5):
        sums = 0
        for j in range(0, 3):
            sums = sums + (preference[j] * dictionary[i][languages[j]])
        if sums > high_score:
            high_score = sums
            high_languages.clear()
            high_languages.append(languages[j])
        elif sums == high_score:
            high_languages.append(languages[j])
 
    high_languages.sort()
    ans = high_languages[0]
    return ans
 
 
def Input(n):
    table = []
 
    for i in range(0,5):
        table.append(input().split(" "))
    languages = (input().split(" "))
    preference = (input().split(" "))
 
    return table, languages, preference
 
n = input()
 
table, languages, preference = Input(n)
 
ret = solution(table, languages, preference)
 
 
 
"""
3
JAVA JAVASCRIPT SQL PYTHON C#
JAVASCRIPT JAVA PYTHON SQL C++
C C++ PYTHON JAVA JAVASCRIPT
JAVA JAVASCRIPT PYTHON KOTLIN PHP
C++ C# JAVASCRIPT C JAVA
PYTHON C++ SQL
7 5 5

"""

 

참고삼아~