코딩테스트로 Python 연습하기 - 2차원 배열

문제 내용

 

이차원 배열 arr이 주어졌을 때

배열 arr의 모든 값이

서로 같으면 1을 return

하나라도 다르면 0을 return 해라.

 

이차원 배열이 익숙하지 않으면

처음엔 어려울 수도 있다.

 

일단 여기서는 모든 값을

하나한 비교해야 하기때문에

2중 반복문을 써주고

그 안에서 조건문으로 걸러주면 될 것 같다.

 

문제 풀이

 

3번 라인 : 반복문을 이용해서 행의 길이를 구해준다.

4번 라인 : 반복문을 이용해서 열의 길이를 구해준다.

└ 문제에서 n x n 배열이라고 했으니 4번라인의

arr[i]를 3번 라인처럼 arr 이라고 써도 된다.

 

5번라인 : 조건문을 써서 문제에 주어진 조건을

충족하는지 알아본다.

└ 한개라도 다르면 0을 반환해야하기 때문에

조건문에서 != 를 사용하여 틀린것을 골라내는게 쉽다.

그렇게 한개라도 틀리면 0으로 초기화해둔 answer을

바로 반환해서 끝내준다.

 

8번 라인 : 2중 반복문을 빠져나왔다는 것은 조건을

모두 충족시킨 것이기 때문에 answer에 1을 저장해주고

반환해준다.

 

너무 오랜만이라 다 까먹어서 생각해내는데

간단한 문제라도 시간이 좀 걸렸지만

고민하면서 문제를 풀어내니까

꽤 재밌는 시간이었다.