// Begin include: "../../template/template.hpp" using namespace std; // intrinstic #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include // utility // Begin include: "util.hpp" namespace yamada { using ll = long long; using i64 = long long; using u64 = unsigned long long; using i128 = __int128_t; using u128 = __uint128_t; using lld = long double; template using V = vector; template using VV = vector>; template using VVV = vector>>; template using VVVV = vector>>>; using vl = vector; using vd = V; using vs = V; using vvl = vector>; using vvvl = vector>>; using vvvvl = vector>>>; template using minpq = priority_queue, greater>; template using maxpq = priority_queue, less>; template struct P : pair { template P(Args... args) : pair(args...) {} using pair::first; using pair::second; P &operator+=(const P &r) { first += r.first; second += r.second; return *this; } P &operator-=(const P &r) { first -= r.first; second -= r.second; return *this; } P &operator*=(const P &r) { first *= r.first; second *= r.second; return *this; } template P &operator*=(const S &r) { first *= r, second *= r; return *this; } P operator+(const P &r) const { return P(*this) += r; } P operator-(const P &r) const { return P(*this) -= r; } P operator*(const P &r) const { return P(*this) *= r; } template P operator*(const S &r) const { return P(*this) *= r; } P operator-() const { return P{-first, -second}; } }; using pl = P; using vp = V; using vvp = VV; constexpr int inf = 1001001001; constexpr long long infLL = 4004004004004004004LL; template inline bool amin(T &x, U y) { return (y < x) ? (x = y, true) : false; } template inline bool amax(T &x, U y) { return (x < y) ? (x = y, true) : false; } template inline T Max(const vector &v) { return *max_element(begin(v), end(v)); } template inline T Min(const vector &v) { return *min_element(begin(v), end(v)); } template inline long long Sum(const vector &v) { return accumulate(begin(v), end(v), T(0)); } template int lb(const vector &v, const T &a) { return lower_bound(begin(v), end(v), a) - begin(v); } template int ub(const vector &v, const T &a) { return upper_bound(begin(v), end(v), a) - begin(v); } constexpr long long TEN(int n) { long long ret = 1, x = 10; for (; n; x *= x, n >>= 1) ret *= (n & 1 ? x : 1); return ret; } template vector mkrui(const vector &v, bool rev = false) { vector ret(v.size() + 1); if (rev) { for (int i = int(v.size()) - 1; i >= 0; i--) ret[i] = v[i] + ret[i + 1]; } else { for (int i = 0; i < int(v.size()); i++) ret[i + 1] = ret[i] + v[i]; } return ret; }; template vector mkuni(const vector &v) { vector ret(v); sort(ret.begin(), ret.end()); ret.erase(unique(ret.begin(), ret.end()), ret.end()); return ret; } template vector mkord(int N, F f) { vector ord(N); iota(begin(ord), end(ord), 0); sort(begin(ord), end(ord), f); return ord; } template vector mkinv(vector &v) { int max_val = *max_element(begin(v), end(v)); vector inv(max_val + 1, -1); for (int i = 0; i < (int)v.size(); i++) inv[v[i]] = i; return inv; } vector mkiota(int n) { vector ret(n); iota(begin(ret), end(ret), 0); return ret; } template T mkrev(const T &v) { T w{v}; reverse(begin(w), end(w)); return w; } template bool nxp(T &v) { return next_permutation(begin(v), end(v)); } // 返り値の型は入力の T に依存 // i 要素目 : [0, a[i]) template vector> product(const vector &a) { vector> ret; vector v; auto dfs = [&](auto rc, int i) -> void { if (i == (int)a.size()) { ret.push_back(v); return; } for (int j = 0; j < a[i]; j++) v.push_back(j), rc(rc, i + 1), v.pop_back(); }; dfs(dfs, 0); return ret; } template vector Digit(T a, const U &x, int siz = -1) { vector ret; while (a > 0) { ret.emplace_back(a % x); a /= x; } if (siz >= 0) while ((int)ret.size() < siz) ret.emplace_back(0); return ret; } // F : function(void(T&)), mod を取る操作 // T : 整数型のときはオーバーフローに注意する template T Power(T a, long long n, const T &I, const function &f) { T res = I; for (; n; f(a = a * a), n >>= 1) { if (n & 1) f(res = res * a); } return res; } // T : 整数型のときはオーバーフローに注意する template T Power(T a, long long n, const T &I = T{1}) { return Power(a, n, I, function{[](T &) -> void {}}); } template T Rev(const T &v) { T res = v; reverse(begin(res), end(res)); return res; } template vector Transpose(const vector &v) { using U = typename T::value_type; if(v.empty()) return {}; int H = v.size(), W = v[0].size(); vector res(W, T(H, U{})); for (int i = 0; i < H; i++) for (int j = 0; j < W; j++) res[j][i] = v[i][j]; return res; } template vector Rotate(const vector &v, int clockwise = true) { using U = typename T::value_type; int H = v.size(), W = v[0].size(); vector res(W, T(H, U{})); for (int i = 0; i < H; i++) for (int j = 0; j < W; j++) { if (clockwise) res[W - 1 - j][i] = v[i][j]; else res[j][H - 1 - i] = v[i][j]; } return res; } template T bisect(T ok, T bad, F pred) { if (ok == bad) return ok; if (!pred(ok)) return ok; while (bad - ok > 1) { T mid = ok + (bad - ok) / 2; (pred(mid) ? ok : bad) = mid; } return bad; } template T bisect_double(T ok, T bad, F pred, int iter = 100) { if (ok == bad) return ok; if (!pred(ok)) return ok; while (iter--) { T mid = ok + (bad - ok) / 2; (pred(mid) ? ok : bad) = mid; } return bad; } template bool inLR(T L, T x, T R){ return (L <= x && x < R); } bool YESNO(bool b) { cout << (b ? "YES\n" : "NO\n"); return b; } bool YesNo(bool b) { cout << (b ? "Yes\n" : "No\n"); return b; } template void mout(mint a, int M = 100) { if (a == 0) { cout << 0 << "\n"; return; } for (int i = 0; i <= M; i++) for (int j = 1; j <= M; j++) { mint val = (mint)i / j; if (val == a) { if (j == 1) cout << i << "\n"; else cout << i << "/" << j << "\n"; return; } else if (val == -a) { if (j == 1) cout << -i << "\n"; else cout << -i << "/" << j << "\n"; return; } } cout << "NF\n"; } template void mout(std::vector A, int M = 100) { int N = A.size(); for (int pos = 0; pos < N; pos++) { if (A[pos] == 0) { cout << 0 << (pos == N - 1 ? "\n" : " "); continue; } bool found = false; for (int i = 0; i <= M; i++) { for (int j = 1; j <= M; j++) { mint val = (mint)i / j; if (val == A[pos]) { if (j == 1) cout << i << (pos == N - 1 ? "\n" : " "); else cout << i << "/" << j << (pos == N - 1 ? "\n" : " "); found = true; break; } else if (val == -A[pos]) { if (j == 1) cout << -i << (pos == N - 1 ? "\n" : " "); else cout << -i << "/" << j << (pos == N - 1 ? "\n" : " "); found = true; break; } } if (found) break; } if (!found) cout << "NF" << (pos == N - 1 ? "\n" : " "); } } bool is_square(uint64_t n) { if (n < 2) return true; uint64_t r = static_cast(sqrtl(static_cast(n))); if (r * r == n) return true; ++r; return r * r == n; } template struct CumulativeSum { std::vector S; CumulativeSum(std::vector &A) { int N = A.size(); S.resize(N + 1); for (int i = 0; i < N; i++) S[i + 1] = S[i] + A[i]; } T query(int l, int r) { return (l <= r ? S[r] - S[l] : (T)0); } T get_val(int i) { return S[i + 1] - S[i]; } }; template T extgcd(T a, T b, T &x, T &y) { T d = a; if(b != 0) { d = extgcd(b, a % b, y, x); y -= (a / b) * x; } else x = 1, y = 0; return d; } // floor(sqrt(x)) long long isqrt (long long x) { long long y = sqrt(x); while (y * y > x) y--; while ((y + 1) * (y + 1) <= x) y++; return y; } template T Floor(T a, T b) { return a / b - (a % b && (a ^ b) < 0); } template T Ceil(T a, T b) { return a / b + (a % b && (a ^ b) >= 0); } } // namespace yamada // End include: "util.hpp" // bit operation // Begin include: "bitop.hpp" namespace yamada { __attribute__((target("popcnt"))) inline int popcnt(const u64 &a) { return __builtin_popcountll(a); } inline int lsb(const u64 &a) { return a ? __builtin_ctzll(a) : 64; } inline int ctz(const u64 &a) { return a ? __builtin_ctzll(a) : 64; } inline int msb(const u64 &a) { return a ? 63 - __builtin_clzll(a) : -1; } template inline int gbit(const T &a, int i) { return (a >> i) & 1; } template inline void sbit(T &a, int i, bool b) { if (gbit(a, i) != b) a ^= T(1) << i; } constexpr long long PW(int n) { return 1LL << n; } constexpr long long MSK(int n) { return (1LL << n) - 1; } } // namespace yamada // End include: "bitop.hpp" // inout // Begin include: "inout.hpp" namespace yamada { template ostream &operator<<(ostream &os, const pair &p) { os << p.first << " " << p.second; return os; } template istream &operator>>(istream &is, pair &p) { is >> p.first >> p.second; return is; } template ostream &operator<<(ostream &os, const vector &v) { int s = (int)v.size(); for (int i = 0; i < s; i++) os << (i ? " " : "") << v[i]; return os; } template istream &operator>>(istream &is, vector &v) { for (auto &x : v) is >> x; return is; } istream &operator>>(istream &is, __int128_t &x) { string S; is >> S; x = 0; int flag = 0; for (auto &c : S) { if (c == '-') { flag = true; continue; } x *= 10; x += c - '0'; } if (flag) x = -x; return is; } istream &operator>>(istream &is, __uint128_t &x) { string S; is >> S; x = 0; for (auto &c : S) { x *= 10; x += c - '0'; } return is; } ostream &operator<<(ostream &os, __int128_t x) { if (x == 0) return os << 0; if (x < 0) os << '-', x = -x; string S; while (x) S.push_back('0' + x % 10), x /= 10; reverse(begin(S), end(S)); return os << S; } ostream &operator<<(ostream &os, __uint128_t x) { if (x == 0) return os << 0; string S; while (x) S.push_back('0' + x % 10), x /= 10; reverse(begin(S), end(S)); return os << S; } void in() {} template void in(T &t, U &...u) { cin >> t; in(u...); } void out() { cout << "\n"; } template void out(const T &t, const U &...u) { cout << t; if (sizeof...(u)) cout << sep; out(u...); } struct IoSetupYamada { IoSetupYamada() { cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(15); cerr << fixed << setprecision(7); } } iosetupyamada; } // namespace yamada // End include: "inout.hpp" // macro // Begin include: "macro.hpp" #define each(x, v) for (auto&& x : v) #define each2(x, y, v) for (auto&& [x, y] : v) #define each3(x, y, z, v) for (auto&& [x, y, z] : v) #define all(v) (v).begin(), (v).end() #define rep1(a) for (long long _ = 0; _ < (long long)(a); ++_) #define rep2(i, a) for (long long i = 0; i < (long long)(a); ++i) #define rep3(i, a, b) for (long long i = a; i < (long long)(b); ++i) #define rep4(i, a, b, c) for (long long i = a; i < (long long)(b); i += c) #define overload4(a, b, c, d, e, ...) e #define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__) #define rep1r(a) for (long long i = (long long)(a)-1; i >= 0LL; --i) #define rep2r(i, a) for (long long i = (long long)(a)-1; i >= 0LL; --i) #define rep3r(i, a, b) for (long long i = (long long)(b)-1; i >= (long long)(a); --i) #define overload3(a, b, c, d, ...) d #define repr(...) overload3(__VA_ARGS__, rep3r, rep2r, rep1r)(__VA_ARGS__) #define eb emplace_back #define mkp make_pair #define mkt make_tuple #define fi first #define se second #define vv(type, name, h, ...) \ vector > name(h, vector(__VA_ARGS__)) #define vvv(type, name, h, w, ...) \ vector>> name( \ h, vector>(w, vector(__VA_ARGS__))) #define vvvv(type, name, a, b, c, ...) \ vector>>> name( \ a, vector>>( \ b, vector>(c, vector(__VA_ARGS__)))) #define ini(...) \ int __VA_ARGS__; \ in(__VA_ARGS__) #define inl(...) \ long long __VA_ARGS__; \ in(__VA_ARGS__) #define ins(...) \ string __VA_ARGS__; \ in(__VA_ARGS__) #define in2(s, t) \ for (int i = 0; i < (int)s.size(); i++) { \ in(s[i], t[i]); \ } #define in3(s, t, u) \ for (int i = 0; i < (int)s.size(); i++) { \ in(s[i], t[i], u[i]); \ } #define in4(s, t, u, v) \ for (int i = 0; i < (int)s.size(); i++) { \ in(s[i], t[i], u[i], v[i]); \ } #define die(...) \ do { \ yamada::out(__VA_ARGS__);\ return; \ } while (0) // End include: "macro.hpp" namespace yamada { void solve(); } int main() { yamada::solve(); } // End include: "../../template/template.hpp" // Begin include: "../../string/rolling-hash.hpp" #include #include using namespace std; // Begin include: "../internal/internal-hash.hpp" namespace internal { using i64 = long long; using u64 = unsigned long long; using u128 = __uint128_t; template struct Hash : array { using array::operator[]; static constexpr int n = BASE_NUM; Hash() : array() {} static constexpr u64 md = (1ull << 61) - 1; constexpr static Hash set(const i64 &a) { Hash res; fill(begin(res), end(res), cast(a)); return res; } Hash &operator+=(const Hash &r) { for (int i = 0; i < n; i++) if (((*this)[i] += r[i]) >= md) (*this)[i] -= md; return *this; } Hash &operator+=(const i64 &r) { u64 s = cast(r); for (int i = 0; i < n; i++) if (((*this)[i] += s) >= md) (*this)[i] -= md; return *this; } Hash &operator-=(const Hash &r) { for (int i = 0; i < n; i++) if (((*this)[i] += md - r[i]) >= md) (*this)[i] -= md; return *this; } Hash &operator-=(const i64 &r) { u64 s = cast(r); for (int i = 0; i < n; i++) if (((*this)[i] += md - s) >= md) (*this)[i] -= md; return *this; } Hash &operator*=(const Hash &r) { for (int i = 0; i < n; i++) (*this)[i] = modmul((*this)[i], r[i]); return *this; } Hash &operator*=(const i64 &r) { u64 s = cast(r); for (int i = 0; i < n; i++) (*this)[i] = modmul((*this)[i], s); return *this; } Hash operator+(const Hash &r) { return Hash(*this) += r; } Hash operator+(const i64 &r) { return Hash(*this) += r; } Hash operator-(const Hash &r) { return Hash(*this) -= r; } Hash operator-(const i64 &r) { return Hash(*this) -= r; } Hash operator*(const Hash &r) { return Hash(*this) *= r; } Hash operator*(const i64 &r) { return Hash(*this) *= r; } Hash operator-() const { Hash res; for (int i = 0; i < n; i++) res[i] = (*this)[i] == 0 ? 0 : md - (*this)[i]; return res; } friend Hash pfma(const Hash &a, const Hash &b, const Hash &c) { Hash res; for (int i = 0; i < n; i++) res[i] = modfma(a[i], b[i], c[i]); return res; } friend Hash pfma(const Hash &a, const Hash &b, const i64 &c) { Hash res; u64 s = cast(c); for (int i = 0; i < n; i++) res[i] = modfma(a[i], b[i], s); return res; } Hash pow(long long e) { Hash a{*this}, res{Hash::set(1)}; for (; e; a *= a, e >>= 1) { if (e & 1) res *= a; } return res; } static Hash get_basis() { static auto rand_time = chrono::duration_cast( chrono::high_resolution_clock::now().time_since_epoch()) .count(); static mt19937_64 rng(rand_time); Hash h; for (int i = 0; i < n; i++) { while (isPrimitive(h[i] = rng() % (md - 1) + 1) == false) ; } return h; } private: static u64 modpow(u64 a, u64 b) { u64 r = 1; for (a %= md; b; a = modmul(a, a), b >>= 1) r = modmul(r, a); return r; } static bool isPrimitive(u64 x) { for (auto &d : vector{2, 3, 5, 7, 11, 13, 31, 41, 61, 151, 331, 1321}) if (modpow(x, (md - 1) / d) <= 1) return false; return true; } static inline constexpr u64 cast(const long long &a) { return a < 0 ? a + md : a; } static inline constexpr u64 modmul(const u64 &a, const u64 &b) { u128 d = u128(a) * b; u64 ret = (u64(d) & md) + u64(d >> 61); return ret >= md ? ret - md : ret; } static inline constexpr u64 modfma(const u64 &a, const u64 &b, const u64 &c) { u128 d = u128(a) * b + c; u64 ret = (d >> 61) + (u64(d) & md); return ret >= md ? ret - md : ret; } }; } // namespace internal /** * @brief ハッシュ構造体 * @docs docs/internal/internal-hash.md */ // End include: "../internal/internal-hash.hpp" template struct RollingHash { using Hash = internal::Hash; Str data; vector hs, pw; int siz; static Hash basis; RollingHash(const Str &S = Str()) { build(S); } void build(const Str &S) { data = S; siz = S.size(); hs.resize(siz + 1); pw.resize(siz + 1); pw[0] = Hash::set(1); hs[0] = Hash::set(0); for (int i = 1; i <= siz; i++) { pw[i] = pw[i - 1] * basis; hs[i] = pfma(hs[i - 1], basis, S[i - 1]); } } Hash get(int l, int r = -1) const { if (r == -1) r = siz; return pfma(hs[l], -pw[r - l], hs[r]); } // T の hash を返す static Hash get_hash(const Str &T) { Hash ret = Hash::set(0); for (int i = 0; i < (int)T.size(); i++) ret = pfma(ret, basis, T[i]); return ret; } // a + b の hash を返す // 引数 : a, b, b の長さ static Hash unite(Hash a, Hash b, long long bsize) { return pfma(a, basis.pow(bsize), b); } int find(Str &T, int lower = 0) const { auto ths = get_hash(T); for (int i = lower; i <= siz - (int)T.size(); i++) if (ths == get(i, i + (int)T.size())) return i; return -1; } static int lcp(const RollingHash &a, const RollingHash &b, int al, int bl) { int ok = 0, ng = min(a.size() - al, b.size() - bl) + 1; while (ok + 1 < ng) { int med = (ok + ng) / 2; (a.get(al, med + al) == b.get(bl, med + bl) ? ok : ng) = med; } return ok; } static int strcmp(const RollingHash &a, const RollingHash &b, int al, int bl, int ar = -1, int br = -1) { if (ar == -1) ar = a.size(); if (br == -1) br = b.size(); int n = min({lcp(a, b, al, bl), ar - al, br - bl}); return al + n == ar ? bl + n == br ? 0 : -1 : bl + n == br ? 1 : a.data[al + n] < b.data[bl + n] ? -1 : 1; } int size() const { return siz; } }; template typename RollingHash::Hash RollingHash::basis = internal::Hash::get_basis(); using roriha = RollingHash; /** * @brief Rolling Hash * @docs docs/string/rolling-hash.md */ // End include: "../../string/rolling-hash.hpp" void yamada::solve() { inl(N); vl A(N); in(A); vl B(N-1); rep(i,N-1)B[i]=A[i+1]-A[i]; vl C=B; reverse(all(C)); RollingHash rhB(B); RollingHash rhC(C); ll ans=0; rep(i,1,N){ ll bns=rhB.lcp(rhB,rhC,i,N-i); /* out(i,":",bns); */ ans+=bns+1; } rep(i,N){ ll bns=rhB.lcp(rhB,rhC,i,N-i-1); /* out(i,":",bns); */ ans+=bns+1; } out(ans); }