인프런 커뮤니티 질문&답변

JiyoonPark님의 프로필 이미지
JiyoonPark

작성한 질문수

파이썬 알고리즘 문제풀이 입문(코딩테스트 대비)

9. 봉우리

section3_봉우리

작성

·

201

0

 안녕하세요 선생님, 강의 너무 잘 듣고 있습니다 :)

section3_봉우리 파트를 수강 중에 있습니다.

해당 문제에서는 주변을 0으로 감싸주었으나

저는 주변을 0으로 감싸지 않고 범위 체크를 해주었습니다. 

def f(i, j):
    global count
    si = i
    sj = j
    dx = [0, 1, 0, -1]
    dy = [1, 0, -1, 0]
    for k in range(4):
        ni = si + dx[k]
        nj = sj + dy[k]
        if 0 <= ni < N and 0 <= nj < N:
            if heights[ni][nj] < heights[si][sj]:
                continue
            else:
                break
    else:
        count += 1

N = int(input())
for _ in range(N):
    heights = [list(map(int, input().split())) for _ in range(N)]
    count = 0
    for i in range(N):
        for j in range(N):
            f(i,j)

    print(count)

이렇게 푼 코드가 타임 리밋이 걸렸습니다.

선생님 강의를 듣고 선생님 풀이 코드로도 채점기를 돌려봤으나 judge(python), judge(python5) 두개 다 똑같이 시간 초과가 뜨는 상황인데 인풋 아웃풋이 알맞게 나오면 그냥 넘어가도 되는 부분일까요?

또 이런 상황에 범위체크와 0으로 주변을 감싸주는 것,

두가지 방법에 효율성 문제에서 크게 차이가 있는지 궁금합니다.

답변 2

1

김태원님의 프로필 이미지
김태원
지식공유자

감사합니다^^ 

효율성을 논할 만한 차이는 아닙니다. 별 차이 없습니다.

메인쪽에서 3중 for 문이 돌면 안됩니다. 그래서 타임리밋이 난것입니다. 아래 코드는 100점 나옵니다.

def f(i, j):
    global count
    si = i
    sj = j
    dx = [0, 1, 0, -1]
    dy = [1, 0, -1, 0]
    for k in range(4):
        ni = si + dx[k]
        nj = sj + dy[k]
        if 0 <= ni < N and 0 <= nj < N:
            if heights[ni][nj] < heights[si][sj]:
                continue
            else:
                break
    else:
        count += 1

N = int(input())
heights = [list(map(int, input().split())) for _ in range(N)]
count = 0
for i in range(N):
    for j in range(N):
        f(i, j)

print(count)

0

JiyoonPark님의 프로필 이미지
JiyoonPark
질문자

아아....  바보같은 실수를 했습니다.ㅠㅠ 

친절한 설명 감사드립니다.

JiyoonPark님의 프로필 이미지
JiyoonPark

작성한 질문수

질문하기