请问大家,逃出农场那道题我用python3写的回溯法求解的,我在本地idle都通过了,但是牛客网系统说我的运行时间超了,可是系统给出的时间是处了c类语言外用时不超过4s,平时我在leetcode上写同类的回溯也就48ms之类的运行时间,无解?是对python不友好?代码如下:
import sys
def comb(n, k):
result = []
tmp = []
def helper(result, tmp, n, pos, k):
total = sum(tmp)
if k == 0:
if total % n == 0:
result.append(tmp[:])
return
else:
return
for i in range(pos, n):
tmp.append(i)
helper(result, tmp, n, i + 1, k - 1)
tmp.pop()
helper(result, tmp, n, 0, k)
return len(result)
for line in sys.stdin:
n, k = map(int, line.split())
print(comb(n, k))