题目:http://www.lydsy.com/JudgeOnline/problem.php?id=1529
很DT的一道题,被卡了好几次空间,最后改了手写栈另外重复利用了一个数组才过。。。。
思路很简单,缩一下SCC,然后统计一下有几个入度为0的点就可以了。。
代码:
#include <cstdio>
#include <algorithm>
#include <cstring>
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 ;
}