#line 1 "/workspaces/compro/lib/data_structure/kth-smallest.hpp" #include #include #include #ifndef COMPRO_MATRIX #define COMPRO_MATRIX // 一点更新、一点削除、k番目に小さい値の取得ができるデータ構造を定義します template class KthSmallest { public: KthSmallest(int N) { data = std::vector(N + 1, T(0)); } // 0-indexでi番目の要素にvalを加える void add(int index, T val) { // 内部的には1-indexで扱いたいので1足す index++; int N = data.size(); for (int i = index; i <= N; i += i & -i) { data[i] += val; } } // 1-indexで昇順に並べたときのk番目の要素を返す T kth(int k) const { int N = data.size(); assert(1 <= k && k <= N); int x = 0; int init = 1; while (2 * init <= N) { init *= 2; } for (int i = init; i > 0; i /= 2) { if (x + i <= N && data[x + i] < k) { k -= data[x + i]; x += i; } } return x + 1; } private: std::vector data; }; #endif #line 1 "/workspaces/compro/lib/template.hpp" #line 3 "/workspaces/compro/lib/io/vector.hpp" #ifndef IO_VECTOR #define IO_VECTOR template std::ostream &operator<<(std::ostream &out, const std::vector &v) { int size = v.size(); for (int i = 0; i < size; i++) { std::cout << v[i]; if (i != size - 1) std::cout << " "; } return out; } template std::istream &operator>>(std::istream &in, std::vector &v) { for (auto &el : v) { std::cin >> el; } return in; } #endif #line 4 "/workspaces/compro/lib/template.hpp" #include #define REP(i, n) for (int i = 0; i < n; i++) #define FOR(i, m, n) for (int i = m; i < n; i++) #define ALL(v) (v).begin(), (v).end() #define coutd(n) cout << fixed << setprecision(n) #define ll long long int #define vl vector #define vi vector #define MM << " " << using namespace std; template void chmin(T &a, T b) { if (a > b) a = b; } template void chmax(T &a, T b) { if (a < b) a = b; } // 重複を消す。計算量はO(NlogN) template void unique(std::vector &v) { std::sort(v.begin(), v.end()); v.erase(std::unique(v.begin(), v.end()), v.end()); } #line 5 "/workspaces/compro/lib/utils/compress.hpp" #ifndef COMPRO_COMPRESS #define COMPRO_COMPRESS template class Compress { public: explicit Compress(const std::vector &argv) { isBuilt = false; add(argv); } Compress() { isBuilt = false; } void add(const std::vector &argv) { for (const auto &x : argv) { add(x); } } void add(const T x) { assert(!isBuilt); v.emplace_back(x); } void build() { isBuilt = true; std::sort(v.begin(), v.end()); v.erase(unique(v.begin(), v.end()), v.end()); } int get(const T x) const { assert(isBuilt); return lower_bound(v.begin(), v.end(), x) - v.begin(); } const T &operator[](const int k) const { return v[k]; } int size() const { return v.size(); } private: std::vector v; bool isBuilt; }; #endif #line 4 "main.cpp" long long solve(int N, const std::vector &a) { Compress cmp(a); cmp.build(); KthSmallest s(N); ll ans = 0; int right = 0; REP(left, N) { auto judge = [&]() { if (right - left < 2) return true; s.add(cmp.get(a[right]), 1); int m1 = cmp[s.kth(1) - 1]; int m2 = cmp[s.kth(2) - 1]; int M = cmp[s.kth(right + 1 - left) - 1]; s.add(cmp.get(a[right]), -1); return M <= m1 + m2; }; while (right < N && judge()) { s.add(cmp.get(a[right]), 1); right++; } ans += right - left - 1; s.add(cmp.get(a[left]), -1); } return ans; } // generated by online-judge-template-generator v4.7.1 (https://github.com/online-judge-tools/template-generator) int main() { int N; std::cin >> N; std::vector a(N); for (int i = 0; i < N; ++i) { std::cin >> a[i]; } auto ans = solve(N, a); std::cout << ans << std::endl; return 0; }