Question

我正在尝试解决此问题http://www.spoj.pl/problems/PEBBMOV/。我认为我有正确的算法，这不是这个问题的重点。这个问题有一个奇怪的输入文件。一个测试用例的输入应该是表单的输入 n a1 a2 a3 ... an。（所有注册）

这里的问题是，a [i]之间存在杂散的换行符等。我需要能够跳过这些新行并在一个地方收集属于一个测试用例的所有[i]。我怎么知道这一切？还有一系列WAs和运行时错误，以及论坛上的一些研究。我有以下python代码尝试这样做，除了我似乎在关键的地方摇摇欲坠，只是不能完成它。我希望在输入读数结束时在列表行[]中有适当的输入行。

有人可以告诉我我的错误吗？或者建议一个更好的方法？提前谢谢..

import sys
#data = sys.stdin.readlines()
#lines = inp.split('\n')
data = sys.stdin.read()
pos = 0
lno = 0

lines = []
while pos<len(data):
    while not data[pos].isdigit():
                   pos = pos + 1
num =data[pos]
print num
cur = pos + 1
numbers_collected = 0

x = [] # temp list
y = []
while numbers_collected < num:

    if cur<len(data) and data[cur].isdigit():
        y.append(data[cur])
        cur = cur + 1
        numbers_collected += 1
    else:
        if cur<len(data)and numbers_collected < num:
            cur = cur + 1
        else:
            break
print x
pos = cur
x.extend(y)
lines.extend(x)



for line in lines:
    print line

Answer 1

这有助于您回答问题吗？

In [1]: s1 = "1\n2\n\n3\n\n\n4\n\n\n\n5\n\n\n\n\n6"

In [2]: s1
Out[2]: '1\n2\n\n3\n\n\n4\n\n\n\n5\n\n\n\n\n6'

In [3]: s1.splitlines()
Out[3]: ['1', '2', '', '3', '', '', '4', '', '', '', '5', '', '', '', '', '6']

In [4]: [elem for elem in s1.splitlines() if elem]
Out[4]: ['1', '2', '3', '4', '5', '6']

如果没有看到示例输入，则很难回答这个问题。但是，SPOJ problem page不提供示例输入，因此OP无法提供不可用的内容。

Answer 2

这样的事情可以提供帮助：

numbers = map(int, sys.stdin.read().split())
#numbers = [1,2,3,4,5]

这是一个快速而肮脏的解决方案，因为它将所有空格视为分隔符。

Answer 3

spoj问题说每个游戏只有一个输入行：

for line in sys.stdin:
    n, *piles = map(int, line.split())
    assert len(piles) == n

让我们假设同一游戏中的数字之间可能有换行符：

numbers = (int(s) for line in sys.stdin for s in line.split() if line.strip())
for n in numbers:
    piles = list(itertools.islice(numbers, n))
    assert len(piles) == n

将输入拆分为python中的行

3 个答案: