[CodeUP] ID3130 소들의 헤어스타일

어찌 저찌 stack을 활용해서 코테 문제를 풀어 블로그에 올려둔다..

솔직히 로직이나... 한글로 작성 안하고 그냥 때려 맞추기한 느낌이라 연습이 안된 것 같다...

다음 문제는 연습할 수 있도록 로직을 잘 작성해보겠다.

def test(N,cow) :
    result=int(0)
    stack=[]
    for i in range(N) :
        print(stack)
        while stack and stack[-1]<cow[i] :
            stack.pop()
            print(i,stack)
        if not stack or stack[-1] > cow[i] :
            stack.append(cow[i])
        result+=len(stack)

    return result -N

# N=6
# cow = [10,3,7,4,12,2]
N = int(input())
cow = [0]*N
for i in range(N) :
    cow[i]=int(input())

result = test(N,cow)
print(result)