#466749 (C++17) No.1031 いたずら好きなお姉ちゃん

提出ソース

結果

問題	No.1031 いたずら好きなお姉ちゃん
ユーザー	kcvlex
提出日時	2020-04-19 00:19:17
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	316 ms / 3,500 ms
コード長	4,196 bytes
コンパイル時間	1,743 ms
コンパイル使用メモリ	154,120 KB
実行使用メモリ	29,168 KB
最終ジャッジ日時	2024-04-14 21:21:01
合計ジャッジ時間	13,748 ms
ジャッジサーバーID （参考情報）	judge4 / judge1

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,812 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	8 ms 6,940 KB
testcase_04	AC	8 ms 6,944 KB
testcase_05	AC	8 ms 6,944 KB
testcase_06	AC	8 ms 6,944 KB
testcase_07	AC	8 ms 6,944 KB
testcase_08	AC	8 ms 6,944 KB
testcase_09	AC	8 ms 6,940 KB
testcase_10	AC	8 ms 6,940 KB
testcase_11	AC	243 ms 24,500 KB
testcase_12	AC	237 ms 24,228 KB
testcase_13	AC	296 ms 28,944 KB
testcase_14	AC	256 ms 25,196 KB
testcase_15	AC	276 ms 27,400 KB
testcase_16	AC	225 ms 24,024 KB
testcase_17	AC	282 ms 27,540 KB
testcase_18	AC	267 ms 26,668 KB
testcase_19	AC	278 ms 27,744 KB
testcase_20	AC	303 ms 29,076 KB
testcase_21	AC	269 ms 26,604 KB
testcase_22	AC	228 ms 23,888 KB
testcase_23	AC	246 ms 24,592 KB
testcase_24	AC	286 ms 27,844 KB
testcase_25	AC	307 ms 28,900 KB
testcase_26	AC	271 ms 26,884 KB
testcase_27	AC	259 ms 25,440 KB
testcase_28	AC	314 ms 29,040 KB
testcase_29	AC	300 ms 29,168 KB
testcase_30	AC	309 ms 29,164 KB
testcase_31	AC	304 ms 29,044 KB
testcase_32	AC	313 ms 29,148 KB
testcase_33	AC	316 ms 28,908 KB
testcase_34	AC	193 ms 29,016 KB
testcase_35	AC	184 ms 29,012 KB
testcase_36	AC	183 ms 29,144 KB
testcase_37	AC	183 ms 29,020 KB
testcase_38	AC	171 ms 28,660 KB
testcase_39	AC	171 ms 27,972 KB
testcase_40	AC	170 ms 28,476 KB
testcase_41	AC	172 ms 28,652 KB
testcase_42	AC	170 ms 28,484 KB
testcase_43	AC	168 ms 28,556 KB
testcase_44	AC	168 ms 28,608 KB
testcase_45	AC	163 ms 27,736 KB
testcase_46	AC	178 ms 28,360 KB
testcase_47	AC	165 ms 28,132 KB
testcase_48	AC	173 ms 28,880 KB
testcase_49	AC	176 ms 28,948 KB
testcase_50	AC	174 ms 28,680 KB
testcase_51	AC	183 ms 29,040 KB
testcase_52	AC	188 ms 29,040 KB
testcase_53	AC	190 ms 28,960 KB
testcase_54	AC	2 ms 6,944 KB
testcase_55	AC	2 ms 6,944 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <limits>
#include <initializer_list>
#include <utility>
#include <bitset>
#include <tuple>
#include <type_traits>
#include <functional>
#include <string>
#include <array>
#include <deque>
#include <list>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <unordered_map>
#include <unordered_set>
#include <iterator>
#include <algorithm>
#include <complex>
#include <random>
#include <numeric>
#include <iostream>
#include <iomanip>
#include <sstream>
#include <regex>
#include <cassert>
#include <cstddef>
#define endl codeforces
#define ALL(v) std::begin(v), std::end(v)
#define ALLR(v) std::rbegin(v), std::rend(v)
using ll = std::int64_t;
using ull = std::uint64_t;
using pii = std::pair<int, int>;
using tii = std::tuple<int, int, int>;
using pll = std::pair<ll, ll>;
using tll = std::tuple<ll, ll, ll>;
template <typename T> using vec = std::vector<T>;
template <typename T> using vvec = vec<vec<T>>;
template <typename T> const T& var_min(const T &t) { return t; }
template <typename T> const T& var_max(const T &t) { return t; }
template <typename T, typename... Tail> const T& var_min(const T &t, const Tail&... tail) { return std::min(t, var_min(tail...)); }
template <typename T, typename... Tail> const T& var_max(const T &t, const Tail&... tail) { return std::max(t, var_max(tail...)); }
template <typename T, typename... Tail> void chmin(T &t, const Tail&... tail) { t = var_min(t, tail...); }
template <typename T, typename... Tail> void chmax(T &t, const Tail&... tail) { t = var_max(t, tail...); }
template <typename T> const T& clamp(const T &t, const T &low, const T &high) { return std::max(low, std::min(high, t)); }
template <typename T> void chclamp(T &t, const T &low, const T &high) { return t = clamp(t, low, high); }
template <typename T> T make_v(T init) { return init; }
template <typename T, typename... Tail> auto make_v(T init, std::size_t s, Tail... tail) { auto v = std::move(make_v(init, tail...)); return vec<decltype(v)>(s, v); }
template <typename T, std::size_t Head, std::size_t ...Tail> struct multi_dem_array { using type = std::array<typename multi_dem_array<T, Tail...>::type, Head>; };
template <typename T, std::size_t Head> struct multi_dem_array<T, Head> { using type = std::array<T, Head>; };
template <typename T, std::size_t ...Args> using mdarray = typename multi_dem_array<T, Args...>::type;
namespace init__ { struct InitIO { InitIO() { std::cin.tie(nullptr); std::ios_base::sync_with_stdio(false); std::cout << std::fixed << std::setprecision(30); } } init_io; }

const ll inf = 5e15;

vec<ll> cnt_l(vec<ll> p) {
    ll n = p.size();
    p.push_back(n);
    vec<ll> rp(n + 1);
    for (ll i = 0; i < n; i++) rp[p[i]] = i;

    auto adjust = [&] {
        std::set<ll> unused;
        vec<ll> ret(n + 1, n);
        for (ll i = 0; i <= n; i++) unused.insert(i);
        for (ll e = n - 1; 0 <= e; e--) {
            ll pos = rp[e];
            unused.erase(pos);
            auto ite = unused.lower_bound(pos);
            ret[e] = *ite;
        }
        return ret;
    }();

    auto dub = make_v<ll>(0, 20, n + 1);
    for (ll i = 0; i <= n; i++) dub[0][i] = adjust[i];
    for (ll i = 0; i + 1 < 20; i++) for (ll j = 0; j <= n; j++) {
        ll pos = dub[i][j];
        ll val = p[pos];
        dub[i + 1][j] = dub[i][val];
    }


    auto get_cnt = [&](ll pos, ll r) {
        ll ret = 0;
        for (ll i = 19; 0 <= i; i--) {
            ll mask = 1ll << i;
            ll nxt = dub[i][p[pos]];
            if (r <= nxt) continue;
            ret += mask;
            pos = nxt;
        }
        return ret;
    };
   
    vec<ll> ret(n);
    std::set<ll> used;
    used.insert(n);

    for (ll e = n - 1; 0 <= e; e--) {
        ll pos = rp[e];
        ll r = *used.lower_bound(pos);
        ret[e] = get_cnt(pos, r);
        used.insert(pos);
    }
    return ret;
}

int main() {
    ll n;
    std::cin >> n;
    vec<ll> p(n);
    for (ll &e : p) {
        std::cin >> e;
        e--;
    }

    auto lv = cnt_l(p);
    std::reverse(ALL(p));
    auto rv = cnt_l(p);
    ll ans = 0;
    for (ll i = 0; i < n; i++) ans += lv[i] + rv[i];
    std::cout << ans << '\n';
    return 0;
}

yukicoder

結果

テストケース

ソースコード