[1일1문] 인덱스구하기/yield

프로그래머스: 바탕화면 정리

문제

컴퓨터 바탕화면은 각 칸이 정사각형인 격자판입니다. 이때 컴퓨터 바탕화면의 상태를 나타낸 문자열 배열 wallpaper가 주어집니다. 파일들은 바탕화면의 격자칸에 위치하고 바탕화면의 격자점들은 바탕화면의 가장 왼쪽 위를 (0, 0)으로 시작해 (세로 좌표, 가로 좌표)로 표현합니다. 빈칸은 ".", 파일이 있는 칸은 "#"의 값을 가집니다. 드래그를 하면 파일들을 선택할 수 있고, 선택된 파일들을 삭제할 수 있습니다.

머쓱이의 컴퓨터 바탕화면의 상태를 나타내는 문자열 배열 wallpaper가 매개변수로 주어질 때 바탕화면의 파일들을 한 번에 삭제하기 위해 최소한의 이동거리를 갖는 드래그의 시작점과 끝점을 담은 정수 배열을 return하는 solution 함수를 작성해 주세요. 드래그의 시작점이 (lux, luy), 끝점이 (rdx, rdy)라면 정수 배열 [lux, luy, rdx, rdy]를 return하면 됩니다.

제한사항

• 1 ≤ wallpaper의 길이 ≤ 50
• 1 ≤ wallpaper[i]의 길이 ≤ 50
  • wallpaper의 모든 원소의 길이는 동일합니다.
• wallpaper[i][j]는 바탕화면에서 i + 1행 j + 1열에 해당하는 칸의 상태를 나타냅니다.
• wallpaper[i][j]는 "#" 또는 "."의 값만 가집니다.
• 바탕화면에는 적어도 하나의 파일이 있습니다.
• 드래그 시작점 (lux, luy)와 끝점 (rdx, rdy)는 lux < rdx, luy < rdy를 만족해야 합니다.

나의 풀이

def find_index(string):
    for i, c in enumerate(string):
        if c == "#":
            yield i

def solution(wallpaper):
    answer = [-1, -1, -1, -1]

    for i, string in enumerate(wallpaper):
        indices = list(find_index(string))
        if not indices:
            continue
        else:
            if answer[0] == -1:
                answer[0] = i
            answer[2] = i+1
            min_x = min(indices)
            min_y = max(indices)
            if answer[1] < 0 or answer[1] > min_x:
                answer[1] = min_x
            if answer[3] <= min_y:
                answer[3] = min_y +1
    return answer

다른 풀이

def solution(wall):
    a, b = [], []
    for i in range(len(wall)):
        for j in range(len(wall[i])):
            if wall[i][j] == "#":
                a.append(i)
                b.append(j)
    return [min(a), min(b), max(a) + 1, max(b) + 1]

좌표값 x,y를 각각의 배열로 만들어 최소값, 최대값으로 시작점, 끝점 찾기

회고

너무 복잡하게 돌아갔다.
푸는 시간도 너무 오래 걸림.

• 문자열에서 찾을 문자의 인덱스 모두 구하기

find, index 함수는 1개만 찾기 때문에 모두 구하는 방식을 찾고 싶었고 yield를 활용해 구해보았다.

def find_index(string):
    for i, c in enumerate(string):
        if c == "#":
            yield i

indices = list(find_index(string))

하지만 효율적으로 사용한건지는? 잘 모르겠다.

그냥 append를 사용했어도 비슷했을 것 같다.

참고

파이썬의 yield 키워드와 제너레이터(generator)