글 읽기 - 반례가 뭐가 있을까요..? 4시간 넘게 고민중입니다

tyg04308 2년 전 0

안녕하세요!

구글링해서 다른 코드와도 비교해봐도 비슷하게 한 것 같은데 틀렸다고 합니다.

어느 부분에서 틀렸을까요..?

이거 때문에 4시간넘게 붙잡고 있는데 잘 안되네요.. ㅜㅜ

예제는 다 통과했고 추가로 개인적으로 넣어봤는데도 이해가 잘 안갑니다..

감사합니다

from itertools import combinations
import sys
from heapq import heappop, heappush

row, col, max_d = map(int, sys.stdin.readline().split(' '))

matrix = list()

position = [x for x in range(row)]
position = list(combinations(position, 3))

for y in range(row):
    matrix.append(list(map(int, sys.stdin.readline().split(" "))))

def exist_enemy(t_map):
    global row

    for i in range(row):
        if 1 in t_map[i]:
            return True

    return False


def move_forward(t_map):
    global row, col

    for i in range(row-1, 0, -1):
        t_map[i] = t_map[i-1]

    t_map[0] = [0] * col


def attack_enemy(t_map, archers):
    global row, col, max_d

    attacked = set()

    for archer in archers:
        all_distance = list()

        for i in range(row-1, -1, -1):
            for j in range(col):
                if t_map[i][j] == 1:
                    dist = abs(row-i) + abs(archer-j)
                    heappush(all_distance, (dist, j, i))

                else:
                    heappush(all_distance, (11, j, i))
            
            min_d, min_y, min_x = heappop(all_distance)

            if (min_d <= max_d):
                attacked.add((min_x, min_y))
                break

    for attack in attacked:
        x, y = attack
        t_map[x][y] = 0

    return len(attacked)


if __name__ == "__main__":
    result = 0

    for archers in position:
        tmp_map = [line[:] for line in matrix]
    
        cnt = 0
        while exist_enemy(tmp_map):
            cnt += attack_enemy(tmp_map, archers)
            move_forward(tmp_map)

        if result < cnt:
            result = cnt

    print(result)

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

문제

문제

출처

ICPC

반례가 뭐가 있을까요..? 4시간 넘게 고민중입니다

tyg04308 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말