swea. D3- 1220. [S/W 문제해결 기본] 5일차 - Magneti python

🔗문제링크

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=3&contestProbId=AV14hwZqABsCFAYD&categoryId=AV14hwZqABsCFAYD&categoryType=CODE&problemTitle=&orderBy=RECOMMEND_COUNT&selectCodeLang=ALL&select-1=3&pageSize=10&pageIndex=2

SW Expert Academy

 

 

 

 

[입력]

10개의 테스트 케이스가 주어진다.

각 테스트 케이스의 첫 번째 줄에는 정사각형 테이블의 한 변의 길이가 주어진다. (이 값은 항상 100이다)

그 다음 줄부터 100 x 100크기의 테이블의 초기 모습이 주어진다. 1은 N극 성질을 가지는 자성체를 2는 S극 성질을 가지는 자성체를 의미하며 테이블의 윗부분에 N극이 아래부분에 S극이 위치한다고 가정한다.

(N극 성질을 가지는 자성체는 S극에 이끌리는 성질이 있다.)

[출력]

#부호와 함께 테스트 케이스의 번호를 출력하고, 공백 문자 후 교착 상태의 개수를 출력한다.

 

 

 

---

 

생각보다 단순한 문제였다! 

flag 변수와 n극 먼저 만나고 s극을 만나게 되면 result 변수에 +1을 해주면 쉽게 풀렸다. 

 

t=10
for tc in range(1,t+1):
    n=int(input())
    arr = [list(map(int,input().split())) for _ in range(n)]
    result=0

    for i in range(n):
        flag=0
        for j in range(n):
            if arr[j][i]==1:
                flag=1
            elif arr[j][i]==2:
                if flag:
                    result+=1
                    flag=0
    print(f'#{tc} {result}')

세로로 탐색하기 때문에 j가 행 부분에 있다는 것