[COS PRO 1급 #6-2] 단어를 순서대로 적으세요

문제설명

한 줄에 K자를 적을 수 있는 메모장에 영어 단어들을 적으려 합니다. 영어 단어는 정해진 순서로 적어야 하며, 단어와 단어 사이는 공백 하나로 구분합니다. 단, 한 줄의 끝에 단어 하나를 완전히 적지 못한다면, 그 줄의 나머지 부분을 모두 공백으로 채우고 다음 줄부터 다시 단어를 적습니다.

예를 들어 한 줄에 10자를 적을 수 있고, 주어진 단어가 순서대로 `["nice", "happy", "hello", "world", "hi"]` 인 경우 각 줄에 다음과 같이 적을 수 있습니다.('_'는 공백을 나타냅니다.)

* 첫째 줄 : `"nice_happy"`
* 둘째 줄 : `"hello_____"`
* 셋째 줄 : `"world_hi"`

이때, 둘째 줄에 `"hello"`를 적으면 단어를 적을 수 있는 남은 칸은 5칸이며, `"world"`를 이어서 적으려면 공백 하나를 포함하여 총 6칸이 필요합니다. 따라서 단어가 잘리게 되므로 남은 칸을 모두 공백으로 채운 후, 다음 줄에 `"world"`부터 다시 단어를 적어 나갑니다.

한 줄에 적을 수 있는 글자 수 K와 적을 단어가 순서대로 담긴 리스트 words가 매개변수로 주어질 때, 단어를 모두 적으면 몇 줄이 되는지 return 하도록 solution 함수를 완성해주세요.

매개변수 설명

한 줄에 적을 수 있는 글자 수 K와 적을 단어가 순서대로 담긴 리스트 words가 solution 함수의 매개변수로 주어집니다.

* K는 5 이상 30 이하인 자연수입니다.
* words 리스트의 길이는 1 이상 100 이하입니다.
* words 리스트에 담겨있는 모든 단어는 알파벳 소문자로만 이루어져 있으며, 각 단어의 길이는 K 이하입니다.

return 값 설명

단어를 모두 적으면 몇 줄이 되는지 return해주세요.

예시

| K  | words                                       | return |
|----|---------------------------------------------|--------|
| 10 | `["nice", "happy", "hello", "world", "hi"]` | 3      |

예시 설명

예제#1
문제의 예시와 같으며, 단어를 모두 적으면 3줄이 됩니다.

문제 코드

def solution(K, words):
    #여기에 코드를 작성해주세요.
    answer = 0
    return answer

풀이

def solution(K, words):
	answer = []
	i = 0
	while i < len(words)-1:
		temp = words[i]
        
        # 현재 글자에 _다음글자를 추가한 길이가 K 이하면 한 줄에 작성
        # 현재 글자에 _다음글자를 추가한 길이가 K 초과면 다음 줄로 넘어감
		for j in range(i+1, len(words)):
			if len(temp+"_"+words[j]) <= K:
				temp = temp + "_" + words[j]
			else:
				answer.append(temp)
				i = j
				break
			if j == len(words)-1:
				answer.append(temp)
				i = j
		
	return len(answer)