Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | 6 | 7 |
8 | 9 | 10 | 11 | 12 | 13 | 14 |
15 | 16 | 17 | 18 | 19 | 20 | 21 |
22 | 23 | 24 | 25 | 26 | 27 | 28 |
29 | 30 |
Tags
- level3
- 코딩테스트
- 가상메모리
- 프로그래머스
- 재귀
- BFS
- 운영체제
- 딕셔너리
- BOJ
- dict
- 구현
- level2
- python
- 브루트포스
- DP
- level1
- level0
- programmers
- 스택
- 다익스트라
- 힙
- 다이나믹 프로그래밍
- 수학
- 백준
- dfs
- MYSQL
- 파이썬
- N과M
- 가상메모리 관리
- 그리디
Archives
- Today
- Total
동캄의 코딩도장
프로그래머스 level2 [뒤에 있는 큰 수 찾기] 파이썬 본문
https://school.programmers.co.kr/learn/courses/30/lessons/154539
# 프로그래머스 level2 뒤에 있는 큰 수 찾기
def solution(numbers):
answer = [-1] * (len(numbers)) #전체 -1로 세팅
stack=[] #값을 임시로 저장할 스택
for i in range(len(numbers)-1):
now=numbers[i] #현재 값
next=numbers[i+1] #다음 값
if now>=next: #현재 값 >= 다음 값 이면
stack.append((i,now)) # 스택에 현재 인덱스, 현재 값 저장
else: #현재 값< 다음 값이면
answer[i]=next #answer[i]에 다음 값 저장
while stack: # 스택이 있다면
idx,val=stack.pop() #스택의 값을 pop 해서
if val<next: #값이 다음 값(i+1번째 값)보다 작으면
answer[idx]=next # answer[idx]값 갱신
else:
stack.append((idx,val)) # 그렇지 않다면, 다시 스택에 삽입
break
return answer
# print(solution([2,3,3,5]))
# print(solution([9, 1, 5, 3, 6, 2]))
스택이 생각이 안나서 헤맸다...
모든 인자를 한 번씩 stack에 넣고, 조건(stack[-1]<numbers[i])이 만족되면 pop 해주면 되는 문제였다.
더욱 깔끔하게 푸신 분의 풀이를 첨부한다.
def solution(numbers):
stack = []
result = [-1] * len(numbers)
for i in range(len(numbers)):
while stack and numbers[stack[-1]] < numbers[i]:
result[stack.pop()] = numbers[i]
stack.append(i)
return result
'코테 > 프로그래머스' 카테고리의 다른 글
프로그래머스 level2 [롤케이크 자르기] 파이썬 (0) | 2023.04.06 |
---|---|
프로그래머스 level1 [과일 장수] 파이썬 (0) | 2023.04.06 |
프로그래머스 level2 [숫자 변환하기] 파이썬 (0) | 2023.04.05 |
프로그래머스 level2 [할인 행사] 파이썬 (0) | 2023.04.05 |
프로그래머스 level2 [연속 부분 수열 합의 개수] 파이썬 (0) | 2023.04.05 |