[SWEA] 파이썬 220. Magnetic | 그리디 알고리즘

 

~목차~

문제

문제 해결 포인트

작성 코드

---

 

문제

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

SW Expert Academy

 

 

문제 해결 포인트

- 행 크기(100)만큼의 리스트 선언

- 한 행의 각 열을 방문하며(이중 for문) 이전에 N극 방향 자석이 있었고 지금 S극 방향 자석이 발견된다면 교착 발견

- 열 값이 0이 아니었다면(방향이 있다면/ 자석이 있다면) 해당 열 값 업데이트

 

 

작성 코드

 

for test in range(1, 11):
    n = int(input())  # 100
    result = 0
    column = [0]*100  # 의미있는 열 값 저장/ 1: N극 2: S극 0: 발견 안 됨 

    for i in range(n):  # 행
        line = list(map(int, input().split()))  # 한 행
        for j in range(100):  # 열
            if column[j] == 1 and line[j] == 2:  # 이전에 N, 지금 S
                result += 1
            if line[j] != 0:  # 열 값이 0이 아니라면(방향이 있다면)
                column[j] = line[j]
                
    print("#"+str(test), result)

 

 

 

도움이 되셨다면 좋아요 눌러주세요💚