백준 BAEKJOON 21965번 드높은 남산 위에 우뚝 선 [PYTHON/파이썬]

백준 BAEKJOON 21965번 드높은 남산 위에 우뚝 선 [PYTHON/파이썬]

---

<문제 출처>

https://www.acmicpc.net/problem/21965

21965번: 드높은 남산 위에 우뚝 선

<풀이>

산의 높이를 리스트로 받은 뒤 산의 모양인지 아닌지 확인해주면 된다.

산이 맞는지 아닌지 확인하는 check 변수와 산이 모양이 올바른지 확인하는 shape 변수를 True로 지정하여 풀었다.

(내 코드에서는 for 문 속에 있는 else 부분을 이해하면 풀기 쉬워진다.)

 

i 번째 산의 높이가 i+1 번째 산의 높이보다 작다면 산이 맞으나

shape이 False라면 산이 아니기 때문에 check을 False로 바꾸고 break

i 번째 산의 높이와 i+1 번째 산의 높이가 같다면 그건 산이 아니기 때문에 check을 False로 바꾸고 break

i 번째 산이 높이가 i+1 번째 산의 높이 보다 크다면 shape을 False로만 바꾸고 for 문을 계속 진행

 

즉, 산의 높이가 처음부터 증가하는지 확인하고 중간에 같은 높이라면 False

증가하다가 감소하는 과정에서 다시 또 증가하면 False가 되는 것이다.

 

<코드>

N = int(input())
A = list(map(int, input().split()))
check = True	# 산이 맞는지 아닌지 확인 변수
shape = True 	# 올라가다 내려오고 다시 올라가는지 확인 변수

for i in range(N-1) :
    if A[i] < A[i+1] :
        if shape == False :	# 올라가고 있으나 한 번 내려간 적이 있다면 break
            check = False
            break
    elif A[i] == A[i+1] :	# 높이가 같으면 break
        check = False
        break
    else :	# 올라가다가 감소할 때
        shape = False

if check :	# 산이면 YES, 아니면 NO
    print('YES')
else :
    print('NO')