仓库

题目描述:

某仓库记录每天出入库情况，正数表示入库数量，负数表示出库数量。编写程序，查询出入库总量等于k 的连续时间段个数。某星期出入库情况为“10,-10,5,6,7,8,-5”， k 值为 18 的连续时间段有两个，分别为第 1 天～第 5 天，第 3 天～第 5 天 。

输入格式:

第一行一行数字，表示出入库情况（不超过10000个数字）。

第二行一个数字k，表示出入库总量。

输出格式:

一行一个数字，表示出入库总量等于k 的连续时间段个数。

样例输入:

10 -10 5 6 7 8 -5
18

样例输出:

2

提示:

请完善以下代码：

a = list(map(int, input().split()))
k = int(input())
pos={0:[-1]};count =0
_①____________________
for i in range(len(a)):
    cur += a[i]
    _②____________________
    if pre in pos:
        _③____________________
    if cur in pos:
        pos[cur].append(i)
    else:
        pos[cur]=[i]
print(count)

时间限制: 1000ms
空间限制: 256MB

---