‍♂️개발일지 ‍♀️

[백준] 12851번: 숨바꼭질 2

ㅊhsilver — Sun, 16 Jan 2022 16:06:04 +0900

알고리즘 정리

문제

✍ 풀이과정

import sys
from collections import deque
MAX = 100001

n, k = map(int, input().split())
queue = deque([n])
dist = [-1]*MAX
dist[n] = 0
cnt=0

while queue:
    i = queue.popleft()

    if i == k:
        cnt+=1

    if i*2 < MAX and dist[i*2] == -1:
        queue.append(i*2)
        dist[i*2] = dist[i]+1
    if i-1>=0 and dist[i-1] == -1:
        queue.append(i-1)
        dist[i-1] = dist[i] + 1
    if i+1< MAX and dist[i+1] == -1:
        queue.append(i + 1)
        dist[i+1] = dist[i] + 1

print(dist[k])
print(cnt)

3번을 먼저 풀어서 3번에서 appendleft를 append로 바꾸고

i==k일 때마다 cnt를 1씩 추가하는 것으로 수정하였다.

출력해봤을 때, 최소 시간은 맞지만 방법 수가 틀리게 나왔다.

생각해보니 처음 방문했을 때만 queue에 추가했기 때문에 1만 출력되는 것이었다.

import sys
from collections import deque
MAX = 100001

n, k = map(int, input().split())
queue = deque([n])
dist = [-1]*MAX
dist[n] = 0
cnt=0

while queue:
    i = queue.popleft()

    if i == k:
        cnt+=1

    if i*2 < MAX:
        if dist[i*2] == -1 or dist[i*2] == dist[i]+1:
            dist[i*2] = dist[i]+1
            queue.append(i * 2)
    if i-1>=0:
        if dist[i-1] == -1 or dist[i-1] == dist[i]+1:
            dist[i-1] = dist[i]+1
            queue.append(i - 1)
    if i+1< MAX:
        if dist[i+1] == -1 or dist[i+1] == dist[i]+1:
            dist[i+1] = dist[i]+1
            queue.append(i + 1)

print(dist[k])
print(cnt)

반례를 생각해보다가 도무지 모르겠어서 블로그를 참고했다.

좀 거지같은디,,,이러한 경우에도 추가를 해줘야 한다!

[백준] 13549번: 숨바꼭질 3

ㅊhsilver — Sun, 16 Jan 2022 14:58:16 +0900

알고리즘 정리

문제

https://www.acmicpc.net/problem/13549

13549번: 숨바꼭질 3

수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다. 수빈이는 걷거나 순간이동을 할 수 있다. 만약, 수빈이의 위치가 X일

www.acmicpc.net

✍ 풀이과정

import sys
from collections import deque
MAX = 100001

n, k = map(int, input().split())
queue = deque([n])
dist = [-1]*MAX
dist[n] = 0

while queue:
    i = queue.popleft()

    if i == k:
       print(dist[i])
       break

    if i*2 < MAX and dist[i*2] == -1:
        queue.appendleft(i*2)
        dist[i*2] = dist[i]
    if i-1>=0 and dist[i-1] == -1:
        queue.append(i-1)
        dist[i-1] = dist[i] + 1
    if i+1< MAX and dist[i+1] == -1:
        queue.append(i + 1)
        dist[i+1] = dist[i] + 1

옛날에 풀어본 문제였고 bfs로 풀었다.

(문제 A->B)에서는 +"1"과 *2밖에 없어서 뒤로 돌아오지 못하는 반면에

이 문제는 -1도 있기 때문에 dist를 두어 해당 위치에 방문했는지를 확인해야 한다.

* (3 -> 4 -> 3) 같은 경우를 방지하기 위해서

(*2)를 하는 경우에는 초가 증가하지 않으므로 appendleft를 해주었다.

[백준] 16953번: A->B

ㅊhsilver — Sun, 16 Jan 2022 14:19:28 +0900

알고리즘 정리

문제

https://www.acmicpc.net/problem/16953

16953번: A → B

첫째 줄에 A, B (1 ≤ A < B ≤ 109)가 주어진다.

www.acmicpc.net

✍ 풀이과정

import sys
from collections import deque

a, b = map(int, input().split())
queue = deque([(a, 1)])

while queue:
    i, cnt = queue.popleft()

    if i==b:
        print(cnt)
        exit(0)

    queue.append((i * 2, cnt+1))
    queue.append((int(str(i)+"1"), cnt+1))

연산의 최솟값을 구하는 것이기 때문에 bfs로 풀었다.

처음에 특별한 조건없이 i에 2를 곱했을 때, 1을 붙였을 때를 모두 추가했는데 "메모리 초과"가 되었다.

import sys
from collections import deque

a, b = map(int, input().split())
queue = deque([(a, 1)])

res=-1
while queue:
    i, cnt = queue.popleft()

    if i==b:
        res = cnt
        break

    if i*2<=b:
        queue.append((i * 2, cnt+1))
    if int(str(i)+"1") <=b:
        queue.append((int(str(i)+"1"), cnt+1))

print(res)

A->B가 불가능한 경우와 b를 초과하는 경우들을 추가해주었더니 통과했다.

[백준] 2606번: 바이러스

ㅊhsilver — Sun, 9 Jan 2022 16:00:18 +0900

알고리즘 정리

문제

https://www.acmicpc.net/problem/2606

2606번: 바이러스

첫째 줄에는 컴퓨터의 수가 주어진다. 컴퓨터의 수는 100 이하이고 각 컴퓨터에는 1번 부터 차례대로 번호가 매겨진다. 둘째 줄에는 네트워크 상에서 직접 연결되어 있는 컴퓨터 쌍의 수가 주어

www.acmicpc.net

✍ 풀이과정

import sys
from collections import deque

n = int(input())
e = int(input())

graph =[[] for _ in range(n+1)]
visited =[0] * (n+1)

for _ in range(e):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)

queue = deque()
queue.append(1)
visited[1] = 1
cnt=0

while queue:
    cur = queue.popleft()

    for i in graph[cur]:
        if visited[i] == 0:
            queue.append(i)
            visited[i] = 1
            cnt+=1
print(cnt)

사실 bfs, dfs 중에 어느 것을 쓰는 게 더 효율적인지는 생각해보진 않았는데 bfs가 편해서 bfs로 풀었다.

앞선 '음식물 쓰레기', '미로찾기' 문제와 동일하게 visited 리스트를 생성하였고 graph 리스트도 생성하였다.

queue에 방문하지 않은 인접한 노드를 추가해주고 cnt를 늘려주면서 1번과 연결될 수 있는 노드의 수를 구했다.

[백준] 2178번: 미로 탐색

ㅊhsilver — Sun, 9 Jan 2022 14:52:25 +0900

알고리즘 정리

문제

https://www.acmicpc.net/problem/2178

2178번: 미로 탐색

첫째 줄에 두 정수 N, M(2 ≤ N, M ≤ 100)이 주어진다. 다음 N개의 줄에는 M개의 정수로 미로가 주어진다. 각각의 수들은 붙어서 입력으로 주어진다.

www.acmicpc.net

✍ 풀이과정

최단 경로를 구하는 문제이므로 bfs를 통해 문제를 해결함

'1743번: 음식물 쓰레기' 문제를 먼저 풀었는데 다시 돌아와서 문제를 보니 비슷하게 풀면 되겠다는 생각이 들었다.

import sys
from collections import deque

n, m = map(int, input().split())

board=[]
visited =[[0]*m for _ in range(n)]
for _ in range(n):
    board.append(list(input()))

dx=[-1, 1 ,0 ,0]
dy=[0,0,-1,1]

# 시작점
queue = deque()
visited[0][0] = 1
queue.append((0, 0))

while queue:
    cur_x, cur_y = queue.popleft()
    if cur_x == n-1 and cur_y == m-1:
        print(visited[cur_x][cur_y])
        break

    for i in range(4):
        X = cur_x + dx[i]
        Y = cur_y + dy[i]

        if 0 <= X < n and 0 <= Y < m and visited[X][Y] == 0 and board[X][Y] == '1':
            queue.append((X, Y))
            visited[X][Y] = visited[cur_x][cur_y] + 1

1743번과 다르게 visited 리스트에 지나온 칸 수를 저장하였다.

Deque 초기화 Error

# 오류 1
queue = deque((0, 0))
visited[0][0] = 1

# 오류 2
queue = deque([0, 0])
## deque([0, 0])

# 올바른 코드 1
queue = deque()
visited[0][0] = 1
queue.append((0, 0))

# 올바른 코드 2
queue = deque([(0, 0)])
## deque([(0, 0)])

while queue:
    cur_x, cur_y = queue.popleft()
    if cur_x == n-1 and cur_y == m-1:
        print(visited[cur_x][cur_y])
        break

TypeError: 'int' object is not iterable

시작점을 deque에 넣는 코드를 (오류 1), (오류 2) 처럼 작성했더니 TypeError가 발생하였다.

deque 내에 tuple이나 list로 초기화를 하게 되면 deque([0, 0]) 처럼 하나의 리스트가 생성되므로

while 문 내에서 cur_x는 0을 할당받고, cur_y는 할당받는 것이 없어서 에러가 발생한 것이다.

popleft로 x, y를 얻고 싶다면,

1) append 사용

2) [( )] 사용

https://trey-de.tistory.com/14

Python에서 deque.popleft() 실행 시, iterable error가 뜬다면?

BFS문제들을 풀다보면, x와 y좌표를 큐에 넣고 빼는 경우가 많습니다. (예: www.acmicpc.net/problem/1926) 물론 상황에 맞게 적용하면 되겠지만, 1) 큐를 생성한 후에 리스트 또는 튜플 자료형으로 큐에

trey-de.tistory.com

[백준] 1743번: 음식물 피하기

ㅊhsilver — Sun, 9 Jan 2022 13:30:41 +0900

알고리즘 정리

문제

https://www.acmicpc.net/problem/1743

1743번: 음식물 피하기

첫째 줄에 통로의 세로 길이 N(1 ≤ N ≤ 100)과 가로 길이 M(1 ≤ M ≤ 100) 그리고 음식물 쓰레기의 개수 K(1 ≤ K ≤ N×M)이 주어진다. 그리고 다음 K개의 줄에 음식물이 떨어진 좌표 (r, c)가 주어진다

www.acmicpc.net

✍ 풀이과정

import sys
from collections import deque

n, m, k = map(int, sys.stdin.readline().split())

board =[[0]*m for _ in range(n)]

for _ in range(k):
    r, c = map(int, input().split())
    board[r-1][c-1] = 1

# 상하좌우
dx = [-1, 1,0,0]
dy = [0, 0,-1,1]

visited = [[0]*m for _ in range(n)]

def check(x,y):
    queue = deque()
    queue.append((x, y))

    visited[x][y] = 1
    cnt=1

    while queue:
        cur_x, cur_y = queue.popleft()
        for i in range(4):
            X = cur_x + dx[i]
            Y = cur_y + dy[i]

            if 0 <= X < n and 0 <= Y < m and visited[X][Y] == 0 and board[X][Y] == 1:
                queue.append((X, Y))
                visited[X][Y] = 1
                cnt += 1
    return cnt

results=[]
for i in range(n):
    for j in range(m):
        if board[i][j] ==1: # 쓰레기인 경우
            results.append(check(i,j))
print(max(results))

이 문제는 bfs, dfs로 모두 풀 수 있는 문제이다.

나는 bfs가 더 편해서 bfs로 풀었다.

- 쓰레기의 위치에 대한 정보를 담고 있는 board와 방문 여부를 알려주는 visited 리스트를 생성

- for 문을 돌면서 쓰레기가 위치한 지점에서부터 탐색을 시작하는 check 함수를 생성

* check 함수에서는 해당 지점의 상하좌우를 확인하여 쓰레기가 위치하며, 방문한 적 없는 곳을 queue에 추가한다.

[백준] 2293번: 동전 1

ㅊhsilver — Sun, 21 Nov 2021 13:06:22 +0900

문제

n가지 종류의 동전이 있다. 각각의 동전이 나타내는 가치는 다르다. 이 동전을 적당히 사용해서, 그 가치의 합이 k원이 되도록 하고 싶다. 그 경우의 수를 구하시오. 각각의 동전은 몇 개라도 사용할 수 있다.

사용한 동전의 구성이 같은데, 순서만 다른 것은 같은 경우이다.

입력

첫째 줄에 n, k가 주어진다. (1 ≤ n ≤ 100, 1 ≤ k ≤ 10,000) 다음 n개의 줄에는 각각의 동전의 가치가 주어진다. 동전의 가치는 100,000보다 작거나 같은 자연수이다.

출력

첫째 줄에 경우의 수를 출력한다. 경우의 수는 2³¹보다 작다.

✍ 풀이과정

처음에 떠올린 생각은 수능에서 풀었던 것처럼 가장 큰 수들을 먼저 넣고 경우의 수를 찾을라고 했었다.

그래서 재귀를 쓰거나 top-down 방식을 사용해서 값을 저장해둘라고 했는데

잘 모르겠어서 동영상을 참고해서 문제를 풀었다.

	dp[1]	dp[2]	dp[3]	dp[4]	dp[5]	dp[6]	dp[7]	dp[8]	dp[9]	dp[10]
1	1	1	1	1	1	1	1	1	1	1
2	0	1	2 (1+2)	2 (2+2, 1+1+2)	2 (2+2+1, 1+1+1+2)
5

행 - k원

열 - 동전의 가치

즉 dp(1, k)을 1원을 사용하여 k원을 만드는 방법이라고 할 수 있다.

dp(2,k)에서는 1,2원을 사용하여 k원을 만드는데 "2원을 추가할 때"와 "2원을 추가하지 않을 때"로 나눌 수 있다.

2원을 추가하지 않을 때는 dp[k]
2원을 추가할 때는 dp[k-2]

=> dp[k] = dp[k] + dp[k-2]

n, k=map(int, input().split())

v=[]
for _ in range(n):
    v.append(int(input()))

dp=[0]*(k+1)
dp[0]=1

for i in v:
    for j in range(i, k+1):
        dp[j] += dp[j-i]

print(dp[k])

[백준] 3085번: 사탕 게임

ㅊhsilver — Sun, 21 Nov 2021 11:37:11 +0900

문제

상근이는 어렸을 적에 "봄보니 (Bomboni)" 게임을 즐겨했다.

가장 처음에 N×N크기에 사탕을 채워 놓는다. 사탕의 색은 모두 같지 않을 수도 있다. 상근이는 사탕의 색이 다른 인접한 두 칸을 고른다. 그 다음 고른 칸에 들어있는 사탕을 서로 교환한다. 이제, 모두 같은 색으로 이루어져 있는 가장 긴 연속 부분(행 또는 열)을 고른 다음 그 사탕을 모두 먹는다.

사탕이 채워진 상태가 주어졌을 때, 상근이가 먹을 수 있는 사탕의 최대 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 보드의 크기 N이 주어진다. (3 ≤ N ≤ 50)

다음 N개 줄에는 보드에 채워져 있는 사탕의 색상이 주어진다. 빨간색은 C, 파란색은 P, 초록색은 Z, 노란색은 Y로 주어진다.

사탕의 색이 다른 인접한 두 칸이 존재하는 입력만 주어진다.

출력

첫째 줄에 상근이가 먹을 수 있는 사탕의 최대 개수를 출력한다.

✍ 풀이과정

어렵따...

우선 연속되는 사탕의 수를 구하는 함수를 작성했다.

def count(board, n):
    # 행
    for i in range(n):
        cnt=1
        for j in range(1, n):
            if board[i][j-1]==board[i][j]:
                cnt+=1
            else:
                cnt = 1
    # 열
    for i in range(n):
        cnt=1
        for j in range(1, n):
            if board[j-1][i]==board[j][i]:
                cnt+=1
            else:
                cnt = 1

n=int(input())

board =[list(input()) for _ in range(n)]

모든 경우의 서로 다른 색의 인접한 두 칸을 확인해봐야할지

최대가 발생할 수 있는 위치를 찾아 주변을 탐색해봐야할지...?

[백준] 1789번: 수들의 합

ㅊhsilver — Sun, 21 Nov 2021 11:31:14 +0900

문제

오랜만에 문제 짧아서 너무 신난당~!

✍ 풀이과정

n=int(input())
sum=0
tmp=1

while True:
    sum+=tmp
    if sum>n:
        print(tmp-1)
        break
    elif sum==n:
        print(tmp)
        break
    tmp+=1

서로 다른 자연수이기 때문에 1부터 더해주면서 값을 1씩 늘려준다.

S이 18일 때

1+2+3+4+5=15이고 마지막으로 3을 더해야지 18이 되는데 서로 다른 자연수이기 때문에 5을 더하지 않고 8을 더해야함

=> "1+2+3+4+8"

즉 합이 S를 넘을 때 그 갯수에서 1을 빼줘야하고 합이 S인 경우에는 그 갯수를 출력해주면 된다.

[백준] 16916번: 부분 문자열

ㅊhsilver — Sun, 14 Nov 2021 16:22:21 +0900

알고리즘 정리

문제

✍ 풀이과정

KMP 알고리즘을 사용해서 다들 푼 것 같다.

그냥 설명만 봤을 때는 한 번에 잘 이해안가고 너무 생소하다,,,ㅠ

차근차근 보면서 걍 따라했더니

def make_table():
    length = len(p)
    table = [0] * len(p)
    j = 0
    for i in range(1, length):
        while j > 0 and p[i] != p[j]:
            j = table[j - 1]
        if p[i] == p[j]:
            j += 1
            table[i] = j
    return table


def kmp():
    table = make_table()
    j = 0
    for i in range(len(s)):
        while j > 0 and s[i] != p[j]:
            j = table[j - 1]
        if s[i] == p[j]:
            if j == len(p) - 1:
                return True
            else:
                j += 1
    return False


s = input()
p = input()
if kmp():
    print(1)
else:
    print(0)

그냥 따라 적은 것이 되어버렸다....

그냥 풀라하면 다음 번에 풀 수 있을지ㅠㅠㅠ