글 읽기 - 진짜 미칠것 같습니다. 테케,질문게시판 반례 다 통과하는데 어디가 문젠가요 ㅠㅠ

qkrwnstns52 3년 전 0

진짜 돌아버리겠습니다....... 도와주세요 ㅠㅠ

import copy


def attack(enemy):
    kill = []
    # cnt = 0
    for i in range(n + 1):
        for j in range(m):
            if board[i][j] == 5:
                enemy.sort(key=lambda x: (abs(i - x[0]) + abs(j - x[1]), x[1]))
                for k in range(len(enemy)):
                    if abs(i - enemy[k][0]) + abs(j - enemy[k][1]) <= d:
                        e_x, e_y = enemy[k][0], enemy[k][1]
                        # board[e_x][e_y] = 0
                        kill.append((e_x, e_y))
                        # enemy.pop(0)
                        # cnt += 1
                        break
    kill = list(set(kill))
    for i in range(len(kill)):
        board[kill[i][0]][kill[i][1]] = 0
    return len(kill)
    # return cnt


def move():
    save = []
    for i in range(n):
        for j in range(m):
            if board[i][j] == 1:
                nx = i + dx[1]
                ny = j + dy[1]
                board[i][j] = 0
                if -1 < nx < n and -1 < ny < m:
                    save.append((nx, ny))
    for i in range(len(save)):
        board[save[i][0]][save[i][1]] = 1


def check():
    flag = True
    for i in range(n):
        for j in range(m):
            if board[i][j] == 1:
                flag = False
    return flag


def arrange(x):
    global max_cnt
    if x == 3:
        cnt = 0
        while not check():
            enemy = []
            for i in range(n):
                for j in range(m):
                    if board[i][j] == 1:
                        enemy.append((i,j))
            cnt += attack(enemy)
            move()
        max_cnt = max(max_cnt, cnt)
        return
    for i in range(n):
        for j in range(m):
            board[i][j] = cboard[i][j]
    for s in range(m):
        if board[n][s] == 0:
            board[n][s] = 5
            arrange(x + 1)
            board[n][s] = 0


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

add = [0 for i in range(m)]
board.append(add)

cboard = copy.deepcopy(board)


dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

max_cnt = 0
arrange(0)
print(max_cnt)

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

문제

문제

출처

ICPC

진짜 미칠것 같습니다. 테케,질문게시판 반례 다 통과하는데 어디가 문젠가요 ㅠㅠ

qkrwnstns52 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말