[프로그래머스] Level1 과일장수(python)

https://school.programmers.co.kr/learn/courses/30/lessons/135808

프로그래머스

 

1. 내가 작성한 코드

while len(score) >= m:

 

으로 반복문을 돌렸더니 계속 시간초과가 발생했다.

 

그래서 for 문으로 수정해주었다.

 

def solution(k, m, score):
    
    result = 0
    score = sorted(score, reverse=True)
    
    for i in range(0, len(score), m):
        tmp = score[i:i+m]
        
        if len(tmp) == m:
            result += min(tmp) *m
        
    return result

 

 

1. range(start, end, step) 좀 써먹자

 

2. 인덱스에서 list 길이 이상의 위치를 인덱스로 가르키면 오류가 발생하지만, 슬라이싱은 존재하는 원소까지만 넣어준다. 

 

score = [1,2,3,1,2,3,1]

m = 4

for i in range(0, len(score), m):
  tmp = score[i:i+m]
  print(tmp)
  
  #[1,2,3,1]
  #[2,3,1]