HDU_4125 Moles 线段树

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 ;
}

HDU_4125 Moles 线段树相关推荐

HDU 4125 Moles 线段树+KMP
题意: 给定n, 下面是1-n的排列. 下面一个二进制子串. 先按给定的排列建出二叉树. 然后遍历树(根->左子树->根->右子树->根) 遍历这个节点时若权值为奇数入栈一个 ...
[线段树or笛卡尔树+简单KMP]poj4005 or hdu4125 Moles
题意:N只编号1-N的鼹鼠打洞,第i只编号为a[i],编号不重复.打的洞的样子符合以a[i]为值,以下标为插入顺序的二叉搜索树.现在从根出发,存在左子树则先走左子树,否则往右走,每经过一个洞(结点), ...
二逼平衡树——树套树（线段树套Splay平衡树）
题面 Bzoj3196 解析线段树和Splay两棵树套在一起,常数直逼inf,但最终侥幸过了思路还是比较简单, 在原数组维护一个下标线段树,再在每一个线段树节点,维护一个对应区间的权值Splay. ...
线段树——HDU - 1698
题目含义就是初始化一堆数为1 可以经过操作把一个区间的数都改变并求这堆数的总大小题目分析有一个 #include<iostream> #include<stdio.h> ...
BZOJ.1558.[JSOI2009]等差数列(线段树差分)
BZOJ 洛谷首先可以把原序列\(A_i\)转化成差分序列\(B_i\)去做. 这样对于区间加一个等差数列\((l,r,a_0,d)\),就可以转化为\(B_{l-1}\)+=\(a_0\),\(B ...
【线段树分治线性基】luoguP3733 [HAOI2017]八纵八横
不知道为什么bzoj没有HAOI2017 题目描述 Anihc国有n个城市,这n个城市从1~n编号,1号城市为首都.城市间初始时有m条高速公路,每条高速公路都有一个非负整数的经济影响因子,每条高速公路 ...
[bzoj1582][Usaco2009 Hol]Holiday Painting 节日画画_线段树
Holiday Painting 节日画画 bzoj-1582 Usaco-2009 Hol 题目大意:给定两个n*m的01网格图.q次操作,每次将第二个网格图的子矩阵全部变成0或1,问每一次操作后两 ...
codefores 786B. Legacy(最短路，线段树优化拆点，好题)
题目链接 B. Legacy time limit per test2 seconds memory limit per test256 megabytes inputstandard input o ...
【题解】BZOJ 3065: 带插入区间K小值——替罪羊树套线段树
题目传送门题解 orz vfk的题解 3065: 带插入区间K小值系列题解一二三四惨一开始用了一种空间常数很大的方法,每次重构的时候merge两颗线段树,然后无限RE(其实是MLE). ...

HDU_4125 Moles 线段树

HDU_4125 Moles 线段树相关推荐

最新文章

热门文章