728x90
반응형

# 효율성 test 탈락

# 다익스트라, 힙 알고리즘 배우고 다시 풀어보기. 

https://programmers.co.kr/learn/courses/30/lessons/42891?language=python3

 

코딩테스트 연습 - 무지의 먹방 라이브

 

programmers.co.kr

# 1초에 1번씩 음식이 바뀐다. = 음식의 time -1 
# 만약 time이 0이면 다음 음식으로 넘어가야함. 이 때, 시간이 증가하면 안됨.. 
# 모든 음식이 0 이면 -1 반환하기 

def solution(food_times, k):
    answer = 0  # 현재 먹고 있는 음식의 번호 -1 
    zero = k  # 0 인 음식의 수 
    
    while zero :
        z = 0 
        for i in range(zero):   
            if food_times[answer] != 0 :
                food_times[answer] -= 1 
            else :
                z += 1 
                
            answer += 1 
            if answer == len(food_times):
                 answer = 0 
            if sum(food_times) == 0 : 
                 answer = -2
                 break
        if (answer == -2): break
        zero = z
    
    answer += 1 
    return answer
728x90
반응형

+ Recent posts