14503.로봇 청소기 - python

https://www.acmicpc.net/problem/14503

14503번: 로봇 청소기

 

구현으로 가능한 문제 

단순하게 문제의 조건대로 뒤로가는 기능 ,앞으로 가능기능 , 90도 회전하는 기능 함수를 만들어서 현재 시작점부터 시작해서 끝나는 조건이 나올때 까지 단계적으로 함수를 실행한다 

 

r,l=map(int,input().split())
x,y,d=map(int,input().split())
t=[]
for i in range(r):
    t.append(list(map(int,input().split())))
def back(x,y,d):
    if d==0:x+=1
    elif d==1:y-=1
    elif d==2:x-=1
    else:y+=1
    if 0<=x <r and 0<=y<l and t[x][y]!=1:
        return x,y
    return False

def turn(d):
    return d-1 if d!=0 else 3

def go(x,y,d):
    if d==0:x-=1
    elif d==1:y+=1
    elif d==2:x+=1
    else:y-=1
    if 0<=x <r and 0<=y<l and t[x][y]==0:
        return x,y
    return False

w=0
while True:
    an=0
    if t[x][y]!=2:w+=1
    t[x][y]=2
    
    for i,j in [[-1,0],[1,0],[0,-1],[0,1]]:
        if 0<=x+i<r and  0<=y+j<l and t[x+i][y+j]==0:
            an=1
    if an:
        while True:
            d=turn(d)
            if go(x,y,d):
                
                x,y=go(x,y,d)
                break
    else:
        if back(x,y,d):
            
            x,y=back(x,y,d)
        else:
            break
print(w)