[백준 알고리즘] 2630번: 색종이 만들기 (Python / 파이썬)
·
알고리즘/백준
문제 접근 n x n 행렬을 받아, 안의 모든 요소가 같다면 그대로 리턴, 같지 않다면 행과 열을 절반으로 나누어 4개의 행렬로 분할하여 각각 재귀호출한다. 재귀함수를 하나 만들었고, 하얀 색종이와 파란 색종이 영역이 각 몇 개씩인지 세야 하므로 count[]라는 배열을 만들어 영역이 1로 채워진 경우 count[1]에, 영역이 0으로 채워진 경우 count[0]에 더해지도록 했다. 가장 중요한 부분은 행렬을 4개로 분할하는 것인데, 파이썬의 slicing을 사용했다. 현재 행(열) 사이즈를 절반으로 나누어 new_size에 할당하고 이를 이용해 2*2 for loop를 돌며 새로 만들(분할된) 행렬의 범위를 슬라이싱으로 정해주었다. 정답 코드 #2630: 색종이 만들기 import sys def cou..
[백준 알고리즘] 1012번: 유기농 배추 (Python / 파이썬)
·
알고리즘/백준
문제 접근 간단한 bfs 너비우선탐색 문제이다. 1) 밭을 돌면서 1을 발견할 때마다 bfs를 실행한다. 2) bfs 안에서, 이어진 1인 곳들을 방문하며 0으로 바꿔놓는다. 3) bfs가 끝날 때마다 정답에 1을 더한다. 정답 코드 #1012번: 유기농 배추 import sys from collections import deque def bfs(field, x, y): queue = deque() queue.append((x, y)) while queue: cur = queue.popleft() dx, dy = [0, 1, -1, 0], [1, 0, 0, -1] for i in range(4): nx, ny = cur[0] + dx[i], cur[1] + dy[i] if 0
[백준 알고리즘] 7576번: 토마토 (Python / 파이썬)
·
알고리즘/백준
문제 접근 헷갈렸던 문제이다. bfs를 사용하는데 시작 지점이 여러개다. 단순히 queue 사이클이 몇번 돌아갔는지 체크하면 당연히 더 큰 값이 나와 틀릴 것이기 때문에 한 사이클에 몇개의 요소가 큐에 들어갔고, 그것들에 의해서는 몇개가 큐에 추가되었는지 체크해야 한다. temp, count, answer 세 개의 변수를 사용해서 특정 주기에 초기화시켜서 풀었다. 예를 들어, 시작점(1)이 2개, 그 시작점으로 인해 생긴 것(2)이 2개, (2)에 의해 3개가 큐에 들어갔다면 날짜 카운트를 bfs 반복 주기의 2번째, 4번째(2+2), 7번째(4+3)마다 해줘야 한다. 내가 짠 대로 하면 마지막 토마토까지 다 익고 나서 그 토마토가 영향을 주는 시간까지 고려되기 때문에 -1을 해준다. 불가 판단(-1 ..
[백준 알고리즘] 2504번: 괄호의 값 (Python / 파이썬)
·
알고리즘/백준
문제 접근 쉬운 문제인 줄 알았는데 시간이 많이 걸렸다. 올바른 괄호 체크는 ], )가 들어올 때만 체크해주면 된다. 숫자 계산이 헷갈리는데, 곱연산에 (,[ 왼쪽 괄호를 사용하고 합연산에 ),] 오른쪽 괄호를 사용한다. 이를 위해 temp와 answer 두 가지 변수를 선언했다. (()[[]]) 이 예제를 보고 설명해보면, 이는 2 * ( 2 + 3 * 3 ) = 22 로 나타낼 수 있다. 왼쪽 괄호가 들어올 때 1로 선언된 temp에 2 또는 3을 곱해준다. 그리고 (), []와 같이 바로 닫힐 때 answer에 temp를 더해준다. 이때 바로 닫히는 (), []와 같은 형태에서 합연산을 해주는 이유는 그것이 가장 깊은 괄호이기 때문이다. (), []의 형태가 등장할때까지 스택에는 곱해야 하는 (, ..
[백준 알고리즘] 10799번: 쇠막대기 (Python / 파이썬)
·
알고리즘/백준
문제 접근 () 와 같이 내부가 없는 괄호는 레이저이다. 이외의 괄호들은 쇠막대를 표현한다. 이 부분을 보자. 스택에는 (, (, ( 3개가 들어가 있는 상태에서 레이저가 들어온다. 스택에 무언가 들어가 있다는 것은, 쇠막대기의 끝이 아직 발견되지 않았다는 뜻이고 이는 곧 레이저에 의해 영향을 받는다는 의미가 된다. 하나의 막대가 잘리면 두개가 된다. 따라 레이저가 들어올 때 현재 스택의 길이만큼 카운트를 추가해주면 된다. 이때 주의사항으로는 쇠막대기의 시작 지점에서 카운트를 1씩 미리 해줘야한다. 정답 코드 #10799번: 쇠막대기 import sys str = sys.stdin.readline().rstrip() stack = [] count = 0 for i in range(0, len(str)-1..
[백준 알고리즘] 5430번: AC ( Python / 파이썬 )
·
알고리즘/백준
문제 설명 문제 접근 reverse(), delete()는 각 O(N)의 시간이 걸린다. 따라 70만 합 조건을 고려하여 계산했을 때 시간초과가 난다. D가 첫 번째 수를 버리는 것이므로 deque의 pop(popleft)를 사용해 시간을 줄일 수 있다. 이는 평균적으로 O(1)의 시간이 걸린다. R 또한 해결해야 하는데 두번 뒤집기(RR) = 처음과 똑같은 상태, 세번 뒤집기(RRR) = 한번 뒤집기(R) 라는 것을 이용한다. 뒤집은 횟수만 기억하였다가, 마지막에 홀수인지 짝수인지 판단만 하면 된다. 그러나 뒤집힌 상태에서 D가 수행되는 경우 이는 pop 또는 popleft (처음 부분 삭제 or 마지막 부분 삭제)로 나뉠 수 있다. 이를 R이 몇번째 등장하였는지 카운트하여 D가 수행되는 시점에서 뒤집..
[백준 알고리즘] 1021번: 회전하는 큐 (Python / 파이썬)
·
알고리즘/백준
문제 설명 지민이는 N개의 원소를 포함하고 있는 양방향 순환 큐를 가지고 있다. 지민이는 이 큐에서 몇 개의 원소를 뽑아내려고 한다. 지민이는 이 큐에서 다음과 같은 3가지 연산을 수행할 수 있다. 첫 번째 원소를 뽑아낸다. 이 연산을 수행하면, 원래 큐의 원소가 a1, ..., ak이었던 것이 a2, ..., ak와 같이 된다. 왼쪽으로 한 칸 이동시킨다. 이 연산을 수행하면, a1, ..., ak가 a2, ..., ak, a1이 된다. 오른쪽으로 한 칸 이동시킨다. 이 연산을 수행하면, a1, ..., ak가 ak, a1, ..., ak-1이 된다. 큐에 처음에 포함되어 있던 수 N이 주어진다. 그리고 지민이가 뽑아내려고 하는 원소의 위치가 주어진다. (이 위치는 가장 처음 큐에서의 위치이다.) 이때..
[백준 알고리즘] 2493번: 탑 (Python / 파이썬)
·
알고리즘/백준
문제 접근 간단히 생각해보면 오른쪽 -> 왼쪽으로 레이저를 쏴주므로 탑 리스트를 역순회하면서 오른쪽보다 큰 값인지 확인해야 한다. 6->5->4->7 과 같이 순회하는 경우 6,5,4 높이의 탑은 모두 7에서 수신한다. 따라 아직 수신되지 못한 신호를 보낸 탑들을 저장해 두어야 하는데 스택이 적합하다. 하나의 탑을 체크할때마다 스택의 top을 계속 확인하면 될것 같다. 스택이 비지 않았다면 top에 쌓여있는 걸 현재 탑의 높이와 비교하면서, 현재탑보다 낮은 것이 쌓여있다면 그것들은 다 pop하고 정답 배열에 추가하거나 바로 출력하면 된다. 위 작업을 하고 나서 스택에 현재 탑을 넣어준다. 중요한 부분은, 스택에 쌓여 있는 탑들과 현재 탑을 비교하고, pop하는 거까지는 문제가 없어도 그 스택에 쌓인 탑들..
[백준 알고리즘] 3273번: 두 수의 합 (Python / 파이썬)
·
알고리즘/백준
문제 접근 이중포문을 돌리면 쉽게 풀 수 있지만, 시간초과가 날 것 같아 계산을 해봤다. 파이썬 기준 제한시간 1초 => 1*3+2 = 5초, 5 * 2000만번(연산) = 약 1억번 연산이 가능 n이 최대 10만이므로 O(n^2)의 알고리즘을 적용할 경우 최악의 경우 10,000,000,000(백억번)의 연산을 해야 한다. 즉 시간초과가 나므로 다른 방법을 사용하는데, 투 포인터 방법이 정석인 것 같고 파이썬의 경우 in 연산자로 푸는 것도 가능하다. 정답 코드 1 (투 포인터) import sys n = int(sys.stdin.readline()) a = sorted(list(map(int, sys.stdin.readline().split(" ")))) x = int(sys.stdin.readlin..
[백준 알고리즘] 14502번: 연구소 (Python / 파이썬)
·
알고리즘/백준
문제 인체에 치명적인 바이러스를 연구하던 연구소에서 바이러스가 유출되었다. 다행히 바이러스는 아직 퍼지지 않았고, 바이러스의 확산을 막기 위해서 연구소에 벽을 세우려고 한다. 연구소는 크기가 N×M인 직사각형으로 나타낼 수 있으며, 직사각형은 1×1 크기의 정사각형으로 나누어져 있다. 연구소는 빈 칸, 벽으로 이루어져 있으며, 벽은 칸 하나를 가득 차지한다. 일부 칸은 바이러스가 존재하며, 이 바이러스는 상하좌우로 인접한 빈 칸으로 모두 퍼져나갈 수 있다. 새로 세울 수 있는 벽의 개수는 3개이며, 꼭 3개를 세워야 한다. 예를 들어, 아래와 같이 연구소가 생긴 경우를 살펴보자. 2 0 0 0 1 1 0 0 0 1 0 1 2 0 0 1 1 0 1 0 0 0 1 0 0 0 0 0 0 0 0 0 0 1 1 ..