UVALive 3942 Remember the Word Tire+DP-白红宇

UVALive 3942 Remember the Word Tire+DP

阅读量：5826 次

发布时间：2019-06-18

本文共 1948 字，大约阅读时间需要 6 分钟。

dp[i]=sum{dp[j]} ,s[i...j-1]为出现过的单词，每次从s[i]开始查找全部出现的前缀s[i..j],前缀在Tire中存在则转移状态。

//#pragma comment(linker, "/STACK:1024000000,1024000000")#include
      
       #include
       
        #include
        
         #include
         
          #include
          
           #include
           
            #include
            
             #include
             
              #include
              
               #include
               #include
                
                 #include
                 
                  #include
                  
                   #include
                   
                    using namespace std;typedef long long ll;typedef pair
                    
                      pii;#define pb(a) push_back(a)#define INF 0x1f1f1f1f#define lson idx<<1,l,mid#define rson idx<<1|1,mid+1,r#define PI 3.1415926535898template
                     
                       T min(const T& a,const T& b,const T& c) { return min(min(a,b),min(a,c));}template
                      
                        T max(const T& a,const T& b,const T& c) { return max(max(a,b),max(a,c));}void debug() {#ifdef ONLINE_JUDGE#else freopen("d:\\in.txt","r",stdin); freopen("d:\\out1.txt","w",stdout);#endif}int getch() { int ch; while((ch=getchar())!=EOF) { if(ch!=' '&&ch!=' ')return ch; } return EOF;}struct TireNode{ int flag; TireNode *next[26]; TireNode() { flag=0; for(int i=0;i<26;i++) next[i]=NULL; }}*root;int Insert(const char *s){ TireNode *p=root; for(int i=0;s[i]!='\0';i++) { int v=s[i]-'a'; if(p->next[v]==NULL) { p=p->next[v]=new TireNode; }else p=p->next[v]; } p->flag=1; return 0;}char str[300300];int dp[300300];int f(int k){ if(str[k]=='\0')return 1; if(dp[k]>=0)return dp[k]; int x=0; TireNode *p=root; for(int i=0;str[i+k]!='\0';i++) { int v=str[i+k]-'a'; if(p->next[v]==NULL)break; else p=p->next[v]; if(p->flag) { x=(x+f(i+k+1))%20071027; } } return dp[k]=x;}int del(TireNode *p){ for(int i=0;i<26;i++) { if(p->next[i]!=NULL) del(p->next[i]); } delete p;}int main(){ int ca=0; while(scanf("%s",str)!=EOF) { root=new TireNode; int n; scanf("%d",&n); for(int i=1;i<=n;i++) { char s[110]; scanf("%s",s); Insert(s); } memset(dp,-1,sizeof(dp)); int num=f(0); printf("Case %d: %d\n",++ca,num); del(root); } return 0;}