2011 Asia Fuzhou Regional Contest-1005 hdu4125 Moles

题目描述：

    给N个数（标号为1-N）按顺序建一棵二叉排序树，然后按数字从小到大遍历这棵树，每走过一个标号为奇数的节点记录1，偶数记录0，访问完所有节点回到地面上之后便能得到一串0-1序列，之后给另外一个0-1串s，问遍历得到的串包含多少个s（可以重叠） n不大于600000

北大出的题还是很好的，题目读起来很简单，不是什么模板题不是超难的论文题也不是特简单的水题。解题思路很清晰，但却需要不断优化，否则10^5的数据很容易超时或者爆内存。这题改了N次，在TLE，MLE，RE折磨N次之后才勉强A掉

题目很简单，但有三个地方需要注意：

1. 建树不能只是模拟插入过程递归写，N为10＾5，给0、1、2、3.......N 时会超时，需要稳定Nlog(N)
2. 遍历的时候不能递归写，会栈溢出
3. 字符串匹配next数组需要处理到匹配串的长度，即需要算出next[m]

另外，建树遍历什么的一定不要用动态内存写，会很麻烦，最好提前开个数组

#include<iostream>#include<cstdio>#include<cstring>#include<iterator>#include<memory.h>#include<set>using namespace std;bool visited[600005];char str[1800005];int cnt,ct,size,res,next[7007];struct Node{int num;Node* left;Node* right;};struct cmp{bool operator()(const Node* a,const Node* b){return (a->num)<(b->num);}};Node* stac[1800005];Node* node = new Node[600005];set<Node*,cmp> visit;set<Node*,cmp>::iterator it;void Next(const char s[]){memset(next,0,sizeof(next));int j,t,m;t=next[0]=-1;j=0;m=strlen(s);while(j<m){if(t<0||s[j]==s[t]) next[++j]=++t;else t=next[t];}return;}void Mat(const char s1[],const char s2[]){int i,j,m;m=strlen(s2);i=j=0;while(i<ct){    if(j==m) res++;        if(j<0||s1[i]==s2[j]){            i++;            j++;        }        else j=next[j];    }    if(j==m) res++;}void dfs(Node* curr){size=0;memset(visited,0,sizeof(visited));stac[size++]=curr;while(size>0){curr=stac[size-1];size--;if((curr->num)%2) str[ct++]='1';else str[ct++]='0';if(!visited[curr->num]){visited[curr->num]=true;if(curr->right){stac[size++]=curr;stac[size++]=curr->right;}if(curr->left){stac[size++]=curr;stac[size++]=curr->left;}}}}int main(){Node* tree;Node* tmpNode;int k,n,num;char harStr[7007];while(scanf("%d",&k)!=EOF){for(int p=1;p<=k;p++){visit.clear();ct=res=cnt=0;tree=NULL;scanf("%d",&n);scanf("%d",&num);node[cnt].num=num;node[cnt].left=NULL;node[cnt].right=NULL;tree=&node[cnt];cnt++;visit.insert(tree);for(int i=1;i<n;i++){scanf("%d",&num);node[cnt].num=num;node[cnt].left=NULL;node[cnt].right=NULL;tmpNode=&node[cnt];cnt++;it=visit.lower_bound(tmpNode);if(it!=visit.end()&&((*it)->left==NULL)){(*it)->left=tmpNode;}else{it--;(*it)->right=tmpNode;}visit.insert(tmpNode);}dfs(tree);scanf("%s",harStr);Next(harStr);Mat(str,harStr);printf("Case #%d: %d\n",p,res);}}return 0;}