Cod sursa(job #1475324)

Utilizator cojocarugabiReality cojocarugabi Data 23 august 2015 20:14:30
Problema Numarare Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.71 kb
# include <bits/stdc++.h>
using namespace std;
# define db double
# define ll long long
# define pi 3.14159265359
# define rad(x) (x * pi / 180)
const int p_1[] = {1,-1};
const ll mod1 = 100000004987ll;
const int mod2 = 1e9 + 7;
const int mod3 = 666013;
const int mod4 = 997;
const int mod5 = 1000004123;
template < class T > inline T sqr(T x) {return x*x;}
template < class T > inline T min(T a,T b,T c) {return min(a,min(b,c));}
template < class T > inline T min(T a,T b,T c,T d) {return min(min(a,b),min(b,c));}
template < class T > inline T max(T a,T b,T c) {return max(a,max(b,c));}
template < class T > inline T max(T a,T b,T c,T d) {return max(max(a,b),max(b,c));}
template < class T > inline T det(T a,T b,T c,T d) {return a * c - b * d;}
template < class T > inline T det(T s[3][3])
{
    return - s[0][0] * det(s[1][1],s[1][2],s[2][1],s[2][2]) + s[0][1] * det(s[1][0],s[1][2],s[2][0],s[2][2]) - s[0][2] * det(s[1][0],s[1][1],s[2][0],s[2][1]);
}
template < class T1 , class T2 , class T3 > inline T3 pow(T1 a,T2 b,T3 mod) {T3 ans = 1;while (b){if (b&1) ans = (1LL * ans * a) % mod;a = (1LL * a * a) % mod;b >>= 1;}return ans;}
template < class T > T phi(T n){T cnt = n,p = n,ans = n;for (T i = 2;i*i <= p;++i)if (!(cnt%i)){ans /= i;ans *= (i-1);while (!(cnt%i)) cnt /= i;}if (cnt > 1) ans /= cnt,ans *= (cnt - 1);return ans;}
template < class T > T f(T a,T b) {return a < b ? 0 : a / b + f(a / b);}
template < class T > T gcd(T a,T b) {return __gcd(a,b);}
class Reader {
  public:
    Reader(const string& filename):
            m_stream(filename),
            m_pos(kBufferSize - 1),
            m_buffer(new char[kBufferSize]) {
        next();
    }

    Reader& operator>>(int& value) {
        value = 0;
        while (current() < '0' || current() > '9')
            next();
        while (current() >= '0' && current() <= '9') {
            value = value * 10 + current() - '0';
            next();
        }
        return *this;
    }

  private:
    const int kBufferSize = 32768;

    char current() {
        return m_buffer[m_pos];
    }

    void next() {
        if (!(++m_pos != kBufferSize)) {
            m_stream.read(m_buffer.get(), kBufferSize);
            m_pos = 0;
        }
    }

    ifstream m_stream;
    int m_pos;
    unique_ptr<char[]> m_buffer;
};
int v[100005];
struct automat
{
    int link,len;
    map < int , int > next;
};
automat s[200005];
int n,last;
ll dp[200005];
void go(int k)
{
    int nw = ++n;
    s[nw].len = s[last].len + 1;
    dp[nw] = 1;
    int g = last;
    for (;g != -1 && !s[g].next[k];g = s[g].link) s[g].next[k] = nw;
    if (g == -1) s[nw].link = 0;
    else
    {
        int to = s[g].next[k];
        if (s[to].len == s[g].len + 1)
            s[nw].link = to;
        else
        {
            int clon = ++n;
            s[clon] = s[to];
            s[clon].len = s[g].len + 1;
            for (;g != -1 && s[g].next[k] == to;g = s[g].link) s[to].next[k] = clon;
            s[to].link = s[nw].link = clon;
        }
    }
    last = nw;
}
vector < int > inv[200005];
ll get(int k)
{
    for (auto it : inv[k]) dp[k] += get(it);
    return dp[k];
}
int main(void)
{
    s[0].link = -1;
    ifstream fi("numarare.in");
    ofstream fo("numarare.out");
    int len;
    fi>>len;
    for (int i = 1;i <= len;++i) fi>>v[i];
    int nxt = v[len];
    for (int i = len-1;i;--i)
    {
        int to = v[i];
        v[i] -= nxt;
        nxt = to;
    }
    --len;
    for (int i = 1;i <= len;++i) go(v[i]);
    for (int i = 1;i <= n;++i) inv[s[i].link].push_back(i);
    ll ans = 0;
    dp[0] = get(0);
    for (int i = 1;i <= n;++i) ans += dp[i] * (dp[i] - 1) / 2;
    return fo << ans << '\n',0;
}