글 읽기 - [python] 로봇 청소기 작동 질문

splato88 2년 전 0

문제 이해를 잘못했는지 질문 드립니다.

우선 테스트 케이스는 맞아요

게시판 반례 중 아래를 입력했는데 저는 7이 나오거든요

7 7
4 2 1
1 1 1 1 1 1 1
1 0 0 0 1 0 1
1 0 1 1 0 0 1
1 0 0 0 0 1 1
1 0 0 1 0 0 1
1 0 0 0 0 0 1
1 1 1 1 1 1 1
>> 11

그래서 그래프 상태를 출력해봤는데

[[1, 1, 1, 1, 1, 1, 1],

[1, 0, 0, 0, 1, 0, 1],

[1, 0, 1, 1, 0, 0, 1],

[1, 0, 0, 0, 0, 1, 1],

[1, 0, 2, 1, 2, 2, 1],

[1, 0, 2, 2, 2, 0, 1],

[1, 1, 1, 1, 1, 1, 1]]

이 상태에서 로봇 청소기가 4,5위치에 방향은 남쪽을 보고 있는 상태인데요 그래서 아래로 전진하여

[[1, 1, 1, 1, 1, 1, 1],

[1, 0, 0, 0, 1, 0, 1],

[1, 0, 1, 1, 0, 0, 1],

[1, 0, 0, 0, 0, 1, 1],

[1, 0, 2, 1, 2, 2, 1],

[1, 0, 2, 2, 2, 2, 1],

[1, 1, 1, 1, 1, 1, 1]]

구석까지 청소를 하였고(5,5) 더 이상 전진할 수 없어서 뒤로 후진했어요(4,5)

그리고 거기서 다시 모든 칸이 막혀있는데 뒤로 후진할 수 없으니까 작동을 멈춰야 하지 않은가요?

잘못된 부분을 알려주시면 감사하겠습니다!

#세로: n,y,j
#가로: m,x,i
n,m=map(int,input().split())
robot_y,robot_x,head=map(int,input().split())

graph=[]
for _ in range(n):
  graph.append(list(map(int,input().split())))


graph[robot_y][robot_x]=2

cnt=1

#한바퀴 돌았는지 표시
rotate=0

while True:
  #print('robot:',robot_y,robot_x,head)
  #print('rotate:',rotate)
  #print(graph)
  head+=1
  
  #4 이상이면 0으로 초기화
  if head>=4:
    head=0
  
  #방향에 따른 조건 설정
  if head==0:
    #전진
    robot_y-=1

    #0일때 청소
    if graph[robot_y][robot_x]==0:
      rotate=0
      graph[robot_y][robot_x]=2
      cnt+=1

    #벽이거나 이미 청소한 곳일때 제자리로 돌아오고 rotate 표시
    else:
      robot_y+=1
      rotate+=1
  elif head==1:
    robot_x-=1
    if graph[robot_y][robot_x]==0:
      rotate=0
      graph[robot_y][robot_x]=2
      cnt+=1
    else:
      robot_x+=1
      rotate+=1
  elif head==2:
    robot_y+=1
    if graph[robot_y][robot_x]==0:
      rotate=0
      graph[robot_y][robot_x]=2
      cnt+=1
    else:
      robot_y-=1
      rotate+=1
  elif head==3:
    robot_x+=1
    if graph[robot_y][robot_x]==0:
      rotate=0
      graph[robot_y][robot_x]=2
      cnt+=1
    else:
      robot_x-=1
      rotate+=1

  #한바퀴 돌면
  if rotate==4:
    if head==0:
      #뒤로 한칸
      robot_y+=1
      if graph[robot_y][robot_x]==1:
        #뒤가 벽이면 break
        break
      else:
        rotate=0
    elif head==1:
      robot_x+=1
      if graph[robot_y][robot_x]==1:
        break
      else:
        rotate=0
    elif head==2:
      robot_y-=1
      if graph[robot_y][robot_x]==1:
        break
      else:
        rotate=0
    elif head==3:
      robot_x-=1
      if graph[robot_y][robot_x]==1:
        break
      else:
        rotate=0
#print(graph)
print(cnt)
  

zigui 2년 전 0

틀린 부분이 크게 두 가지가 있는데, 하나는 로봇이 왼쪽이 아니라 오른쪽을 보고 있다는 것이고, 나머지 하나는 동쪽과 서쪽의 방향이 정확하지 않은 것입니다. (44, 50, 62, 68, 82, 94번째 줄)

댓글을 작성하려면 로그인해야 합니다.