Cod sursa(job #1159560)

Utilizator RaduGabriel2012Dinu Radu RaduGabriel2012 Data 29 martie 2014 18:26:39
Problema Prefix Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.88 kb
#include <iostream>
#include <fstream>
#include <cstring>
using namespace std;
ifstream f("prefix.in");
ofstream g("prefix.out");
 int n,k;
 char a[1000005];
 int p[1000005],last[1000005],cr[1000005],per;
int main()
{ int i,t;
    f>>t; f.getline(a,2);
  for(;t;t--)
   {
    f>>(a+1); a[0]=' ';
    n=strlen(a)-1;

   k=0;
   for(i=2;i<=n;i++)
   {
     while (k && a[k+1]!=a[i])
      k=p[k];

      if (a[k+1]==a[i]) k++;

      p[i]=k;
   }

   for(i=1;i<=n;i++)
    {if (p[i]==1)
     last[i]=i; else last[i]=last[i-1];
      if (p[i]==1) cr[i]=1;
       else
        {if (p[i]>p[i-1]) cr[i]=cr[i-1]+1;
         else cr[i]=0;}
    }

   for(i=n;i>=1;i--)
    if (last[i]>0 && last[i]==(i-cr[i]+1))
     { per=last[i]-1;
        if (i%per==0 && ((p[i]/per+1)*per)==i)
         {g<<i<<"\n"; break;}
     }


    if (!i) g<<"0\n";

   }
    return 0;
}