코딩테스트 연습문제
👇문제👇
문제 설명
지나다니는 길을 'O', 장애물을 'X'로 나타낸 직사각형 격자 모양의 공원에서 로봇 강아지가 산책을 하려합니다. 산책은 로봇 강아지에 미리 입력된 명령에 따라 진행하며, 명령은 다음과 같은 형식으로 주어집니다.
["방향 거리", "방향 거리" … ]
예를 들어 "E 5"는 로봇 강아지가 현재 위치에서 동쪽으로 5칸 이동했다는 의미입니다. 로봇 강아지는 명령을 수행하기 전에 다음 두 가지를 먼저 확인합니다.
주어진 방향으로 이동할 때 공원을 벗어나는지 확인합니다.
주어진 방향으로 이동 중 장애물을 만나는지 확인합니다.
위 두 가지중 어느 하나라도 해당된다면, 로봇 강아지는 해당 명령을 무시하고 다음 명령을 수행합니다.
공원의 가로 길이가 W, 세로 길이가 H라고 할 때, 공원의 좌측 상단의 좌표는 (0, 0), 우측 하단의 좌표는 (H - 1, W - 1) 입니다.
공원을 나타내는 문자열 배열 park, 로봇 강아지가 수행할 명령이 담긴 문자열 배열 routes가 매개변수로 주어질 때, 로봇 강아지가 모든 명령을 수행 후 놓인 위치를 [세로 방향 좌표, 가로 방향 좌표] 순으로 배열에 담아 return 하도록 solution 함수를 완성해주세요.
제한사항
3 ≤ park의 길이 ≤ 50
3 ≤ park[i]의 길이 ≤ 50
park[i]는 다음 문자들로 이루어져 있으며 시작지점은 하나만 주어집니다.
S : 시작 지점
O : 이동 가능한 통로
X : 장애물
park는 직사각형 모양입니다.
1 ≤ routes의 길이 ≤ 50
routes의 각 원소는 로봇 강아지가 수행할 명령어를 나타냅니다.
로봇 강아지는 routes의 첫 번째 원소부터 순서대로 명령을 수행합니다.
routes의 원소는 "op n"과 같은 구조로 이루어져 있으며, op는 이동할 방향, n은 이동할 칸의 수를 의미합니다.
op는 다음 네 가지중 하나로 이루어져 있습니다.
N : 북쪽으로 주어진 칸만큼 이동합니다.
S : 남쪽으로 주어진 칸만큼 이동합니다.
W : 서쪽으로 주어진 칸만큼 이동합니다.
E : 동쪽으로 주어진 칸만큼 이동합니다.
1 ≤ n ≤ 9
입출력 예
| park | routes | result |
| ["SOO","OOO","OOO"] | ["E 2","S 2","W 1"] | [2,1] |
| ["SOO","OXX","OOO"] | ["E 2","S 2","W 1"] | [0,1] |
| ["OSO","OOO","OXO","OOO"] | ["E 2","S 3","W 1"] | [0,0] |
입출력 예 #1
입력된 명령대로 동쪽으로 2칸, 남쪽으로 2칸, 서쪽으로 1칸 이동하면 [0,0] -> [0,2] -> [2,2] -> [2,1]이 됩니다.
입출력 예 #2
입력된 명령대로라면 동쪽으로 2칸, 남쪽으로 2칸, 서쪽으로 1칸 이동해야하지만 남쪽으로 2칸 이동할 때 장애물이 있는 칸을 지나기 때문에 해당 명령을 제외한 명령들만 따릅니다. 결과적으로는 [0,0] -> [0,2] -> [0,1]이 됩니다.
입출력 예 #3
처음 입력된 명령은 공원을 나가게 되고 두 번째로 입력된 명령 또한 장애물을 지나가게 되므로 두 입력은 제외한 세 번째 명령만 따르므로 결과는 다음과 같습니다. [0,1] -> [0,0]
문제를 이해해야 뭘 풀든 말든 하는데,,,
이해하는데 한참 걸림,,,
아니 좀 더 명확하고 쉽게 표현할 방법이 없는 건 아닌 것 같은데
일부러 그러는 건가?
알듯말듯 헷갈리는 문제라서 차근차근 풀어보기로 함
# 시작 좌표 찾기
for i in range(len(park)):
if 'S' in park[i]:
start = [i, park[i].find('S')]
breakroutes가 ["E 2","S 2","W 1"] 형태이기 때문에 방향과 거리를 가져와서 쓰기가 편하다.
dir = routes[j][0] # 방향
dis = int(routes[j][2]) # 거리그리고 방향별로 나눠서 움직이는 함수를 만들어주면 끝!
인데 하루종일 걸림
def solution(park, routes):
answer = []
start = []
for i in range(len(park)):
if 'S' in park[i]:
start = [i, park[i].find('S')]
break
# 시작좌표 start
for j in range(len(routes)):
dir = routes[j][0]
dis = int(routes[j][2])
elif dir == 'W':
loc_move = start[1] - dis
if loc_move < 0: # 길없음
continue
if 'X' in park[start[0]][loc_move:start[1]]: # X 위치 찾기
continue
else:
start[1] = loc_move
elif dir == 'E':
loc_move = start[1] + dis
if loc_move >= len(park[0]): # 길없음
continue
if 'X' in park[start[0]][start[1]:loc_move+1]: # X 위치 찾기
continue
else:
start[1] = loc_move
return startpark의 input이 ["OSO","OOO","OXO","OOO"] 로 주어지기 때문에,
가로줄의 속성을 파악하려면 이 리스트 안에서 문자열을 하나 뽑아 슬라이싱 해서 비교하면 된다.
만약 내가 있는 곳이 맨 윗줄, 즉 park[0][]이라면 "OSO"문자열 안에 X가 있는지 찾으면 되는 것이다.
그래서 동쪽(E), 서쪽(W) 일 때는 슬라이싱을 이용해 쉽게 했는데, 문제는 북(N), 남(S) 이었다.
if dir == 'N':
loc_move = start[0] - dis
if loc_move < 0: # 길없음
print(start, '길없음')
continue
if 'X' in park[loc_move:start[0]][start[1]]: # 장애물이 있을 때
print(start, '장애물')
continue
else:
start[0] = loc_move # 없을 때
print(start, '이동')
elif dir == 'S':
loc_move = start[0] + dis
if loc_move >= len(park):
print(start, '길없음')
continue
if 'X' in park[start[0]:loc_move+1][start[1]]: # 장애물이 있을 때
print(start, '장애물')
continue
else:
start[0] = loc_move # 없을 때
print(start, '이동')☝️동, 서쪽과 같은 방식으로 이번엔 0번째 값을 슬라이싱 해서 가져왔는데,
자꾸 장애물을 뚫고 가버리는 것이어따...
그래서 for문을 사용해 각 문자열의 해당 자리를 먼저 가져오고, 그 뒤에 X값이 있는지 비교해보려고 했다.
👇그랬더니 역시나 이번에도 X값을 지나서 뚫고 가버린다...👇
if dir == 'N':
loc_move = start[0] - dis
if loc_move < 0: # 길없음
print(start, '길없음')
continue
for i in range(loc_move,start[0]):
if 'X' in park[i][start[1]]: # 장애물이 있을 때
print(start, '장애물')
continue
else:
start[0] = loc_move # 없을 때
print(start, '이동')
elif dir == 'S':
loc_move = start[0] + dis
if loc_move >= len(park):
print(start, '길없음')
continue
for i in range(start[0],loc_move + 1):
if 'X' in park[i][start[1]]: # 장애물이 있을 때
print(start, '장애물')
continue
else:
start[0] = loc_move # 없을 때
print(start, '이동')그래서 결국 2차원 배열의 슬라이싱에 대해 검색해보고, 그 결과를 바탕으로 새로운 코드를 짰다.
사실 이게 왜 되는지, 뭔지는 나도 모르겠당 ㅎ
if dir == 'N':
loc_move = start[0] - dis
if loc_move < 0: # 길없음
continue
if 'X' in [park[i][start[1]] for i in range(loc_move,start[0])] : #장애물이 있을 때
continue
else:
start[0] = loc_move # 없을 때
elif dir == 'S':
loc_move = start[0] + dis
if loc_move >= len(park):
continue
if 'X' in [park[i][start[1]] for i in range(start[0],loc_move+1)]: #장애물이 있을 때
continue
else:
start[0] = loc_move # 없을 때👇전체코드👇
def solution(park, routes):
answer = []
start = []
for i in range(len(park)):
if 'S' in park[i]:
start = [i, park[i].find('S')]
break
# 시작좌표 start
for j in range(len(routes)):
dir = routes[j][0]
dis = int(routes[j][2])
if dir == 'N':
loc_move = start[0] - dis
if loc_move < 0: # 길없음
continue
if 'X' in [park[i][start[1]] for i in range(loc_move,start[0])] : #장애물이 있을 때
continue
else:
start[0] = loc_move # 없을 때
elif dir == 'S':
loc_move = start[0] + dis
if loc_move >= len(park):
continue
if 'X' in [park[i][start[1]] for i in range(start[0],loc_move+1)]: #장애물이 있을 때
continue
else:
start[0] = loc_move # 없을 때
elif dir == 'W':
loc_move = start[1] - dis
if loc_move < 0: # 길없음
continue
if 'X' in park[start[0]][loc_move:start[1]]:
continue
else:
start[1] = loc_move
elif dir == 'E':
loc_move = start[1] + dis
if loc_move >= len(park[0]): # 길없음
continue
if 'X' in park[start[0]][start[1]:loc_move+1]:
continue
else:
start[1] = loc_move
return start
GitHub 댓글