글 읽기 - [파이썬] 테케는 다 통과 했지만 바로 틀리네요.. 도저히 모르겠습니다..

wksehfdl8 3년 전 1

도와주시면 감사하겠습니다..ㅠㅠㅠㅠㅠ 몇시간 동안 고민해보고 다른사람 풀이도 봐봤지만,

제가 틀린 이유를 잘 모르겠네요..

코드는 비교적 간단하고 주석도 달아 놨습니다...

한번만 훑어봐주시면 정말 감사하겠습니다!!

import sys

input = sys.stdin.readline
N,K = map(int,input().split())
board = [list(map(int,input().split())) for _ in range(N)]  #
chessPieces = []
pos=[[[] for _ in range(N)] for _ in range(N)]
for i in range(K):
    r,c,d = map(int,input().split())
    chessPieces.append([r-1,c-1,d-1])
    pos[r-1][c-1].append(i)

dy,dx= [0,0,-1,1], [1,-1,0,0] #우,좌,상,하

def move(piece_num): # white:0   red:1   blue:2
    y,x,dir = chessPieces[piece_num]
    ny,nx = y+dy[dir], x+dx[dir]

    if ny<0 or nx<0 or ny>=N or nx>=N or board[ny][nx] == 2:        # 파랑,벽 체크   
        if dir == 0 or dir ==2:               
            dir +=1
        else:
            dir -=1
        chessPieces[piece_num][2] = dir
        ny,nx = y+dy[dir], x+dx[dir]

    if ny < 0 or nx < 0 or ny >= N or nx >= N or board[ny][nx] == 2: # 다시 한번 체
        return 0

    stack,tmp=[],-1                                 # 위에있는거 가져고기 pop이라 이미 reversed 되있
    while tmp != piece_num:
        tmp = pos[y][x].pop()
        stack.append(tmp)

    if board[ny][nx] == 1:                          # 빨, reversed 되있으므로 just extend
        pos[ny][nx].extend(stack)

    elif board[ny][nx] == 0:                        # 하 reverse 후 extend
        stack.reverse()
        pos[ny][nx].extend(stack)

    for i in stack:                                 # 체스피스 위치 업뎃
        chessPieces[i][0], chessPieces[i][1] = ny, nx

    if len(pos[ny][nx]) >= 4:                       # 스택이 4 이상 쌓였으면 리턴 1
        return 1
    return 0

cnt=1
while cnt <= 1000:
    for i in range(K):
        if move(i) == 1:
            print(cnt)
            sys.exit()
    cnt+=1

print(cnt)

wksehfdl8 3년 전 0

57 줄

print(-1) 입니다... ㅁㅊㅇ.. ㅠ

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

문제

문제

출처

ICPC

[파이썬] 테케는 다 통과 했지만 바로 틀리네요.. 도저히 모르겠습니다..

wksehfdl8 3년 전 1

wksehfdl8 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말