stack.pop(), 그리고 결과 효율적으로 저장하기 잡기술

 

보통 .pop()을 쓸 때, 리스트 이름 뒤에 .pop()을 붙여서 마지막 element를 빼냈던 것 같다.

 

그런데 어떤 변수에다가 stack.pop() 값을 할당해줄 수 있다.

 

코드를 보자.

 

 

여기 보면 기준점이 되는 index를 stack.pop()으로 할당해주었다.

 

a = stack.pop()으로 할당해주면 마지막 element의 인덱스를 가져올 수 있다.

 

 

그리고 결과를 효율적으로 저장해주는 방법인데,

 

 

코드를 보면 result를 저장할 때에

 

result = [0] * len(prices)로,

 

단순히 빈 배열인 result = [] 를 하기보다

애초에 0이 각 위치에 들어가있는 상태로 초기화를 시킨다.

 

이렇게 저장하면 따로 위치를 관리해줄 필요 없이 result 자체에 인덱스로 접근할 수 있다.

result[prev_index] 에다가 값을 할당해준 마지막 코드처럼.

결과 배열의 인덱스를 가지고 놀려면 단순히 빈 배열인 result = [] 를 하기보다 애초에 0이 각 위치에 들어가있는 상태로 설정하면 된다.  *result = [0] * n