判断题 python

7-5 判断题 (10 分)

判断题的评判很简单，本题就要求你写个简单的程序帮助老师判题并统计学生们判断题的得分。

输入格式：
输入在第一行给出两个不超过 100 的正整数 N 和 M，分别是学生人数和判断题数量。第二行给出 M 个不超过 5 的正整数，是每道题的满分值。第三行给出每道题对应的正确答案，0 代表“非”，1 代表“是”。随后 N 行，每行给出一个学生的解答。数字间均以空格分隔。

输出格式：
按照输入的顺序输出每个学生的得分，每个分数占一行。

输入样例：

3 6
2 1 3 3 4 5
0 0 1 0 1 1
0 1 1 0 0 1
1 0 1 0 1 0
1 1 0 0 1 1

输出样例：

13
11
12

代码如下：

n,m=input().split()
n,m=int(n),int(m)
lst=[int(i) for i in input().split()]
lst1=[int(i) for i in input().split()]
lst2=[]
for i in range(n):
    s=[]
    s=[int(i) for i in input().split()]
    lst2.append(s)
a=[]
for i in range(n):
    su=0
    for j in range(m):
        if lst2[i][j]==lst1[j]:
            su+=lst[j]
    a.append(su)
for i in range(n):
    print(a[i])