728x90

이 문제도 시간 초과가 났다. 이진풀이를 이용하는 방법으로 풀 수 있었다.

def solution(stones, k):
    answer = 0
 
    while True:
 
        # 0이 아니라면 감소 시킴, 0이라면 다음 0까지 count 하고 count>k 면 return answer
        count = 0
 
 
        for i in range(0,len(stones)):
 
 
            if stones[i] == 0:
                count+=1
                if count>=k:
                    return answer
            else:
                count = 0
                stones[i]-=1
 
        answer+=1   
 
 
 
728x90

+ Recent posts