[ Python ] 백준 10157 - 실패..

[ 코드 ]

#(r,c)가 편해서 백준과 반대로 받음
#변수는 내 맘대로받는거니깐
r,c=map(int,input().split())
idx=int(input())

# 답 찾는 함수
def find_ans(p):
    if(p>r*c): #out of range
        print(0)
    for i in range(r):
        for j in range(c):
            if(arr[i][j]==p):
                print(i+1,j+1)

# 배정 된 표 만들기
arr=[[0 for _ in range(c)]for _ in range(r)]
x,y=0,-1
num=1 #넣을 번호
add=1 #1 or -1
for i in range(c,0,-1):
    j=i+(r-c-1)
    for _ in range(i):
        y+= add
        arr[x][y] = num
        num+=1
    for _ in range(j):
        x+=add
        arr[x][y]=num
        num+=1
    add*=-1 #방향 바꿈


find_ans(idx)

 

 

직접 표를 만들어서 찾는 방식으로 코딩했다.

85%까지는 느리게 느리게 채점이 되더니 결국 틀렸다고 한다..ㅠ

다음에 좀 더 보완해야겠다!