HDU 1251 统计难题

HDU 1251
中文题意，不再说了。
字典树裸题，比赛时用的指针实现的导致MLE||RE很难受。赛后套了一下数组实现的直接A了。思想大家也理解了。具体
说一下数组如何实现。

//ZQW写法
#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <map>
#include <vector>
using namespace std;
int num[400010];
int nex[400010][26];
struct tire
{
    int L,root;
    int newnode()
    {
        for(int i=0;i<26;i++)//初始化这个新节点。因为这个新节点什么也没有连就设置为-1；
            nex[L][i]=-1;
        num[L++]=0;
        return L-1;
    }
    void init()//初始化
    {
        L=0;
        root=newnode();//root为根节点初始化为0
    }
    void insert1(char s[])
    {
        int now=root;//从根节点开始便利
        for(int i=0;s[i];i++)
        {
            if(nex[now][s[i]-'a']==-1)//如果这个节点没有连接下一个s[i]-'a'节点。那么就开一个新节点。并将这两个节点连接
                nex[now][s[i]-'a']=newnode();
            now=nex[now][s[i]-'a'];//遍历下一个节点
            num[now]++;//每次遍历到这个节点，都+1.这样就知道前缀字母出现多少次
        }
    }
    int query(char s[])
    {
        int now=root;//从根节点开始查询
        for(int i=0;s[i];i++)
        {
            if(nex[now][s[i]-'a']!=-1)//如果不是新节点就继续遍历到尾
                now=nex[now][s[i]-'a'];
            else//如果是新节点，那么原字符串就没有出现过，直接返回0
                return 0;
        }
        return num[now];
    }
};
int main()
{
    char str[100];
    int len,sum=0;
    tire tree;
    tree.init();//字典树初始化
    while(gets(str))
    {
        len=strlen(str);
        if(len==0) break;
        tree.insert1(str);
    }
    while(gets(str)!=NULL)
    {
        printf("%d\n",tree.query(str));
    }
//    cout<<sum<<endl;
}

//zxy写法
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <iostream>

using namespace std;
const int inf=0x3f3f3f3f;
#define ll long long
const int maxn=100000+100;
int n,m;
int k;
struct node
{
    int id;
    int l,r;
} e[maxn];
int a[maxn];
int cmp(node a,node b)
{
    if(a.l/k!=b.l/k)
        return a.l/k<b.l/k;
    else
        return a.r<b.r;
}
int ans[maxn];
bool flag[maxn];
void solve()
{
    memset(flag,0,sizeof(flag));
    int l=1,r=0,sum=0;
    for(int i=0; i<m; i++)
    {
        while(r<e[i].r)
        {
            r++;
            int num=a[r];
            flag[num]=true;
            if(flag[num-1]==false&&flag[num+1]==false) sum++;
            if(flag[num-1]==true&&flag[num+1]==true) sum--;
        }
        while(r>e[i].r)
        {
            int num=a[r];
            flag[num]=false;
            if(flag[num-1]==false&&flag[num+1]==false) sum--;
            if(flag[num-1]==true&&flag[num+1]==true) sum++;
            r--;
        }
        while(l<e[i].l)
        {

            int num=a[l];
            flag[num]=false;
            if(flag[num-1]==false&&flag[num+1]==false) sum--;
            if(flag[num-1]==true&&flag[num+1]==true) sum++;
            l++;
        }
        while(l>e[i].l)
        {
            l--;
            int num=a[l];
            flag[num]=true;
            if(flag[num-1]==false&&flag[num+1]==false) sum++;
            if(flag[num-1]==true&&flag[num+1]==true) sum--;

        }
        ans[e[i].id]=sum;
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        k=(int)sqrt(n);
        for(int i=1; i<=n; i++)
            scanf("%d",&a[i]);
        for(int i=0; i<m; i++)
        {
            scanf("%d%d",&e[i].l,&e[i].r);
            e[i].id=i;
        }
        sort(e,e+m,cmp);
        solve();
        for(int i=0; i<m; i++)
            printf("%d\n",ans[i]);

    }
    return 0;
}