흙금이네 블로그

아이디어 차례로 학생 자리를 교실에서 비어 있는 자리 중 규칙에 맞는 자리로 정하고, 학생의 만족도 총합을 구한다. 풀이 import sys input = sys.stdin.readline delta = [(-1, 0), (0, 1), (1, 0), (0, -1)] def solution(): N = int(input()) classroom = [[0]*N for _ in range(N)] likes = [[] for _ in range(N**2+1)] for _ in range(N**2): student, *numbers = map(int, input().split()) likes[student] = numbers max_like = max_empty = 0 pos = [] for r in range(..

아이디어 버스 출발 시간을 구하여 정렬한 후, 이분 탐색으로 탈 수 있는 가장 빠른 버스 출발 시간과의 차이를 구한다. 풀이 import sys input = sys.stdin.readline def solution(): N, T = map(int, input().split()) bus = set() for _ in range(N): S, I, C = map(int, input().split()) for i in range(C): bus.add(S+I*i) bus = sorted(bus) if T > bus[-1]: print(-1) else: s = 0 e = len(bus)-1 while s T: e = m-1 elif bus[m] < T: s = m+1 else: print(0) return pri..

아이디어 이분 탐색으로 조건을 만족하는 가장 작은 음이 아닌 정수 제곱근을 구한다. 풀이 def solution(): N = int(input()) s = 0 e = N while s N: e = m-1 elif m**2 < N: s = m+1 else: print(m) return print(s) solution()

아이디어 투 포인터를 이용하여 동시에 가지고 있는 CD의 개수를 구한다. 풀이 import sys input = sys.stdin.readline def solution(): while 1: N, M = map(int, input().split()) if N == 0 and M == 0: break CD1 = [int(input()) for _ in range(N)] CD2 = [int(input()) for _ in range(N)] res = i = j = 0 while i CD2[j]: j += 1 else: i += 1 print(res) solution()

아이디어 이분 탐색으로 N의 제곱근을 찾는다. 풀이 def solution(): N = int(input()) s = 1 e = N while 1: m = (s+e)//2 if m**2 > N: e = m-1 elif m**2 < N: s = m+1 else: print(m) return solution()

아이디어 조건에 따라 주사위가 이동할 때마다 주사위와 지도의 상태를 변경한다. 풀이 import sys input = sys.stdin.readline delta = [(0, 1), (0, -1), (-1, 0), (1, 0)] dice = [0]*6 turned = [ (3, 1, 0, 5, 4, 2), (2, 1, 5, 0, 4, 3), (4, 0, 2, 3, 5, 1), (1, 5, 2, 3, 0, 4) ] def solution(): N, M, x, y, K = map(int, input().split()) board = [list(map(int, input().split())) for _ in range(N)] for d in map(int, input().split()): d -= 1 nx, ..

아이디어 재귀를 통해 계산에 필요한 수들을 구해 나간다. 풀이 def find_A(i): if A.get(i, 0): return A[i] A[i] = find_A(i//P)+find_A(i//Q) return A[i] N, P, Q = map(int, input().split()) A = dict() A[0] = 1 print(find_A(N))

아이디어 유니온 파인드를 이용하여 상호 배타적 집합의 개수를 구하고, 집합 내 최소 비용의 합을 구한다. 풀이 import sys input = sys.stdin.readline def solution(): for _ in range(M): v, w = map(int, input().split()) union(v, w) for i in range(1, N+1): find(i) res = 0 for i in set(group): res += A[i] if res w: v, w = w, v group[w] = v if A[w] < A[v]: A[v] = A[w] N, M, k = map(int, input().split()) A = [0]+list(map(int, input().split())) group ..

아이디어 탄산 내성 K보다 큰 2의 거듭제곱의 개수를 구하고, 이변을 통해 승리할 수 있는 횟수를 더한다. 풀이 def solution(): N, M, K = map(int, input().split()) n = 1 res = 0 while n < N: n *= 2 if K < n: if M == 0: break M -= 1 res += 1 print(res) solution()

아이디어 최솟값을 찾는 우선순위 큐, 최댓값을 찾는 우선순위 큐, 남은 값의 개수를 저장하는 딕셔너리를 사용하여 해결한다. 풀이 우선순위 큐를 사용하기 위해 heapq 모듈의 heappush와 heappop을 불러온다. 최댓값을 저장하는 우선순위 큐 max_heap과 최솟값을 저장하는 우선순위 큐 min_heap을 생성하고, 남은 데이터의 총 개수 total, 남은 값의 개수를 저장하는 딕셔너리 cnt_dict를 생성한다. k번의 for문에서 연산이 I이면 cnt_dict를 참고해 추가하려는 값의 남은 개수가 0이면 min_heap과 max_heap에 추가하고, 그렇지 않으면 두 우선순위 큐에 추가하지 않고 cnt_dict의 값과 total을 1 증가시킨다. 연산이 D이면 cnt_dict를 참고해 남은 ..