Problem p02564 (python)

Fixed submission:
import io
import os


input = io.BytesIO(os.read(0, os.fstat(0).st_size)).readline
N, M = [int(x) for x in input().split()]
edges = [[int(x) for x in input().split()] for i in range(M)]

from collections import defaultdict

graph = defaultdict(list)
for u, v in edges:
    graph[u].append(v)
    graph[v]

# https://atcoder.jp/contests/practice2/submissions/16566401
def scc(G):
    preorder = {}
    lowlink = {}
    seen = [False] * N
    scc_queue = []
    i = 0  # Preorder counter
    ans = []
    for source in range(N):
        if seen[source]:
            continue
        queue = [source]
        while queue:
            v = queue[-1]
            if v not in preorder:
                i = i + 1
                preorder[v] = i
            done = True
            for w in G[v]:
                if w not in preorder:
                    queue.append(w)
                    done = False
                    break
            if done:
                lowlink[v] = preorder[v]
                for w in G[v]:
                    if seen[w]:
                        continue
                    if preorder[w] > preorder[v]:
                        lowlink[v] = min([lowlink[v], lowlink[w]])
                    else:
                        lowlink[v] = min([lowlink[v], preorder[w]])
                queue.pop()
                if lowlink[v] == preorder[v]:
                    scc = {v}
                    seen[v] = True
                    while scc_queue and preorder[scc_queue[-1]] > preorder[v]:
                        k = scc_queue.pop()
                        scc.add(k)
                        seen[k] = True
                    ans.append(scc)
                else:
                    scc_queue.append(v)
    return ans


SCC = list(scc(graph))

DEBUG = False
print(len(SCC))
if DEBUG:
    seen = set()
for comp in reversed(SCC):
    print(len(comp), *comp)
    if DEBUG:
        for u in comp:
            for v in graph[u]:
                assert v not in seen
        seen.update(comp)

Buggy submission (verdict: Runtime Error):
import networkx as nx

N, M = [int(x) for x in input().split()]
edges = [[int(x) for x in input().split()] for i in range(M)]

graph = nx.DiGraph(edges)

if True:
    import matplotlib.pyplot as plt

    nx.draw_networkx(graph)
    plt.show()

SCC = list(
    nx.algorithms.components.strongly_connected_components(graph)
)
print(len(SCC))
for comp in reversed(SCC):
    print(len(comp), *comp)

Buggy submission (verdict: Runtime Error):
import networkx as nx

N, M = [int(x) for x in input().split()]
edges = [[int(x) for x in input().split()] for i in range(M)]

graph = nx.DiGraph(edges)

if True:
    import matplotlib.pyplot as plt

    nx.draw_networkx(graph)
    plt.show()

SCC = list(
    nx.algorithms.components.strongly_connected_components(graph)
)
print(len(SCC))
for comp in reversed(SCC):
    print(len(comp), *comp)

Buggy submission (verdict: Runtime Error):
import networkx

N, M = [int(x) for x in input().split()]
edges = [[int(x) for x in input().split()] for i in range(M)]

graph = networkx.DiGraph(edges)
SCC = list(networkx.algorithms.components.strongly_connected_components_recursive(graph))
print(len(SCC))
for comp in SCC:
    print(len(comp), *comp)