大致的思路是:通过检索每一个k的值在输入数据集中出现在了几列,出现在列的次数即为这个k值所对应的
输出。
import numpy as np
n,m,k = map( int, input().split(' ') )
flag = 0
count = 0
game = []
a = np.zeros([n,m], dtype = int)
for i in range(n) :
a[i] = input().split(' ')
for day in range(1,k+1):
for x in range(m):
for y in range(n):
if (a[y][x] == (day)):
flag = 1
break
if (flag == 1): count += 1
flag = 0
game.append(count)
count = 0
print(game)
测试数据正确,但是提交无法通过且超时了
请问大佬们程序出错在哪里了?
这个思路是可行的吗?如何优化?