中铁笔试题 火车空闲座位问题

题目如下：

 

 

代码如下：

import sys
line1 = sys.stdin.readline().strip()
W,M = list(map(int, line1.split()))
arr=[]
N=0
for k in range(W):
    N+=k
arr=[]
for i in range(N):
    arr.append(list(map(int,(sys.stdin.readline().strip().split()))))
#print(arr)
print(N)
for i in range(1,W):
    j=i+1
    H=0
    for m in range(len(arr)):
        if arr[m][0] <= i and arr[m][1] >= j:
            H += arr[m][2]
    print(M - H,end=" ")
        #for n in range(len(arr[m])):


结果：