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)

获得：

持续遇到Kattis问题的运行时错误

1 个答案: