[프로그래머스] Lv.0 - 그림 확대.py

문제

• 직사각형 형태의 그림 파일이 있고, 이 그림 파일은 1 × 1 크기의 정사각형 크기의 픽셀로 이루어져 있습니다. 이 그림 파일을 나타낸 문자열 배열 picture과 정수 k가 매개변수로 주어질 때, 이 그림 파일을 가로 세로로 k배 늘린 그림 파일을 나타내도록 문자열 배열을 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ picture의 길이 ≤ 20
• 1 ≤ picture의 원소의 길이 ≤ 20
• 모든 picture의 원소의 길이는 같습니다.
• picture의 원소는 '.'과 'x'로 이루어져 있습니다.
• 1 ≤ k ≤ 10

입출력 예시

• 입출력 예 1
  • 예제 1번의 picture는 다음과 같습니다.

.xx...xx.
x..x.x..x
x...x...x
.x.....x.
..x...x..
...x.x...
....x....

• 이를 가로 세로로 k배, 즉 2배 확대하면 그림 파일은 다음과 같습니다.

..xxxx......xxxx..
..xxxx......xxxx..
xx....xx..xx....xx
xx....xx..xx....xx
xx......xx......xx
xx......xx......xx
..xx..........xx..
..xx..........xx..
....xx......xx....
....xx......xx....
......xx..xx......
......xx..xx......
........xx........
........xx........

• 따라서 ["..xxxx......xxxx..", "..xxxx......xxxx..", "xx....xx..xx....xx", "xx....xx..xx....xx", "xx......xx......xx", "xx......xx......xx", "..xx..........xx..", "..xx..........xx..", "....xx......xx....", "....xx......xx....", "......xx..xx......", "......xx..xx......", "........xx........", "........xx........"]를 return 합니다.
• 입출력 예 2
  • 예제 2번의 picture는 다음과 같습니다.

x.x
.x.
x.x

• 이를 가로 세로로 k배, 즉 3배 확대하면 그림 파일은 다음과 같습니다.

xxx...xxx
xxx...xxx
xxx...xxx
...xxx...
...xxx...
...xxx...
xxx...xxx
xxx...xxx
xxx...xxx

• 따라서 ["xxx...xxx", "xxx...xxx", "xxx...xxx", "...xxx...", "...xxx...", "...xxx...", "xxx...xxx", "xxx...xxx", "xxx...xxx"]를 return 합니다.

코드 1

def solution(picture, k):
    answer = []
    new_pic = ''
    new_picture = []
    for pic in picture:
        for pic_times in pic:
            new_pic += pic_times * k
        for _ in range(k):
            new_picture.append(new_pic)
        new_pic = ''
    answer = new_picture
    return answer

코드 2

def solution(picture, k):
    answer = []
    for num in range(len(picture)):
        answer += [picture[num].replace('.', '.'*k).replace('x', 'x'*k)] * k
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges