1529: [POI2005]ska Piggy banks（SCC）

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=1529

很DT的一道题，被卡了好几次空间，最后改了手写栈另外重复利用了一个数组才过。。。。

思路很简单，缩一下SCC，然后统计一下有几个入度为0的点就可以了。。

代码：

#include 

#include 

#include 

 

using namespace std ;

 

#define MAXN 1000100

 

struct edge {

    edge *next ;

    int t ;

} *head[ MAXN ] ;

 

void AddEdge( int s , int t ) {

    edge *p = new( edge ) ;

    p -> t = t , p -> next = head[ s ] ;

    head[ s ] = p ;

}

 

int n , a[ MAXN ] ;

 

int dfn[ MAXN ] , low[ MAXN ] , scc[ MAXN ] , cnt = 0 , S[ MAXN ] , top = 0 ;

bool f[ MAXN ] ;

 

void Tarjan( int v ) {

    dfn[ v ] = low[ v ] = ++ cnt ;

    S[ ++ top ] = v , f[ v ] = true ;

    for ( edge *p = head[ v ] ; p ; p = p -> next ) if ( ! dfn[ p -> t ] ) {

        Tarjan( p -> t ) ; low[ v ] = min( low[ v ] , low[ p -> t ] ) ;

    } else if ( f[ p -> t ] ) low[ v ] = min( low[ v ] , low[ p -> t ] ) ;

    if ( dfn[ v ] == low[ v ] ) {

        int u ;

        do {

            f[ u = S[ top -- ] ] = false ; scc[ u ] = v ;

        } while ( u != v ) ;

    }

}

 

int main(  ) {

    memset( head , 0 , sizeof( head ) ) ;

    scanf( "%d" , &n ) ;

    for ( int i = 0 ; i ++ < n ; ) {

        scanf( "%d" , &a[ i ] ) ; AddEdge( a[ i ] , i ) ;

    }

    memset( dfn , 0 , sizeof( dfn ) ) ;

    memset( f , false , sizeof( f ) ) ;

    for ( int i = 0 ; i ++ < n ; ) if ( ! dfn[ i ] ) {

        Tarjan( i ) ;

    }

    memset( dfn , 0 , sizeof( dfn ) ) ;

    for ( int i = 0 ; i ++ < n ; ) if ( scc[ a[ i ] ] != scc[ i ] ) {

        dfn[ scc[ i ] ] ++ ;

    }

    int ans = 0 ;

    for ( int i = 0 ; i ++ < n ; ) if ( scc[ i ] == i && ! dfn[ i ] ) ++ ans ;

    printf( "%d\n" , ans ) ;

    return 0 ;

}