본문 바로가기

알고리즘146

[Python] 백준 1138 한 줄로 서기 🧑‍💻 [Python] 백준 1138 한 줄로 서기 Gold 4 - 그리디 입력값은 서있는 사람의 기준으로 왼쪽에, 자신보다 키가 큰 사람의 숫자를 의미한다 예) 키가 1인 사람 왼쪽에는 자신보다 큰 사람이 6명이 있다 문제풀이 result의 0의 개수를 세면서, index + 1 즉 키를 넣어주면 된다 코드 N = int(input()) result = [0] * N for index, num in enumerate(list(map(int, input().split()))): cnt = 0 i = 0 while True: if result[i] == 0 and cnt == num: result[i] = index + 1 break elif result[i] == 0: cnt += 1 i += 1 pri.. 2023. 1. 25.
[Python] 백준 7662 이중 우선순위 큐 🧑‍💻 [Python] 백준 7662 이중 우선순위 큐 Gold 4 - Heapq 최소 힙과, 최대 힙을 사용하는 문제이다 두 개의 힙을 사용하면서, 숫자 관련된 참/거짓의 여부도 파악해야 한다 처음에는 두 힙에 숫자 하나가 모두 들어간다 그리고 숫자를 하나씩 뺄 때에는, 하나의 힙에는 그 숫자가 남게 된다 True의 경우는, 두 개의 힙에 동일한 숫자가 있을 경우를 True로 넣는다 I 는 힙에 숫자를 넣는 것 / D 1 최대 힙에서 숫자를 하나 빼는 것 / D -1 최소 힙에서 숫자를 하나 빼는 것 문제풀이 명령어 I 또는 D를 제일 먼저 확인 한다 I일 경우, 최대 힙과, 최소 힙에 숫자와 인덱스를 튜플로 넣는다 숫자 : 최대 힙과 최소 힙에서, 숫자들끼리 비교를 해야 함 인덱스 : True/Fal.. 2023. 1. 23.
[Python] 백준 2109 순회강연 🧑‍💻 [Python] 백준 2109 순회강연 Gold 3 - 힙 강연을 하는데, 각 강연마다 강연료와 몇 일 안에 강연을 하는지 입력을 받는다 최대한 강연료를 많이 받을 수 있도록 코드를 짜야 한다 문제풀이 1 힙 리스트 안에, 강연료 기준으로 최대 힙으로 넣는다 즉 튜플 형태로 힘 리스트에 넣는 것 그리고 힙을 하나씩 빼면서, 강연을 해야하는 날이, False이면 True로 바꾼다 강연을 해야 하는 날이, True일 경우, 그 전날을 본다. 이것을 False를 찾을 때 까지 뒤로 탐색을 한다 만약 1일차까지 다 True일 경우는, 그 강연은 못 하는 것이다 문제풀이 2 일단 강의 날짜 기준으로 오름차순으로 정렬을 한다 그리고 강연료를 힙 안에 넣어준다 힙의 길이는 날짜와 같거나, 적어야 한다 넣었는데.. 2023. 1. 20.
[Python] 백준 1715 - 카드 정렬하기 🧑‍💻 [Python] 백준 1715 - 카드 정렬하기 Gold 4 - 그리디, 힙 heappop을 2번씩 해서, 나온 숫자들을 더해준다 더해준 숫자들을 다시 heap에 넣어준다 이것을 heap에 숫자가 2개 미만일 때까지 반복해서 해준다 문제풀이 힙에서 2개의 숫자를 pop으로 뽑아와서, 더해주는 것이다 제일 낮은 숫자끼리 더하면, 최소 숫자를 만들 수 있다 코드 import heapq N = int(input()) heap = [] for n in range(N): heapq.heappush(heap, int(input())) answer = 0 while True: if len(heap) 2023. 1. 18.
[Python] 프로그래머스 - 소수 구하기 🧑‍💻 [Python] 프로그래머스 - 소수 구하기 Level 2 처음에는 그냥 일반적인 식으로 소수 구하기를 풀었다 시간 초과 n까지의 숫자를 순회하면서, 각 숫자가 소수인지 아닌지를 찾음 에라토스테네스의 체를 사용하면 더 빠르게 소수를 구할 수 있다 2부터 시작을 해서, 2의 배수인 숫자들, 3의 배수인 숫자들, 5의 배수인 숫자들 등을 빼주는 것이다 문제풀이 코드 def solution(n): answer = 0 flag = True if n == 1: answer = 0 else: for num in range(2, n + 1): cnt = 0 for i in range(1, (num // 2) + 1): if num % i == 0: cnt += 1 if cnt > 1: flag = False .. 2023. 1. 18.
[Python] 백준 2696 - 중앙값 구하기 🧑‍💻 [Python] 백준 2696 - 중앙값 구하기 Gold 3 - heapq left와 right 힙 그리고 중간 값을 가지고 문제를 푸는 것이다 left는 최대 힙을 구하는 것이고, right은 최소힙을 구하는 것이다 middle은 제일 최근에 구한 중간 값이다 즉 수열의 원소가 middle보다 작으면 left로 들어가야 하고, 크면 right로 들어가야 한다 코드 import heapq import math T = int(input()) for _ in range(T): N = int(input()) nums = [] for _ in range(int(math.ceil(N / 10))): nums += list(map(int, input().split())) left, right = [], [].. 2023. 1. 17.
🧑‍💻 [Python] 백준 14235 - 크리스마스 선물 🧑‍💻 [Python] 백준 14235 - 크리스마스 선물 Silver 3 - Heap 입력된 값을 순회한다 입력된 값이 0이 아니면 선물을 충전하는 거점 0은 아이들의 집이다 문제풀이 입력 값들을 튜플로 받아서 리스트에 넣는다 그리고 for문을 순회하며 0인지 아닌지 구별한다 0이 아니면 선물을 충전하는 거점 즉 heap에 최대힙으로 선물에 대한 숫자를 넣는다 0이면, heap에 선물이 들어 있는지 없는지를 구별한다 heap에 숫자가 있으면, 제일 앞에 있는 숫자, 즉 제일 큰 숫자를 출력한다 없으면 해당 아이에게 줄 선물이 없다는 것. 즉 -1을 출력한다 코드 import heapq N = int(input()) stops = [] for _ in range(N): stops.append(tuple(.. 2023. 1. 16.
[Python] 백준 13904 - 과제 🧑‍💻 [Python] 백준 13904 - 과제 Gold 5 - 정렬 과제 마감일과, 과제 점수가 주어진다 하루에 한 과제를 끝낼 수 있다 즉 과제 점수 기준으로 내림차순으로 해서, 최대한 많은 점수를 얻으려고 한다 문제풀이 과제 관련 정보를 입력 받는다 그리고 일수가 있는 리스트를 만든다 이 리스트는, 어느 날에 과제를 완료했는지 확인하는 리스트다 즉 False가 있고, True로 그 날 과제를 완료했다고 표시한다 그리고 과제들을 과제 점수가 높은 과제 기준으로 내림차순으로 정렬을 한다 for문을 사용하여 과제들을 순회한다 여기서 과제를 끝내야 하는 날이 제일 기준 index가 된다 만약에 과제를 끝내야 하는 날이 False일 경우, True로 바꿔주고 answer에 과제 점수를 누적시켜준다 과제를 끝.. 2023. 1. 15.
[Python] 백준 1374 - 강의실 🧑‍💻 [Python] 백준 1374 - 강의실 Gold 5 - 정렬, Heap 최대한 적은 수의 강의실을 사용하여, 모든 강의를 진행해야 한다 문제풀이 시간 초과 강의 시간들을 강의 시작 시간 기준으로 오름차순으로 정렬했다 그리고 강의가 끝나는 시간 기준으로 다시 오름차순으로 정렬을 했다 그리고 while문을 돌면서, 강의들을 뺐다 여기서 하고 싶었던 것은 강의실 당 들어가는 강의 시간들을 구하는 것이었다 그렇게 되면 한 강의실에 대한 스케줄이 나오고, 강의 시간이 아직도 남아있으면, 다시 while문을 반복해야 한다 반복하는 것을 강의 시간이 아예 없을 때까지 진행을 하는 것이다 Heap으로 풀기 위에는 한 강의실에 대한 스케줄을 강의 시간이 다 없어질 때까지 짜는 것이었다 Heap으로 풀게 되면 강.. 2023. 1. 14.

티스토리툴바