one Book a Day

재귀 dp로 안쪽에서부터 경우의 수를 뽑아오며 밖으로까지 cnt 값을 누적시켜 꺼내왔습니다. import sys input = sys.stdin.readline ans = list(map(int, input().split())) dp = [[[[-1 for score in range(11)] for pre2 in range(6)] for pre in range(6)] for depth in range(11)] def make_dp(depth, pre, pre2, score): if dp[depth][pre][pre2][score] != -1: return dp[depth][pre][pre2][score] if depth == 10: return 1 if score >= 5 else 0 cnt = 0 fo..

정렬 후 투포인터로 풀면 됩니다. import sys input = sys.stdin.readline N = int(input()) M = int(input()) arr = sorted(map(int, input().split())) left = 0 right = N - 1 answer = 0 while left M: right -= 1 else: left += 1 print(answer)

로직이 완전 깔끔하지는 못하지만 시간 내에 통과하도록 최소한의 연산을 하도록 노력함. 기본적으로 재귀를 통해 폭탄을 한발씩 떨어뜨리는 함수, 폭탄이 블록과 만나면 연쇄 폭발을 일으키는 재귀함수 폭발 후 빈칸에 블록을 중력을 적용해 채우는 함수. 3가지 함수와 남은 블록을 카운팅해주는 함수 1개의 함수 총 4개의 함수로 구성했으며 중간에 블록이 모두 터졌을 경우 바로 종료하지는 못했으나 더이상 N을 파고들지 않아도 W를 0~ W-1까지 순회해서 continue로 넘어가게 만듦. 모두 터지지 않을 경우의 로직은 효율적으로 짰다고 생각함. def delete_block(board, x, y, cnt): board[x][y] = 0 if cnt > 1: for i in range(1, cnt): for dx, ..

조합을 dfs로 짜고, 조합에 속한 요소와 속하지 못한 요소간 차이를 비교. 삼성기출 스타트와 링크. 웰노운 def choose_parts(cur_node, cnt): global min_gap if cnt == N // 2: S1, S2 = 0, 0 for i in range(N - 1): for j in range(i + 1, N): if check[i] and check[j]: S1 += S[i][j] + S[j][i] elif not check[i] and not check[j]: S2 += S[i][j] + S[j][i] min_gap = min(min_gap, abs(S1 - S2)) for next_node in range(cur_node, N): if not check[next_node]:..

dfs 분기를 이용해 사각형을 그리도록 직선이동 시킴. 간만큼 돌아오도록 dx, dy를 이용해 통제. # 기본적으로 방향성 유지 이동, 간만큼 돌아오도록 dx, dy 변수가 늘어났다가 0으로 돌아오게끔 통제. def dfs(x, y, dx, dy, direction, cnt): global max_size nx, ny = x + delta[direction][0], y + delta[direction][1] # 속행 if 0

최소 이동거리 문제로 인접리스트가 주어져 다익스트라로 풀이했습니다. from heapq import heappush, heappop def dijkstra(start): heap = [] dists[start] = 0 heappush(heap, (0, start)) while heap: cur_dist, cur_node = heappop(heap) for next_node, next_dist in graph[cur_node]: dist = cur_dist + next_dist if dists[next_node] > dist: dists[next_node] = dist heappush(heap, (dist, next_node)) for test in range(1, int(input()) + 1): N, ..

가중치가 있는 2차원 배열의 최소거리를 구하는 문제로 다익스트라를 사용했씁니다. from heapq import heappush, heappop def dijkstra(x, y): heap = [] dists[x][y] = 0 heappush(heap, (0, 0, 0)) while heap: cur_dist, x, y = heappop(heap) for dx, dy in delta: nx, ny = x + dx, y + dy if 0

기본적인 MST 문제로 간선이 주어졌기 때문에 크루스칼 알고리즘을 사용해 풀이했습니다. from heapq import heappop, heappush def find(x): if parents[x] == x: return x parents[x] = find(parents[x]) return parents[x] def union(a, b): a = find(a) b = find(b) if a > b: a, b = b, a parents[b] = a for test in range(1, int(input()) + 1): V, E = map(int, input().split()) parents = {i: i for i in range(V + 1)} edges = [] for __ in range(E): a,..

보자마자 price(1~count) 가 생각나서 제출. 했으나 제출 누르자마자 음수인 경우 예외처리 안해줬다는 생각이 나서 결과보니 아니나다를까 testcase 하나 오답.. 음수처리 해주었더니 통과 def solution(price, money, count): answer = price * count * (count + 1) // 2 - money if answer < 0: answer = 0 return answer

중앙값 계속 체크해서 반환. 슬라이딩 윈도우처럼 돌리는건 불가능할까? 사실 세로로 삽입하면 한줄씩은 3개씩 빼고 넣는게 가능할지도? import sys input = sys.stdin.readline def check_mid(x, y): tmp = [] for i in range(x, x + 3): for j in range(y, y + 3): if tmp: if tmp[-1] = T: return True return False R, C = map(int, input().split()) filters = [list(map(int, input().split())) for __ in range(R)] T = int(input()) cnt = 0 for i in range(R - 2): for j in ra..