글 읽기 - 람다식을 이용해서 풀어보려고 했습니다만.. 시간초과가 뜨네요

sonjungwoo9 1년 전 0

아래 방법으로는 풀지 못하는 걸까요..!

import sys
from collections import deque
input = sys.stdin.readline

hills = int(input())
elevation = []
for i in range (hills):
    elevation.append([int(input()), int(i)])

count = deque([0 for _ in range (hills)])

def sorting():
    return elevation.sort(reverse=True, key=lambda x : x[0])

def value():
    tmp = []
    for i in range(len(elevation)):
        tmp.append(elevation[i][0])
    tmp_max = max(tmp)
    tmp_min = min(tmp)
    return tmp.count(tmp_min) >= tmp.count(tmp_max)
    
maxValue = 100
minValue = 5

while (maxValue - minValue) > 17 :
    sorting()
    if value():
        elevation[0][0] -= 1
        count[elevation[0][1]] += 1
        sorting()
    else :
        elevation[-1][0] += 1
        count[elevation[-1][1]] += 1
        sorting()
    maxValue = (max(elevation, key=lambda x: x[0]))[0]
    minValue = (min(elevation, key=lambda x: x[0]))[0]

answer = 0
while count:
    k = count.popleft()
    if k != 0:
        answer += k**2

print(answer)
    

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

람다식을 이용해서 풀어보려고 했습니다만.. 시간초과가 뜨네요

sonjungwoo9 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말