글 읽기 - 청소년상어 파이썬 코드 질문

hjy_stat 1년 전 0

몇 번을 다시 구현해도 계속 틀리는데

어느 부분에서 잘못된 건지 잘 모르겠습니다.

고수 분들 도와주세요..

import copy

dirs = [(-1, 0), (-1, -1), (0, -1), (1, -1), (1, 0), (1, 1), (0, 1), (-1, 1)] # 8방 정의
map_ = [[False] * 4 for _ in range(4)]

# 최대로 먹은 물고기 번호의 합
max_eaten_no_sum = 0

class Fish:
    def __init__(self, type_: str, no: int, dir_: int, row: int, col: int):
        self.type = type_ # shark or fish
        self.no = no # 물고기 번호
        self.dir = dir_# 방향
        self.row = row
        self.col = col
    @classmethod
    def from_fish(self, o):
        self.type = o.type # shark or fish
        self.no = o.no # 물고기 번호
        self.dir = o.dir # 방향
        self.row = o.row
        self.col = o.col

def isIn(r: int, c: int) -> bool:
    if r >= 0 and r <= 3 and c >=0 and c<= 3:
        return True
    else:
        return False

def move_fish(fish: Fish, map_: list[Fish]) -> list[Fish]:
    dir_ = fish.dir
    for d in range(dir_, dir_ + len(dirs)):
        nd = d % 8 # 0부터 7까지 값을 갖도록
        fish.dir = nd # 물고기 방향 재설정
        nr = fish.row + dirs[nd][0]
        nc = fish.col + dirs[nd][1]
        if isIn(nr, nc):
            # 그 공간이 null이면 이동 가능
            if map_[nr][nc] is None:
                map_[fish.row][fish.col] = None #기존자리 Null 처리
                fish.row = nr
                fish.col = nc
                map_[nr][nc] = fish
                return map_
            # 그 공간이 물고기면 자리 바꾸기
            elif map_[nr][nc].type == 'F':
                dest = map_[nr][nc]
                dest.row = fish.row
                dest.col = fish.col
                
                fish.row = nr
                fish.col = nc
                map_[nr][nc] = fish
                map_[dest.row][dest.col] = dest
                return map_
    

def move_fishes(map_: list[Fish]):
    for n in range(1, 17): # 물고기 번호가 낮은것부터 이동(1~16번)
        flag = False
        for r in range(len(map_)):
            for c in range(len(map_[r])):
                if (map_[r][c] is not None) and map_[r][c] == n:
                    flag = True
                    map_ = move_fish(copy.deepcopy(map_[r][c]), copy.deepcopy(map_))
                    break # column에 대한 break
            if flag:
                break # row에 대한 break
    return map_
        
def move(map_ : list[Fish], shark: Fish, eaten: int):
    global max_eaten_no_sum
    max_eaten_no_sum = max(max_eaten_no_sum, eaten)
    # 일단 물고기 이동시키기
    map_ = move_fishes(copy.deepcopy(map_))
    
    # 상어는 1부터 3까지 갈 수 있다.
    for i in range(1,4):
        nr = shark.row + dirs[shark.dir][0] * i
        nc = shark.col + dirs[shark.dir][1] * i
        
        # 새로운 좌표가 범위안에 있고, 물고기가 있다면
        if isIn(nr, nc) and (map_[nr][nc] is not None):
            no = map_[nr][nc].no
            ns = Fish('S', 0, map_[nr][nc].dir, nr, nc)
            # 맵 복제 후 상어심기
            cloned = copy.deepcopy(map_)
            cloned[nr][nc] = ns
            # 기존 위치는 null 처리
            cloned[shark.row][shark.col] = None
            move(copy.deepcopy(cloned), copy.deepcopy(ns), no + eaten)          
            
# 초기 물고기 배치
for row in range(4):
    input_ = list(map(int, input().split()))
    no, dir_ = input_[0:8:2], input_[1:8:2]
    dir_ = [d-1 for d in dir_]
    for col in range(4):
        map_[row][col] = Fish('F', no[col], dir_[col], row, col)
        
# 초기 상어        
shark = Fish('S', 0, map_[0][0].dir, 0, 0)
max_eaten_no_sum = map_[0][0].no

map_[0][0] = shark

move(copy.deepcopy(map_), copy.deepcopy(shark), max_eaten_no_sum)

print(max_eaten_no_sum)

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

문제

문제

출처

ICPC

청소년상어 파이썬 코드 질문

hjy_stat 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말