글 읽기 - [python] 반례를 여쭙습니다..

tmfqksdk 3년 전 0

질문글에 나와있는 반례와 테스트 케이스까지 모두 적용해 보았는데, 도무지 틀린 반례가 나오지 않습니다 ㅠㅠㅠ

고수님들의 고견 듣고자 합니다.. 감사합니다

import sys

num_town, store_max = list(map(int, input().split(' ')))
loop = int(input())

# 1. 배송정보 마을 순으로 정렬
INPUT = []
for n in range(loop):
    INP = list(map(int, sys.stdin.readline().rstrip().split(' ')))
    INPUT.append(INP)

# print(INPUT)
# 2. 박스 배송
STORE = [0]*(num_town + 1)

answer = 0
for town in range(1, num_town + 1):
    # 1) 트럭에서 박스 내리기
    answer += STORE[town]
    STORE[town] = 0


    for n in range(loop):
        s_town, e_town, num_box = INPUT[n]

        # 2) 박스 싣기
        if s_town == town:
            store_reserve = store_max - sum(STORE)
            # print(store_max, sum(STORE), store_reserve)
            if store_reserve > num_box:
                STORE[e_town] += num_box

            else:
                STORE[e_town] += min(num_box, store_reserve)
                store_less = num_box - store_reserve # 남은 용량
                # print("store_less: ", store_less)
                town_tmp = e_town
                while store_less > 0:
                    town_tmp += 1
                    # print(town_tmp)
                    if town_tmp > num_town: break

                    if STORE[town_tmp] > 0:
                        # print('good', store_less, STORE)
                        if STORE[town_tmp] >= store_less:

                            STORE[town_tmp] -= store_less
                            STORE[e_town] += store_less
                            break
                        else:
                            STORE[e_town] += STORE[town_tmp]
                            store_less -= STORE[town_tmp]
                            STORE[town_tmp] = 0


    # print(STORE)


print(answer)

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

문제

문제

출처

ICPC

[python] 반례를 여쭙습니다..

tmfqksdk 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말