삼성 기출에서 자주 나오는 & 숙지하면 좋을 코드 유형들을 추려보았습니다
혹시 자주 나오는 또다른 유형이 있다면 댓글로 알려주시면 감사하겠습니다 ! 🙇🏻♀️
arr = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]
## zip
# 시계 방향 90 (= 반시계 방향 270)
arr_90 = list(map(list, zip(*arr[::-1])))
print(arr_90)
# 시계 방향 180 (= 반시계 방향 180)
arr_180 = [a[::-1] for a in arr[::-1]]
print(arr_180)
# 시계 방향 270 (= 반시계 방향 90)
arr_270 = [x[::-1] for x in list(map(list, zip(*arr[::-1])))[::-1]]
print(arr_270)
## 인덱싱
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
n = 3
# 시계 방향 90 (= 반시계 방향 270)
new_90 = [[0] * n for _ in range(n)]
for i in range(n):
for j in range(n):
new_90[j][n - i - 1] = arr[i][j]
print(new_90)
# 시계 180 & 반시계 180
new_180 = [[0] * n for _ in range(n)]
for i in range(n):
for j in range(n):
new_180[n - i - 1][n - j - 1] = arr[i][j]
print(new_180)
# 시계 270 & 반시계 90
new_270 = [[0] * n for _ in range(n)]
for i in range(n):
for j in range(n):
new_270[n - 1 - j][i] = arr[i][j]
print(new_270)
def rotated_90(a):
m= len(a)
n = len(a[0])
result = [[0]* m for _ in range(n)] # 배열의 가로 세로 길이가 뒤바뀌는 것 주의
for i in range(m): # 범위 주의
for j in range(n): # 범위 주의
result[j][m-i-1] = a[i][j]
return result
def rotated_180(a):
n= len(a)
m = len(a[0])
result = [[0]* m for _ in range(n)]
for i in range(n): # 범위 주의
for j in range(m): # 범위 주의
result[n-i-1][m-j-1] = a[i][j]
return result
def rotated_270(a):
n= len(a)
m = len(a[0])
result = [[0]* n for _ in range(m)] # 배열의 가로 세로 길이가 뒤바뀌는 것 주의
for i in range(n): # 범위 주의
for j in range(m): # 범위 주의
result[m-1-j][i] = a[i][j]
return result
a=[[1,2,3,4],[5,6,7,8],[9,10,11,12]]
print(rotated_90(a))
print(rotated_180(a))
print(rotated_270(a))
부분 회전은 말그대로 2차원배열의 특정 부분만 회전시키는 것을 말한다. 보기엔 간단해 보일 수 있어도 꼭 step 별로 구현하는 것을 추천한다. (나중에 해당 부분에서 꼬이게 되면 디버깅 하기 힘들기 때문)
# 7X7 배열
arr = [[7 * j + i for i in range(1, 8)] for j in range(7)]
new_arr = [[0] * 7 for _ in range(7)]
sy, sx = 2, 2
length = 3
# 배열의 특정 부분(정사각형)을 회전시킴
def rotate_90(sy, sx, length):
global arr, new_arr
# 정사각형을 시계방향으로 90도 회전
for y in range(sy, sy + length):
for x in range(sx, sx + length):
# 1단계 : (0,0)으로 옮겨주는 변환을 진행함
oy, ox = y - sy, x - sx
# 2단계 : 90도 회전했을때의 좌표를 구함
ry, rx = ox, length - oy - 1
# 3단계 : 다시 (sy,sx)를 더해줌
new_arr[sy + ry][sx + rx] = arr[y][x]
# new_arr 값을 현재 board에 옮겨줌
for y in range(sy, sy + length):
for x in range(sx, sx + length):
arr[y][x] = new_arr[y][x]
print(arr[y][x])
rotate_90(sy, sx, length)
for i in range(len(arr)):
print(arr[i])
원본
회전 후
삼성 코테는 itertools 라이브러리 사용 불가하다. 백트래킹으로 직접 구현해서 써야한다.
arr = [1, 2, 3, 4]
visited = [0] * len(arr) # visited도 전역으로 둬도 됨
def permutations(n, new_arr):
global arr
# 순서 상관 0, 중복 X
if len(new_arr) == n:
print(new_arr)
return
for i in range(len(arr)):
if not visited[i]:
visited[i] = 1
permutations(n, new_arr + [arr[i]])
visited[i] = 0
permutations(2, [])
parameter
: 뽑을 개수, 뽑힌 값을 담는 배열arr = [1, 2, 3, 4]
def product(n, new_arr):
global arr
# 순서 상관 0, 중복 0
if len(new_arr) == n:
print(new_arr)
return
for i in range(len(arr)):
product(n, new_arr + [arr[i]])
product(2, [])
parameter
: 뽑을 개수, 뽑힌 값을 담는 배열arr = [1, 2, 3, 4]
# 현재 인덱스 +1 만큼을 매개변수로 계속 넘겨주어야함
# 순서가 상관없고 중복 불가이기 때문에 현재 인덱스보다 같거나 작은 인덱스는 볼 필요가 없기 때문
def combinations(n, new_arr, c):
# 순서 상관 X, 중복 X
if len(new_arr) == n:
print(new_arr)
return
for i in range(c, len(arr)):
combinations(n, new_arr + [arr[i]], i + 1)
combinations(2, [], 0)
현재 인덱스+1
값을 넘김parameter
: 뽑을 개수, 뽑힌 값을 담는 배열, 현재 인덱스arr = [1, 2, 3, 4]
def combinations_with_replacement(n, new_arr, c):
# 순서 상관 X, 중복
if len(new_arr) == n:
print(new_arr)
return
for i in range(c, len(arr)):
combinations(n, new_arr + [arr[i]], i)
combinations_with_replacement(2, [], 0)
현재 인덱스
값을 그대로 넘김parameter
: (뽑을 개수, 뽑힌 값을 담는 배열, 현재 인덱스)arr = [[0, 1, 0], [1, 0, 1], [0, 1, 0], [0, 0, 1], [0, 1, 0]]
print("기존")
for i in range(len(arr)):
print(arr[i])
def gravity():
n = len(arr)
m = len(arr[0])
for i in range(n - 1):
for j in range(m):
p = i
# 현재칸이 아래로 내려갈 수 있다면 그 윗줄도 한 칸 씩 연쇄적으로 내려와야함
while 0 <= p and arr[p][j] == 1 and arr[p + 1][j] == 0:
arr[p][j], arr[p + 1][j] = arr[p + 1][j], arr[p][j]
p -= 1
gravity()
print("변화")
for i in range(len(arr)):
print(arr[i])
2차원 배열에서 시작 지점을 기준으로 나선형으로 값에 접근하는 코드 유형이다
어딘가에선 토네이도 🌪 코드라고도 불리는 듯 하다
arr = [[0] * 5 for _ in range(5)]
def tornado():
global arr
d = [(0, -1), (1, 0), (0, 1), (-1, 0)]
y = len(arr) // 2
x = len(arr) // 2
num = 0 # 칸에 채워넣을 값
dist = 1
d_idx = 0 # 서 남 동 북 순서
move_count = 0 # 2가 되면 dist 길이가 1 늘어나고 move_count는 다시 0으로 초기화
while True:
for _ in range(dist):
dy, dx = d[d_idx]
Y = dy + y
X = dx + x
if (Y, X) == (0, -1): # 0행 -1열이 토네이도가 모두 끝나고 나서의 위치임
return
num += 1
arr[Y][X] = num
y = Y
x = X
move_count += 1
d_idx = (d_idx + 1) % 4
if move_count == 2:
dist += 1
move_count = 0
tornado()
for i in range(5):
print(arr[i])
def solution(n):
if n == 1:
return [[1]]
answer = [[0 for j in range(n)] for i in range(n)] # 배열 초기화
x = 0
y = 0
d_idx=0
for i in range(n * n):
answer[x][y] = i + 1
if d_idx == 0:
y += 1
if y == n - 1 or answer[x][y + 1] != 0: # 맨 끝에 도달했거나 가려는 곳에 이미 값이 있으면 방향 전환
d_idx = 1
elif d_idx == 1:
x += 1
if x == n - 1 or answer[x + 1][y] != 0:
d_idx = 2
elif d_idx == 2:
y -= 1
if y == 0 or answer[x][y - 1] != 0:
d_idx = 3
elif d_idx == 3:
x -= 1
if x == n - 1 or answer[x - 1][y] != 0:
d_idx = 0
return answer
arr=solution(5)
for i in range(len(arr)):
print(arr[i])
낚시왕 문제는 아래 방식대로 코드를 짜지 않으면 시간초과가 난다. 아래 코드의 경우 한 번에 코드를 이해하기 힘들 수 있는데 해당 영상이 이해하는 데 도움이 될 수 있으니 참고해 보면 좋을 듯 하다
def get_next_loc(i, j, speed, dir):
if dir == UP or dir == DOWN: # i
cycle = R * 2 - 2
if dir == UP:
speed += 2 * (R - 1) - i
else:
speed += i
speed %= cycle
if speed >= R:
return (2 * R - 2 - speed, j, UP)
return (speed, j, DOWN)
else: # j
cycle = C * 2 - 2
if dir == LEFT:
speed += 2 * (C - 1) - j
else:
speed += j
speed %= cycle
if speed >= C:
return (i, 2 * C - 2 - speed, LEFT)
return (i, speed, RIGHT)
백준의 배열돌리기1 문제 풀이이다 !
from collections import deque
N, M, R = map(int, input.split())
arr = []
new_arr = [[0]*M for _ in range(N)]
q = deque()
for i in range(N):
arr.append(list(input().split()))
loops = min(N, M) // 2
for i in range(loops):
q.clear()
q.extend(arr[i][i:M-i])
q.extend([row[M-i-1] for row in arr[i+1:N-i-1]])
q.extend(arr[N-i-1][i:M-i][::-1])
q.extend([row[i] for row in arr[i+1:N-i-1]][::-1])
q.rotate(-R)
for j in range(i, M-i): # 상
new_arr[i][j] = q.popleft()
for j in range(i+1, N-i-1): # 우
new_arr[j][M-i-1] = q.popleft()
for j in range(M-i-1, i-1, -1): # 하
new_arr[N-i-1][j] = q.popleft()
for j in range(N-i-2, i, -1): # 좌
new_arr[j][i] = q.popleft()
return new_arr
좋은 글 감사합니다 :)