[BOJ][Python] 백준 2174번 - 로봇 시뮬레이션

문제 링크: https://www.acmicpc.net/problem/2174

2174번: 로봇 시뮬레이션

 

 

---

문제 풀이

전형적인 시뮬레이션 문제. 이름부터 시뮬레이션이라 나와있다.

일반적인 문제였으면 쉽게 풀렸을거 같지만 평소와 다른 x, y 좌표여서 다르게 처리해줘야 했다. 로봇을 불러오는 경우는 로봇 리스트를 만들어서 $N$번 로봇의 좌표 및 방향은 $N$번 인덱스에 저장시켜서 바로 불러올 수 있었다.

맵을 만들때 빈 공간인 경우는 0으로, 로봇이 있는 경우는 i번 로봇이라면 i로 해줬다.

 

코드

	a, b = map(int, input().split())
	maps = [[0 for _ in range(a)] for _ in range(b)]
	n, m = map(int, input().split())
	robot = [0]
	direct = {'N': 0, 'E': 1, 'S': 2, 'W': 3}
	dx = [-1, 0, 1, 0]
	dy = [0, 1, 0, -1]
	for i in range(1, n+1):
	x, y, d = input().split()
	x, y = int(x)-1, int(y)
	maps[b-y][x] = i
	robot.append((b-y, x, direct[d]))
	orders = []
	for _ in range(m):
	rb, o, repeat = input().split()
	orders.append((int(rb), o, int(repeat)))
	for order in orders:
	rb, o, repeat = order
	if o == 'F':
	for _ in range(repeat):
	x, y, d = robot[rb]
	nx = x + dx[d]
	ny = y + dy[d]
	if nx < 0 or nx >= b or ny < 0 or ny >= a:
	print('Robot '+str(rb)+' crashes into the wall')
	exit()
	if maps[nx][ny] != 0:
	print('Robot '+str(rb)+' crashes into robot '+str(maps[nx][ny]))
	exit()
	maps[x][y], maps[nx][ny] = 0, rb
	robot[rb] = (nx, ny, d)
	elif o == 'L':
	x, y, d = robot[rb]
	for _ in range(repeat):
	d = (d+3)%4
	robot[rb] = (x, y, d)
	elif o == 'R':
	x, y, d = robot[rb]
	for _ in range(repeat):
	d = (d+1)%4
	robot[rb] = (x, y, d)
	print('OK')

view raw 2174.py hosted with ❤ by GitHub