[코드트리] 일렬로 서있는 소 2 / 완전탐색, 1차원배열

https://www.codetree.ai/missions/5/problems/cattle-in-a-rowing-up-2?&utm_source=clipboard&utm_medium=text 

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

 

 

 

 

일렬로 서있는 소2

---

전체 배열을 순회해야한다. 다만 i<j<k 에 대해 A[i] <= A[j] <= A[k] 를 만족해야하므로 완전탐색을 이용하면 i에 대해 가능한 모든 j, k 에 대해서 해당 조건이 맞는지 확인한다.

 

배열의 인덱스에 주의하기 위해 i는 범위가 l-2 로 j는 l-1 로 k는 l 로 설정해야한다.

그렇지 않으면 i 와 j 는 범위에 벗어날 수 있다.

 

 

 

코드

---

 

n = int(input())

cows_h = list(map(int, input().split()))

l = len(cows_h)
cnt = 0

for i in range(l-2):
    for j in range(i+1, l-1):
        for k in range(j+1, l):
            if(cows_h[i] <= cows_h[j] <= cows_h[k]):
                cnt+=1


print(cnt)

만약 조건에 맞다면 cnt 값을 증가시킨다.