<aside> 💬 lv0, 행렬

행렬 속 지뢰 찾기

문제 설명

M x N 크기의 행렬이 주어집니다. 이 행렬은 0과 1로 구성되어 있으며, 1은 지뢰를 나타냅니다. 주어진 행렬에서 지뢰의 총 개수를 계산하는 코드를 작성해주세요.

제한 사항

• 행렬의 크기 M과 N은 각각 1 이상 100 이하의 정수입니다.
• 행렬의 각 요소는 0 또는 1입니다.

입출력 예

입출력 설명

각 행렬에서 1로 표시된 지뢰의 개수를 세어 출력합니다.

</aside>

😺 풀이 1. count() 함수

def solution(data):
    matrix = data **#1 초기 변수 설정**
    mine_count = 0

    for row in matrix: **#2 모든 행 순회**
        mine_count += row.count(1)

    return mine_count **#3 결과 반환**

단계별 풀이 전략

1. 초기 변수 설정

   주어진 data를 matrix라는 변수에 할당한다. 1의 개수를 세기 위한 변수 mine_count를 초기화한다.

2. 모든 행 순회

   for문을 통해 matrix의 각 행 row를 순회한다. 각 행 row에서 1의 갯수를 세고, 이를 mine_count에 더한다.

3. 결과 반환

   순회를 마친 mine_count값을 반환한다.

알아둬야 할 개념

2차원 배열

• 여러개의 1차원 배열이 모여 만들어진 배열로, 각각의 2차원 배열은 1차원 배열의 한 행을 나타낸다.

반복문

• 반복문 for으로 순회가 이루어진다.
• 이 예제에서는 for row in matrix로 각 일차원 배열, 즉 각 행에 접근하였다.

함수 count()

• 순회 가능한 자료형에서 특정 요소의 갯수를 세는 메서드. 리스트.count(요소)형태로 사용된다.
• ex) "frogy in flowers".count("f") → 2

풀이 2. 평탄화