hdu 4125 Moles(kmp+树状数组)

题目链接：hdu 4125 Moles

题意：

给你n个数，让你按键值建一个平衡二叉树，然后奇数为0，偶数为1，然后可以遍历这颗树得到一个欧拉序列，现在给你一个串，问你出现了几次。

题解：

建树的时候要引用一个结论：就是新插入的数的父亲，要么是比它大的最小的那个元素，要么是比它小的最大的那个元素。

这样才能nloglogn建树，其实可以用set nlogn建树，但是空间要被卡，然后换用树状数组来替代set。

我这份代码没用KMP跑了4000+，用了kmp只会跑2000+。kmp还是在大串上比较有优势。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int N=6e5+7;
int t,n,cas,a[N],l[N],r[N],S[N],top,len,C[N],root,mi,mx;
char s[N*2],b[N],vis[N];

void add(int x){while(x<=n)C[x]++,x+=x&-x;}
inline int sum(int x){int an=0;while(x)an+=C[x],x-=x&-x;return an;}

void insert(int x)
{
    add(x);
    int tt=sum(x);
    if(!root)
    {
        root=mi=mx=x;
        return;
    }
    if(x>mx)
    {
        r[mx]=x,mx=x;
        return;
    }
    if(x<mi)
    {
        l[mi]=x,mi=x;
        return;
    }
    mi=min(mi,x),mx=max(mx,x);
    int ll=1,rr=x-1,mid;
    while(ll<rr)
    {
        mid=ll+rr>>1;
        if(sum(mid)==tt-1)rr=mid;
        else ll=mid+1;
    }
    if(!r[rr])
    {
        r[rr]=x;
        return;
    }
    ll=x+1,rr=n;
    while(ll<rr)
    {
        mid=ll+rr>>1;
        if(sum(mid)!=tt)rr=mid;
        else ll=mid+1;
    }
    l[rr]=x;
}

void init()
{
    root=0;
    F(i,0,n)C[i]=0;
    F(i,1,n)l[i]=r[i]=0;
    F(i,1,n)insert(a[i]);
}

void dfs()
{
    F(i,0,n)vis[i]=0;
    top=1,S[1]=a[1];
    while(top)
    {
        int now=S[top];
        s[len++]='0'+(now&1);
        if(l[now]&&!vis[l[now]]){S[++top]=l[now];continue;}
        if(r[now]&&!vis[r[now]]){S[++top]=r[now];continue;}
        top--;
        vis[now]=1;
    }
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        F(i,1,n)scanf("%d",a+i);
        scanf("%s",b);
        int lenb=strlen(b);
        init();
        len=0;
        dfs();
        int ans=0;
        F(i,0,len-1)
        {
            if(s[i]==b[0])
            {
                int p=i,fg=1;
                F(j,0,lenb-1)if(p+j>=len||s[p+j]!=b[j]){fg=0;break;}
                ans+=fg;
            }
        }
        printf("Case #%d: %d
",++cas,ans);
    }
    return 0;
}