코딩 테스트/삼성 기출

시뮬레이션과 구현) 20057.마법사 상어와 토네이도

Carnival7 2022. 10. 1. 13:49

20057번: 마법사 상어와 토네이도 (acmicpc.net)

 

20057번: 마법사 상어와 토네이도

마법사 상어가 토네이도를 배웠고, 오늘은 토네이도를 크기가 N×N인 격자로 나누어진 모래밭에서 연습하려고 한다. 위치 (r, c)는 격자의 r행 c열을 의미하고, A[r][c]는 (r, c)에 있는 모래의 양을

www.acmicpc.net


# 회전 : 가운데 칸부터 시작해서, 계속해서 방향을 90도 반시계 방향 회전하며 이동하다가, 회전해서 이동할 칸이 이미 처리한 칸이면 회전하지 않는 방식으로 탐색
# 모래 : 모래의 이동과 비율은 기준이 되는 방향을 하나 미리 구하고(좌) 그 기준을 계속해서 90도 반시계 방향 회전 시키면 4가지 방향(상,하,좌,우)에 대한 이동과 기준 구할 수 있다.
# 방향 : 방향 (dx,dy) -> (-dy,dx)로 바꾸면 90도 반시계 방향 회전

# 바람에 의해 모래가 옮겨지는 지점 및 비율
class Wind:
    def __init__(self,dx,dy,ratio):
        self.dx=dx
        self.dy=dy
        self.ratio=ratio

# 반시계방향
#  좌,하,우,상
dx=[0,1,0,-1]
dy=[-1,0,1,0]
spread=[]

# 반시계 방향으로 바람 경로 회전
def rotate(points):
    res=[]
    for p in points:
        nx=-p.dy
        ny=p.dx
        res.append(Wind(nx,ny,p.ratio))
    return res

# 상하좌우 방향별 모래 흩뿌리는 지점 리스트 저장
def make_spread():
    # 좌 방향
    points=[]
    points.append(Wind(1,-1,10))
    points.append(Wind(-1,-1,10))
    points.append(Wind(-1,0,7))
    points.append(Wind(1,0,7))
    points.append(Wind(-1,1,1))
    points.append(Wind(1,1,1))
    points.append(Wind(0,-2,5))
    points.append(Wind(-2,0,2))
    points.append(Wind(2,0,2))
    spread.append(points)
    # 하, 우, 상 방향
    for _ in range(3):
        points=rotate(points)
        spread.append(points)

# 격자 내부/외부 선별하여 내부이면 모래양 +, 외부이면 ans +
def inout(x,y,cur):
    global a,n
    if 0<=x<n and 0<=y<n:
        a[x][y]+=cur
        return 0
    else:
        return cur

make_spread()
n=int(input())
a=[list(map(int,input().split())) for _ in range(n)]
visit=[[False]*n for _ in range(n)] # 해당 칸이 처리되었는지 나타냄
order=[[0] * n for _ in range(n)] # 디버깅용
ans=0
x,y=n//2,n//2
visit[x][y]=True
cnt=2 # 디버깅용
order[x][y]=1
k=0 # 방향

while True:
    # 다음 이동할 칸 선별/방문
    nx,ny=x+dx[k],y+dy[k]
    order[nx][ny]=cnt
    cnt+=1
    visit[nx][ny]=True
    # (x,y) -> (nx,ny)
    sand=a[nx][ny]
    used=0
    a[nx][ny]=0
    # 칸 이동
    x,y=nx,ny

    # 해당 방향으로 모래 흩뿌리기
    for p in spread[k]:
        nx,ny=x+p.dx,y+p.dy
        cur=sand*p.ratio//100
        used+=cur
        ans+=inout(nx,ny,cur)
    sand-=used
    nx,ny=x+dx[k],y+dy[k]
    ans+=inout(nx,ny,sand)

    # (0,0) 칸이면 중지
    if (x,y)==(0,0):
        break

    # 다음 탐색할 방향 설정
    nk=(k+1)%4
    nx,ny=x+dx[nk],y+dy[nk]
    if not visit[nx][ny]:
        k=nk

print(ans)