为什么第二实施比第一实施快?

时间:2019-05-29 05:40:06

标签: python graph disjoint-sets

我最近尝试针对一种代码问题提交我的解决方案。我成功解决了这个问题,并尝试了各种测试用例,并且代码在本地计算机上正常运行。问题中的时间限制为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。

先谢谢您。

0 个答案:

没有答案