Cod sursa(job #338926)

Utilizator GavrilaVladGavrila Vlad GavrilaVlad Data 7 august 2009 15:21:50
Problema PScPld Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.28 kb
#include <stdio.h>
#include <string.h>

using namespace std;

#define maxn 2000010

long n, i, j, k, l, r, lung, poz, sol, x, v[maxn], t[maxn], op, l1, o2, uac;
char s[maxn];

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