我最近尝试针对一种代码问题提交我的解决方案。我成功解决了这个问题,并尝试了各种测试用例,并且代码在本地计算机上正常运行。问题中的时间限制为2秒。我试图提交解决方案,但是它说超过了期限。
代码在这里(我的实现)
def find(u):
if parent[u] == u:
return u
parent[u] = find(parent[u])
return parent[u]
n,m = map(int,input().split())
parent = [i for i in range(n+1)]
ans = [0 for i in range(n+1)]
for i in range(1,m+1):
grp = [int(i) for i in input().split()][1:]
if len(grp) == 0:
continue
x = find(grp[0])
for i in grp[1:]:
parent[find(i)] = x
for i in range(1,n+1):
ans[find(i)] += 1
ans = " ".join(map(str,[ans[find(u)] for u in range(1, n + 1)]))
print(ans)
然后,我对代码进行了一些更改,并成功在Codeforce上提交了该代码。
第二实施:
from sys import stdin
from sys import stdout
def find(u):
if parent[u] == u:
return u
parent[u] = find(parent[u])
return parent[u]
n, m = map(int, stdin.readline().split())
parent = [i for i in range(n+1)]
ans = [0 for i in range(1+n)]
for i in range(1,m+1):
grp= [int(u) for u in stdin.readline().split()][1:]
if(len(grp)==0):
continue
x = find(grp[0])
for i in grp[1:]:
parent[find(i)] = x
for i in range(1,n+1):
ans[find(i)]+=1
pprintit = ' '.join(map(str, [ans[find(u)] for u in range(1, n + 1)]))
stdout.write('%s' % pprintit)
我的问题是:
stdin比input()更快,因为仅通过更改I / O方法,我就可以提交解决方案
我正在使用python 3.7。
先谢谢您。