83번 빛 섞어 색 만들기

https://codeup.kr/problem.php?id=6083 

[기초-종합] 빛 섞어 색 만들기(설명)(py)

 

r, g, b = map(int, input().split())

color_list = []

for i in range(r):
    for j in range(g):
        for k in range(b):
            color_list.append([i,j,k])

[print(*color_list[i]) for i in range(len(color_list))]
print(len(color_list))

처음에 이렇게 리스트를 이용해 풀었지만 메모리초과가 났다.

리스트를 사용하지 않고 바로 print를 이용해 출력하도록 코드를 수정했지만 위 코드에서 리스트 컴프리헨션을 통해 인덱스를 반복문으로 돌리며 리스트를 벗기는 형식으로 출력한 코드 접근이 마음에 들어 포스팅을 하게 되었다.

 

아래 코드는 정답을 맞추기 위해 print로 수정한 코드이다.

r, g, b = map(int, input().split())

for i in range(r):
    for j in range(g):
        for k in range(b):
            print(i,j,k)

print(r*g*b)