http://acm.hdu.edu.cn/showproblem.php?pid=4125
题意:
给你N个数字,先按照给数的顺序建一棵二叉查找树,然后按中序遍历树的结点,并记录访问结点的奇偶顺序,这样就会得到一个序列,再给你一个0,1序列,现在要你求给出的序列在遍历出来的序列中出现的次数,序列允许相互覆盖。
思路:
暴力建树如果二叉树退化成一个树链的话,时间复杂度就会变成O(N^2),可能会超时。这样我们就要求找一种logn的建树策略,很容易就会想到线段树,对于每个元素x,我们找到在它前面的比x小的最大值y和比x大的最小值z,可以证明的是x要么插在y的右边,要么插在z的左边。 做一我们可以用线段树在O( n*logn )的时间内将树建好,然后就是dfs求遍历序列, 最后kmp就ok 了。
代码:
#include <stdio.h>
#include <string.h>
#include <stack>
using namespace std;
//#pragma comment(linker, "/STACK:1024000000,1024000000")
const int inf = (1 << 30) ;
const int NN = 600010 ;
int N ;
int lft[NN<<2] ,rgh[NN<<2] ;
void build(int l ,int r, int idx){
lft[idx] = rgh[idx] = -1 ;
if(l == r) return ;
int mid = (l + r) >> 1 , ls = idx<<1 , rs = idx<<1|1 ;
build(l , mid ,ls) ;
build(mid+1 , r , rs) ;
}
int left[NN] , right[NN] ;
char seq[NN * 3] , str[7010];
int cnt , len1 , len2 ;
int next[7010] ;
struct Node{
int now , f;
Node(){}
Node( int _now , int _f)
:now( _now ) , f( _f ){}
} ;
stack< Node > ss ;
void dfs(int now){
aaa :
seq[cnt++] = ( now&1 ) + '0' ;
int l = left[now] ;
if(l != -1 ){
ss.push( Node(now , 0) ) ;
now = l ;
goto aaa ;
bbb1 :
seq[ cnt++ ] = ( now&1 ) + '0' ;
}
int r = right[now] ;
if( r!=-1 ){
ss.push( Node(now , 1) ) ;
now = r ;
goto aaa ;
bbb2 :
seq[ cnt++ ] = ( now&1 ) + '0' ;
}
if( ss.empty() ) return ;
now = ss.top().now ;
int ff = ss.top().f ; ss.pop() ;
if( ff ) goto bbb2 ;
else goto bbb1 ;
}
void get_next(){
len1 = strlen( str ) ;
int i , j ;
next[0] = -1 ;
i = 1 ; j = -1 ;
for( ; i<len1 ;i ++ ){
while( j!=-1 && str[i]!=str[j+1] ) j = next[j] ;
if( str[i] == str[j+1] ) j++ ;
next[i] = j ;
}
}
int kmp(){
get_next() ;
len2 = strlen( seq ) ;
int i, j ; i = 0 ; j = -1 ;
int cc = 0 ;
for( ;i<len2 ;i++ ){
while( j!=-1 && str[j+1]!=seq[i] ) j = next[j] ;
if( seq[i] == str[j+1]) j++ ;
if( j==len1-1 ){
cc ++ ;
j = next[j] ;
}
}
return cc ;
}
inline int MIN(int a, int b){ return a < b ? a : b ;}
inline int MAX(int a, int b){ return a > b ? a : b ;}
void up(int l , int r, int idx){
int ls = idx<<1 ,rs = idx<<1|1 ;
if( lft[ls] == -1 ) lft[idx] = lft[rs] ;
else lft[idx] = lft[ls] ;
if( rgh[rs] == -1 ) rgh[idx] = rgh[ls] ;
else rgh[idx] = rgh[rs] ;
}
void update(int l , int r, int idx, int pos ){
if(l == r){
lft[idx] = rgh[idx] = pos ;return ;
}
int mid = (l + r) >> 1 , ls = idx<<1 ,rs = idx<<1|1 ;
if( pos<=mid ) update(l , mid , ls , pos) ;
else update(mid+1 , r , rs , pos) ;
up(l , r , idx);
}
int ask_small(int l , int r, int idx , int a, int b){
if( rgh[idx] == -1 ) return -1;
if(l==a && r==b) return rgh[idx] ;
int mid = (l + r) >> 1 , ls = idx<<1 , rs = idx<<1|1 ;
if( b<=mid ) return ask_small(l , mid , ls , a , b) ;
else if( mid<a ) return ask_small( mid+1 ,r ,rs , a ,b) ;
else{
int aa = ask_small( mid+1 , r , rs , mid+1 , b) ;
if( aa != -1 ) return aa ;
aa = ask_small( l , mid , ls , a , mid ) ;
return aa ;
}
}
int ask_big(int l , int r, int idx, int a, int b){
if( lft[idx] == -1 ) return -1 ;
if(l==a && r==b) return lft[idx] ;
int mid = (l + r) >> 1 , ls = idx<<1 , rs = idx<<1|1 ;
if( b<=mid ) return ask_big(l , mid , ls , a , b) ;
else if( mid<a ) return ask_big( mid+1 ,r ,rs , a ,b) ;
else{
int aa = ask_big( l , mid , ls , a , mid ) ;
if( aa != -1 ) return aa ;
aa = ask_big( mid+1 , r , rs , mid+1 , b ) ;
return aa ;
}
}
int main(){
int T , a , b; scanf("%d",&T) ;
int s_pos , m_pos , cas = 0 ;
while( T-- ){
scanf("%d",&N) ;
build(1 , N ,1);
for(int i=1;i<=N;i++) left[i] = right[i] = -1 ;
for(int i=1;i<=N;i++){
scanf("%d",&a);
if( i == 1 ) b = a ;
if( a == 1 ) s_pos = -1 ;
else s_pos = ask_small( 1 , N , 1 , 1 , a-1 ) ;
if( a == N ) m_pos = -1 ;
else m_pos = ask_big( 1 , N , 1 , a+1 , N ) ;
if( s_pos == -1 && m_pos == -1 ){
;
}
else if( s_pos==-1 && m_pos!=-1 ){
left[ m_pos ] = a ;
}
else if( s_pos!=-1 && m_pos==-1 ){
right[ s_pos ] = a ;
}
else{
if( right[ s_pos ] == -1 ) right[ s_pos ] = a ;
else left[ m_pos ] = a ;
}
update(1 , N , 1 , a ) ;
}
cnt = 0 ;
while( !ss.empty() ) ss.pop() ;
dfs( b ) ;
seq[cnt] = 0 ;
scanf("%s",str);
printf("Case #%d: %d\n",++cas,kmp() );
}
return 0 ;
}