Merge remote-tracking branch 'origin/main'

Author: zaqquum

.DS_Store

@@ -0,0 +1,34 @@
+from collections import deque
+import sys
+input = sys.stdin.readline
+
+dr = [1, -1, 0, 0]  # down, up, right, left
+dc = [0, 0, 1, -1]
+
+N, K = map(int, input().split())
+board = []  # 바이러스 정보 받아올 배열
+for _ in range(N):
+    board.append(list(map(int, input().split())))
+
+S, X, Y = map(int, input().split())  # 시간, 행, 열
+
+q = []
+for i in range(N):
+    for j in range(N):
+        if board[i][j] != 0:  # 바이러스가 있는 위치
+            q.append((board[i][j], i, j, 0))  # 바이러스 정보, 행, 열, 흘러간 시간
+q.sort()  # (주의) 리스트로 받아 정렬 후 queue로 변경!
+q = deque(q)
+
+while q:  # BFS
+    viru, row, col, time = q.popleft()
+    if time == S:  # 시간이 S만큼 지났다면 종료
+        break
+    for d in range(4):  # 상하좌우 이동
+        r = row + dr[d]
+        c = col + dc[d]
+        if -1 < r < N and -1 < c < N and not board[r][c]:  # 범위를 벗어나지 않고 위치 값이 0일 때
+            board[r][c] = viru  # 바이러스 배치
+            q.append((viru, r, c, time+1))  # 다음 번 위치를 큐에 저장
+
+print(board[X-1][Y-1])

.idea/.gitignore

@@ -0,0 +1,81 @@
+import sys
+from collections import deque
+
+#1. 입력값 받기
+#2. 모든 배열을 돌며 인구이동 조건을 만족하는 나라가 있는지 확인. (1. 인접한 나라 2. 두 국가의 인구 차의 절댓값이 l이상 r이하)
+#3. 인구 이동조건을 만족하는 나라가 있으면 bfs로 그래프를 탐색하면서 나라의 개수와 인구의 총합을 구해서 새로운 인구를 구해준다.
+#4. visited 배열은 인구 이동날마다 초기화 해야 한다.
+
+# 실제로 인구를 이동시키는 함수
+def movePopulation(i, j):
+    global n, l, r
+    global visited, arr
+
+    dx = [-1, 1, 0, 0]
+    dy = [0, 0, -1, 1]
+
+    queue = deque([(i, j)])
+    visited[i][j] = 1
+    queue2 = deque([(i, j)])
+    populationSum = arr[i][j]
+    countrySum = 1
+
+    while queue:
+        x, y = queue.popleft()
+
+        for d in range(4): 
+            row = x + dx[d]
+            col = y + dy[d]
+
+            if 0 <= row < n and 0 <= col < n and visited[row][col] == 0 and l <= abs(arr[x][y] - arr[row][col]) <= r:
+                visited[row][col] = 1
+                countrySum += 1
+                populationSum += arr[row][col]
+                queue.append((row, col))
+                queue2.append((row, col))
+
+    newPopulation = populationSum // countrySum
+
+    while queue2:
+        x, y = queue2.popleft()
+        arr[x][y] = newPopulation
+
+# 인구 이동 조건을 만족하는 나라가 있는지 확인하는 함수
+def checkCountries():
+    global ans
+
+    dx = [-1, 1, 0, 0]
+    dy = [0, 0, -1, 1]
+
+    for i in range(n):
+        for j in range(n):
+            for k in range(4):
+                row = i + dx[k]
+                col = j + dy[k]
+                if 0 <= row < n and 0 <= col < n and l <= abs(arr[i][j] - arr[row][col]) <= r:
+                    ans += 1
+                    return True
+
+    return False
+
+
+inp = sys.stdin.readline
+
+n, l, r = map(int, inp().split())
+
+arr = []
+visited = [[0] * n for _ in range(n)]
+
+ans = 0
+
+for i in range(n):
+    arr.append(list(map(int, inp().split())))
+
+while checkCountries():
+    for i in range(n):
+        for j in range(n):
+            if visited[i][j] == 0:
+                movePopulation(i, j)
+    visited = [[0] * n for _ in range(n)]
+
+print(ans)

.idea/codingtest_algorithm_study.iml

@@ -0,0 +1,21 @@
+import sys
+
+inp = sys.stdin.readline
+
+arr = inp().split()
+
+n = arr[0]
+b = int(arr[1])
+
+ans = 0
+i = 1
+length = len(n)
+
+for char in n:
+    if char >= 'A' and char<='Z':
+        ans = ans + pow(b,length - i) * (ord(char) - ord('A') + 10)
+    else:
+        ans = ans + pow(b,length - i) * int(char)
+    i+=1
+
+print(ans)