글 읽기 - 테케 다맞고 제출하면 바로 틀림..

nk1318 1년 전 0

테케 다맞고 제출하면 바로 틀렸습니다 뜨네요 어디서 틀린걸까요.. 미치겠네여

def IsBlock(r, c, d, dd):
    if d == 1:
        if dd == 0 and (walls[r][c] == 0 or walls[r - 1][c] == 1):
            return True
        elif dd == 1 and walls[r][c] == 1:
            return True
        elif dd == 2 and (walls[r + 1][c] == 0 or walls[r + 1][c] == 1):
            return True
    elif d == 2:
        if dd == 0 and (walls[r][c] == 0 or walls[r - 1][c - 1] == 1):
            return True
        elif dd == 1 and walls[r][c - 1] == 1:
            return True
        elif dd == 2 and (walls[r + 1][c] == 0 or walls[r + 1][c - 1] == 1):
            return True
    elif d == 3:
        if dd == 0 and (walls[r][c - 1] == 0 or walls[r][c - 1] == 1):
            return True
        elif dd == 1 and walls[r][c] == 0:
            return True
        elif dd == 2 and (walls[r][c] == 1 or walls[r][c + 1] == 0):
            return True
    elif d == 4:
        if dd == 0 and (walls[r][c - 1] == 1 or walls[r + 1][c - 1] == 0):
            return True
        elif dd == 1 and walls[r + 1][c] == 0:
            return True
        elif dd == 2 and (walls[r][c] == 1 or walls[r + 1][c + 1] == 0):
            return True
    return False

def hot():
    result = [[0] * C for _ in range(R)]
    for r in range(R):
        for c in range(C):
            if 1 <= rooms[r][c] <= 4:
                temp_rooms = [[0] * C for _ in range(R)]
                queue = []
                d = rooms[r][c]

                # 온도 5 설정
                nr = r + dx[d][1]
                nc = c + dy[d][1]
                if 0 <= nr < R and 0 <= nc < C and not IsBlock(r, c, d, 1):
                    temp_rooms[nr][nc] = 5
                    queue.append([nr, nc])
                else:
                    break

                # 온도 1 ~ 4 설정
                i = 4
                new_queue = []
                while queue and i > 0:
                    qr, qc = queue.pop()
                    for j in range(3):
                        tr = qr + dx[d][j]
                        tc = qc + dy[d][j]
                        if 0 <= tr < R and 0 <= tc < C and not IsBlock(qr, qc, d, j):
                            temp_rooms[tr][tc] = i
                            if [tr, tc] not in new_queue:
                                new_queue.append([tr, tc])
                    if not queue:
                        queue = new_queue
                        i -= 1
                        new_queue = []

                for r_ in range(R):
                    for c_ in range(C):
                        result[r_][c_] += temp_rooms[r_][c_]
    return result

def survey():
    for r in range(R):
        for c in range(C):
            if rooms[r][c] == 5 and after_rooms[r][c] < K:
                return False
    return True

R, C, K = map(int, input().split())
rooms = [list(map(int, input().split())) for _ in range(R)]
W = int(input())
walls_list = [list(map(int, input().split())) for _ in range(W)]
after_rooms = [[0] * C for _ in range(R)]
walls = [[-1] * C for _ in range(R)]
dx = ["", [-1, 0, 1], [-1 ,0, 1], [-1, -1, -1], [1, 1, 1]]
dy = ["", [1, 1, 1], [-1, -1, -1], [-1, 0, 1], [-1, 0, 1]]
chocolate = 0

for r in range(R):
    for c in range(C):
        for w in walls_list:
            if w[0] - 1 == r and w[1] - 1 == c:
                walls[r][c] = w[2]

add_hot = hot()

while True:

    # 1. 모든 온풍기에서 바람 나옴
    for r_ in range(R):
        for c_ in range(C):
            after_rooms[r_][c_] += add_hot[r_][c_]

    # 2. 온도 조절
    temp_rooms = [[0] * C for _ in range(R)]
    for r in range(R):
        for c in range(C):
            temp_rooms[r][c] += after_rooms[r][c]
            for d in range(1, 5):
                nr = r + dx[d][1]
                nc = c + dy[d][1]
                if r <= nr < R and c <= nc < C and not IsBlock(r, c, d, 1):
                    diff = after_rooms[r][c] - after_rooms[nr][nc]
                    if diff >= 4:
                        temp_rooms[r][c] -= diff // 4
                        temp_rooms[nr][nc] += diff // 4
                    elif diff <= -4:
                        temp_rooms[r][c] += abs(diff) // 4
                        temp_rooms[nr][nc] -= abs(diff) // 4
    after_rooms = temp_rooms

    # 3. 1 이상의 바깥쪽 칸 온도 1 감소
    for r in range(R):
        for c in range(C):
            if (r == 0 or r == R - 1 or c == 0 or c == C - 1) and after_rooms[r][c]:
                after_rooms[r][c] -= 1

    # 4. 초콜릿 하나 먹음
    chocolate += 1
    if chocolate >= 101:
        break

    # 5. 조사하는 칸의 온도 검사
    if survey():
        break

print(chocolate)

tk4003950 1년 전 0

60번째 줄, 방향이 다른 바람인데도 위치가 같다는 걸로 new_queue에 안들어가고 있네요

8 10
0 0 0 0 0 0 0
4 4 4 4 4 4 4
0 0 0 0 5 0 0
0 0 0 0 0 5 0
3 3 3 3 3 3 3
3
4 1
4 0
6 0

답: 1
출력: 2

nk1318 1년 전 0

말씀해주신 부분이 어디가 틀린지 아무리 찾아도 모르겠네요ㅠ

대신, 이 조건으로

온풍기가 있는 칸과 바람이 나오는 방향에 있는 칸 사이에는 벽이 없다.
온풍기의 바람이 나오는 방향에 있는 칸은 항상 존재한다.

아래 부분을

# 온도 5 설정
nr = r + dx[d][1]
nc = c + dy[d][1]
if 0 <= nr < R and 0 <= nc < C and not IsBlock(r, c, d, 1):
temp_rooms[nr][nc] = 5
queue.append([nr, nc])
else:
break

이렇게 고치니까

# 온도 5 설정
nr = r + dx[d][1]
nc = c + dy[d][1]
temp_rooms[nr][nc] = 5
queue.append([nr, nc])

알려주신 반례가 제대로 1이 출력되네요..

그래도 제출하면 틀렸다고 나와요 ㅎ 하하

아무튼 감사합니다,,ㅠㅠ

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

문제

문제

출처

ICPC

테케 다맞고 제출하면 바로 틀림..

nk1318 1년 전 0

tk4003950 1년 전 0

nk1318 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말