[Python] 지뢰찾기

코딩도장 / 23.7 심사문제 : 지뢰찾기

dojang.io/mod/quiz/attempt.php?attempt=1348879&cmid=2298

---

문제

 

입출력 예

---

 

나의 풀이

 

col, row = map(int, input().split())

matrix = []
for i in range(row):
    matrix.append(list(input()))

for i in range(row):
    for j in range(col):
        if matrix[i][j] == '*':
            continue
        elif matrix[i][j] == '.':
            cnt = 0
            for y in range(i - 1, i + 2):
                for x in range(j - 1, j + 2):
                    if y < 0 or x < 0 or y >= row or x >= col:
                        continue
                    elif matrix[y][x] == '*':
                        cnt += 1
            matrix[i][j] = cnt

for i in range(row):
  for j in range(col):
    print(matrix[i][j], end='')
  print()    

 

너무 힘들었다.... 문제에 대략 두 시간은 걸린다고 되있는데 난 하루하고 두시간 고민했는데 도저히 모르겠어서 해설 봤다. 해설 봐도 모르겠어서 머리 싸매고 이틀정도 고민해서 얻어낸 답...

 

먼저 input().split을 이용해 사용자가 입력한 값을 col, row에 각 각 저장해줍니다. input에 입력한 값은 문자열로 변수에 저장되므로 map을 이용해 정수(int)로 바꿔줍니다.

그 다음 빈 리스트를 만들어줍니다. 빈 리스트에 사용자가 입력한 값이 2차원 리스트로 저장되어야 하므로 for문을 이용해 리스트에 입력값을 리스트로 만들어 넣어주고 세로값만큼 반복합니다. 이렇게 하면 입력값이 2차원 리스트로 저장됩니다. ex) .** 을 입력하면 입력값이 리스트로 변해 빈 리스트에 matrix = [[., *, *]] 로 저장되고 세로값에 3을 입력해 3이 되었다고 하면 range(3)이 되므로 이 과정을 3번 반복하게 됩니다. 이 과정을 반복하면 빈 리스트가 2차원 리스트가 됩니다. (matrix = [., *, *], [*, ., .], [., *, .])

이렇게 2차원 리스트가 만들어졌다면 for문을 이용해 리스트를 검사하여 리스트 요소가 * 이라면 건너뛰고 . 이면 주변 지뢰의 수를 출력하게 합니다. 리스트 요소가 . 이면 cnt라는 변수에 0을 주고 주변 요소들을 검사해 지뢰(*)를 발견하면 1이 증가하게 합니다. 여기서 리스트 밖을 검사하면 안되므로  y < 0 or x < 0 or y >= row or x >= col 로 조건을 주어 조건과 일치하면 건너뜁니다. 마지막으로 matrix[i][j] = cnt 로 요소를 cnt로 바꿔줍니다. 

마지막으로 한번 더 for문을 사용해서 요소를 출력합니다. * 는 건너뛰어 그대로 출력되고 .는 cnt로 바꿔어 숫자로 출력됩니다.