把测试点下载下来本地运行是对的，可一提交就WA，怎么办？

代码如下：

target = input()
mains = input()
s = ' ' + mains + '@' + target
m_size = len(mains)
pi_map = [0 for _ in s]

for i in range(2, len(s)):
	len = i - 1
	while len:
		if s[pi_map[len] + 1] == s[i]:
			pi_map[i] = pi_map[len] + 1
			if pi_map[i] == m_size:
				print(i - 2 * m_size)
			break
		len = pi_map[len]
		
for i in range(1, m_size + 1):
	print(pi_map[i],end=' ')
print('')