Cod sursa(job #338905)

Utilizator GavrilaVladGavrila Vlad GavrilaVlad Data 7 august 2009 14:18:15
Problema PScPld Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.68 kb
#include <stdio.h>
#include <string.h>

using namespace std;

#define maxn 2000010

long n, i, j, k, l, r, lung, poz, sol;
char s[maxn];
long v[maxn], t[maxn];

int main()
{
    freopen("pscpld.in", "r", stdin);
    freopen("pscpld.out", "w", stdout);
    scanf("%s", s+1);
    for(n=1; s[n]>0; ++n)
    {
  //      printf("%c", s[n]);
    }
    --n;
    for(i=1; i<=2*n-1; i++)
    {
        if(i%2)
        {
            l=r=i/2+1;
            lung=1;
            if(t[i])
            {
                lung=v[2*t[i]-i];
                poz=t[i]/2+1;
                if(poz+v[t[i]]-1>i/2+1+lung-1)
                {
                    lung=poz+v[t[i]]-1-(i/2+1)+1;
                }
                l-=lung+1;
                r+=lung-1;
            }
            while(s[l-1]==s[r+1] && l>1 && r<n)
            {
                l--;
                r++;
                lung++;
                t[r*2]=t[r*2-1]=i;
            }
            sol+=lung;
            v[i]=lung;
        }
        else
        {
            l=i/2;
            r=l+1;
            lung=0;
            if(t[i])
            {
                lung=v[2*t[i]-i];
                poz=t[i]/2+1;
                if(poz+v[t[i]]-1>i/2+1+lung-1)
                {
                    lung=poz+v[t[i]]-1-(i/2+1)+1;
                }
                l-=lung;
                r+=lung;
            }
            while(s[l-1]==s[r+1] && l>1 && r<n)
            {
                l--;
                r++;
                lung++;
                t[r*2]=t[r*2-1]=i;
            }
            sol+=lung;
            v[i]=lung;
        }
    }
    printf("%d\n", sol);
    return 0;
}