Cod sursa(job #2505560)

Utilizator ReksioCroftOctavian Florin Staicu ReksioCroft Data 7 decembrie 2019 00:15:08
Problema Componente tare conexe Scor 30
Compilator py Status done
Runda Arhiva educationala Marime 1.14 kb
def dfs( nod ):
    vizitat[nod] = True
    for i in v[nod]:
        if vizitat[i] == False:
            dfs( i )
    stack.append(nod)


def ctc_dfs(nod):
    ctc[nod] = co
    for i in v2[nod]:
        if ctc[i] == 0:
            ctc_dfs( i )


try:
    import sys

    sys.setrecursionlimit(10000)
    fin = open( "ctc.in" )
    s = fin.readline().split()
    n = int( s[0] )
    v = [ [] for i in range (0, n) ]
    v2 = [ [] for i in range (0, n) ]
    for i in fin:
        s = [ int(x)-1 for x in i.split() ]
        v[ s[0] ].append( s[1] )
        v2[ s[1] ].append( s[0] )
    fin.close()

    vizitat = [False] * n
    stack = []
    for i in range( 0, n ):
        if vizitat[i] == False:
            dfs( i )

    co = 0
    ctc = [0] * n
    for i in range( n-1, -1, -1 ):
        if ctc[ stack[i] ] == 0:
            co += 1
            ctc_dfs( stack[i] )

    fout = open( "ctc.out", "w" )
    fout.write( str( co ) + '\n' )

    for i in range( 1, n + 1 ):
        for j in range( 0, n ):
            if i == ctc[j]:
                fout.write( str( j+1 ) + " ")
        fout.write( '\n' )
    fout.close()
except RecursionError as re:
    print( re )