글 읽기 - [Python] 바로 틀렸습니다. 1000년 예제 통과 못함..

kdk124578 2년 전 5

문제의 예제는 다 통과하고

5 2 7
2 3 2 3 2
2 3 2 3 2
2 3 2 3 2
2 3 2 3 2
2 3 2 3 2
2 1 3
3 2 3

답 : 71

이 예제도 통과하는데

천 년 예제

10 1 1000
100 100 100 100 100 100 100 100 100 100
100 100 100 100 100 100 100 100 100 100
100 100 100 100 100 100 100 100 100 100
100 100 100 100 100 100 100 100 100 100
100 100 100 100 100 100 100 100 100 100
100 100 100 100 100 100 100 100 100 100
100 100 100 100 100 100 100 100 100 100
100 100 100 100 100 100 100 100 100 100
100 100 100 100 100 100 100 100 100 100
100 100 100 100 100 100 100 100 100 100
1 1 1

답 : 5443

이게 자꾸 안 나오네요.. 하

이거 뭐 1000개라서 디버깅도 못 하겠고..ㅋㅋㅋ

제 머가리로는 틀린 부분을 찾을 수 없습니다..

시간 초과로 고민 좀 해보고 싶네요.

N, M, K = map(int, input().split())
S2D2 = [list(map(int, input().split())) for _ in range(N)]
energy = [[5] * N for _ in range(N)]

tree_info = [[[] for _ in range(N)] for _ in range(N)]
for _ in range(M):
    x, y, age = map(int, input().split())
    tree_info[x - 1][y - 1].append(age)

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

for k in range(K):
    # 양분 먹기 및 죽기
    for i in range(N):
        for j in range(N):
            tree_info[i][j].sort()
            for idx, tree_age in enumerate(tree_info[i][j]):
                if tree_age <= energy[i][j]:
                    tree_info[i][j][idx] += 1
                    energy[i][j] -= tree_age
                else:  # 현재 인덱스부터 나무는 죽어버림.
                    energy[i][j] += sum(tree_info[i][j][idx:]) // 2
                    tree_info[i][j] = tree_info[i][j][:idx]
                    break
                    
    # 번식하기
    for i in range(N):
        for j in range(N):
            for tree_age in tree_info[i][j]:
                if tree_age % 5 == 0:
                    for d in range(8):
                        nx, ny = i + dx[d], j + dy[d]
                        if not (0 <= nx < N and 0 <= ny < N):
                            continue
                        tree_info[nx][ny].append(1)

    # 양분 추가하기
    for i in range(N):
        for j in range(N):
            energy[i][j] += S2D2[i][j]

ans = 0
for row in tree_info:
    for elem in row:
        ans += len(elem)

print(ans)

kdk124578 2년 전 1

26번째 줄에 죽은 나무의 양분을 더할 때 한꺼번에 더하면 안 되네요!

가만히 생각해보니까 오차가 납니다.

energy[i][j] += sum(tree_info[i][j][idx:]) // 2
    
# 한꺼번에 더하면 안 되네요..ㅋㅋㅋㅋ

for die in tree_info[i][j][idx:]:
    energy[i][j] += die // 2

gadoz 1년 전 1

올리신지 오래 되었지만....

스스로 답변 달아주신 내용 감사합니다. 제꺼 코드 틀린 부분이 이것과 같았는데 큰 도움 되었습니다.

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

문제

문제

출처

ICPC

[Python] 바로 틀렸습니다. 1000년 예제 통과 못함..

kdk124578 2년 전 5

kdk124578 2년 전 1

gadoz 1년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말