[코드트리 조별과제] 시뮬레이션 1차원 바람 Python

문제

 

바람이 불어오는 방향과 행의 숫자가 주어지고

만약 인접한 행의 같은 열에 같은 숫자가 있다면 반대 방향에서 바람을 불어오게 하면 된다.

(바람이 분다 = 한 칸씩 밀어낸다)

주어진 바람의 정보 수만큼 반복해서 진행한다.

 

 

 

풀이 방식

def rotate(row, direction):
    if direction == 'L':
        tmp = graph[row][-1]
        for i in range(m-1, 0, -1):
            graph[row][i] = graph[row][i-1]
        graph[row][0] = tmp
    elif direction == 'R':
        tmp = graph[row][0]
        for i in range(m-1):
            graph[row][i] = graph[row][i+1]
        graph[row][-1] = tmp

 

우선 한 칸씩 밀어 내는 함수를 만들었다.

왼쪽에서 불어 오면 오른쪽으로 한 칸씩 밀고, 오른쪽에서 불어 오면 왼쪽으로 한 칸씩 민다.

 

 

def infection(row, direction):
    if direction == 'U':
        if row - 1 >= 0:
            for i in range(m):
                if graph[row][i] == graph[row-1][i]:
                    return True
    elif direction == 'D':
        if row + 1 < n:
            for i in range(m):
                if graph[row][i] == graph[row+1][i]:
                    return True

    return False

 

다음 인접한 행에 같은 숫자가 있는 지 확인한 후 바람 전파를 진행할 것인지 확인한다.

만약 전파 가능하면 True를 반환한다.

 

 

n, m, q = map(int, input().split())
graph = [list(map(int, input().split())) for _ in range(n)]
for _ in range(q):
    r, d = input().split()
    r = int(r) - 1

    rotate(r, d)

    next_d = 'R' if d=='L' else 'L'
    direction = 'U'
    for i in range(r, 0, -1):
        if infection(i, direction):
            rotate(i-1, next_d)
            next_d = 'R' if next_d=='L' else 'L'
        else:
            break

    next_d = 'R' if d=='L' else 'L'
    direction = 'D'
    for i in range(r, n):
        if infection(i, direction):
            rotate(i+1, next_d)
            next_d = 'R' if next_d=='L' else 'L'
        else:
            break

for i in range(n):
    print(*graph[i])

 

바람이 시작한 행에서 위아래로 전파를 시작한다.

전파 가능해서 True를 반환받으면 rotate를 진행하고, 바람의 방향을 반대로 바꾼다.

바람 정보 수만큼 진행하고 최종 건물의 상태를 출력한다.

 

 

전체 코드 (Python3 / 217ms / 27MB)

def rotate(row, direction):
    if direction == 'L':
        tmp = graph[row][-1]
        for i in range(m-1, 0, -1):
            graph[row][i] = graph[row][i-1]
        graph[row][0] = tmp
    elif direction == 'R':
        tmp = graph[row][0]
        for i in range(m-1):
            graph[row][i] = graph[row][i+1]
        graph[row][-1] = tmp

def infection(row, direction):
    if direction == 'U':
        if row - 1 >= 0:
            for i in range(m):
                if graph[row][i] == graph[row-1][i]:
                    return True
    elif direction == 'D':
        if row + 1 < n:
            for i in range(m):
                if graph[row][i] == graph[row+1][i]:
                    return True

    return False


n, m, q = map(int, input().split())
graph = [list(map(int, input().split())) for _ in range(n)]
for _ in range(q):
    r, d = input().split()
    r = int(r) - 1

    rotate(r, d)

    next_d = 'R' if d=='L' else 'L'
    direction = 'U'
    for i in range(r, 0, -1):
        if infection(i, direction):
            rotate(i-1, next_d)
            next_d = 'R' if next_d=='L' else 'L'
        else:
            break

    next_d = 'R' if d=='L' else 'L'
    direction = 'D'
    for i in range(r, n):
        if infection(i, direction):
            rotate(i+1, next_d)
            next_d = 'R' if next_d=='L' else 'L'
        else:
            break

for i in range(n):
    print(*graph[i])

 

 

 

문제 링크

https://www.codetree.ai/missions/2/problems/The-1D-wind-blows/description

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석