python - 不断收到 Kattis 问题的运行时错误

Question

我正在尝试解决这个问题( https://open.kattis.com/problems/anotherbrick )。

当我提交最终代码时，我不断收到运行时错误。我不确定我的代码有什么问题。

# @Author Gansaikhan Shur
# Another Brick in the Wall 

import sys

# if len(sys.argv) < 2:
#    sys.exit("{}: needs an input file!".format(sys.argv[0]))
#    exit()

sys_input = sys.argv[1]

with open(sys_input, "r") as f:
    data = [line.rstrip('\n') for line in f]
    hw_num = data[0].split()
    len_bricks = data[1].split()
    # Assigning to variables
    height = int(hw_num[0])
    width = int(hw_num[1])
    num_bricks = hw_num[2]

currHeight = 0
for i in range(height):
    currHeight = i+1
    brick_width = 0
    for j in len_bricks:
        brick_width += int(j)
        if currHeight == height and brick_width == width:
            print("YES")
            exit()
        if brick_width == width:
            brick_width = 0
            continue
        elif brick_width > width:
            print("NO")
            exit()
        else:
            continue
    currHeight += 1

请告诉我如何才能让 kattis.com 接受此代码。谢谢

Answer 1

运行时错误可能意味着:

• 您遇到仅在运行时发生的错误
• 您得到了错误的结果(请参阅下面的代码块 - 您的数据收集是错误的)
• 您的代码运行时间太长

<小时/>

问题 1:

您的代码不会增加高度:

 if brick_width == width:
            brick_width = 0
            continue

如果你有像bricks = [1]*10000000这样的数据，你的代码将会运行得很慢，因为你永远不会增加你所在的高度。您需要处理整个列表，直到得到“否”。

问题2:

您为此任务读取数据的方法是错误的 - 它不是基于文件。请参阅the documentation for python on Kattis -您需要从 sys.stdin 读取(代码如下)。

<小时/>

您可以简化它:

def can_he_do_it(h,w,bricks):
    height = 0
    cur_w = 0

    # remove this line for kattis.com
    print("Project: Height: {} Width: {} with {}".format(h,w,bricks)) 

    # process all bricks - not the height or anythin else with range
    # bricks is all you got, place them down one by one, check lenght/height
    # and return False if you are too wide or not high enough
    # return True if you are high enough
    for brick in bricks:
        cur_w += brick        # add to current width and check

        if cur_w > w:         # too wide
            return False
        elif cur_w == w:      # exaclty correct, next row
            height += 1
            cur_w = 0

        if height == h:       # reach target heigth
            return True

    return False              # too few materials


def result(b):
     print("YES" if b else "NO") 

# replace with your number-read-code
result( can_he_do_it(2, 10, [5,5,5,5,5,5,]) )
result( can_he_do_it(2, 10, [5,5,5,3,5,5,]) )
result( can_he_do_it(2, 10, [5,5,5,]) ) 

输出:

Project: Height: 2 Width: 10 with [5, 5, 5, 5, 5, 5]
YES

Project: Height: 2 Width: 10 with [5, 5, 5, 3, 5, 5]
NO

Project: Height: 2 Width: 10 with [5, 5, 5]
NO

<小时/>

用途:

import sys 

data = []
for i in sys.stdin:
    data.append(i)

data = [line.rstrip('\n') for line in data if line]
height, width, *_ = map(int,data[0].split())  # read 2 values, ignore rest, cast to int
bricks = list(map(int, data[1].split()))      # use all bricks, cast to int

# replace with your number-read-code
can_he_do_it(height, width, bricks)

获取: