hdu 4622 Reincarnation

http://acm.hdu.edu.cn/showproblem.php?pid=4622

用字典树把每一个字符串对应成一个整数 相同的字符串对应到相同的整数上

把所用的串对应的整数放在一个数组里 比如书字符串s[l...r]对应的整数是 k

那么二维数组 [l][r] 就等于k

假设一个对应好的二维数组  左下角是原点

3     4     5     2

2     3     4     0

1     6     0     0

2     0     0     0

这样求解 从l到r的不同字符串的个数 其实就是求 从[l][r] 到右下角所在的矩阵所包含不同整数的个数(不包括0)

这里需要一定的去重处理 处理后是

-1    0    1     1

0     1    1     0

1     1    0     0

1     0    0     0

然后一边dp就可以求出所有答案

注意常用的next[26]写法的字典树有可能超内存 要优化

代码：

#include<iostream>

#include<cstdio>

#include<algorithm>

#include<string>

#include<cstring>

#include<cmath>

#include<set>

using namespace std;



typedef long long ll;

typedef pair<double,double>ppd;

const double PI = acos(-1.);

const double eps = (1e-9);

const int N=2005;

const int M=2000000;

const int K=27;

char s[N];

int c[N][N];

int last[M];

struct node

{

    int head;

    int k;

}trie[M];

struct node1

{

    int j,next;

    char c;

}edge[M];

int num,cnt,I;

int add(char a,int &x)

{

    bool flag=false;

    for(int t=trie[x].head;t!=-1;t=edge[t].next)

    {

        if(edge[t].c==a)

        {flag=true;x=edge[t].j;break;}

    }

    if(flag==true)

    return trie[x].k;

    trie[cnt].head=-1;

    trie[cnt].k=num++;

    edge[I].c=a;

    edge[I].j=cnt;

    edge[I].next=trie[x].head;

    trie[x].head=I++;

    x=cnt++;

    return trie[x].k;

}

int main()

{

    //freopen("data.in","r",stdin);

    int T;

    scanf("%d",&T);

    while(T--)

    {

        trie[0].head=-1;

        trie[0].k=-1;

        I=0;num=1,cnt=1;

        memset(c,0,sizeof(c));

        memset(last,-1,sizeof(last));

        scanf(" ");

        gets(s);

        int n=strlen(s);

        for(int j=0;j<n;++j)

        for(int i=j,t=0;i>=0;--i)

        {

            int x=i+1;

            int y=j+1;

            int k=add(s[i],t);

            if(last[k]==-1)

            {

                last[k]=x;

                ++c[x][y];

            }else if(last[k]<x)

            {

                ++c[x][y];

                --c[last[k]][y];

                last[k]=x;

            }

        }

        for(int j=1;j<=n;++j)

        for(int i=n;i>=1;--i)

        c[i][j]+=c[i][j-1]+c[i+1][j]-c[i+1][j-1];

        int q;

        scanf("%d",&q);

        while(q--)

        {

            int x,y;

            scanf("%d %d",&x,&y);

            printf("%d\n",c[x][y]);

        }

    }

    return 0;

}