結果

問題 No.1013 〇マス進む
ユーザー kcvlex
提出日時 2020-03-20 22:23:08
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 3,123 bytes
コンパイル時間 1,875 ms
コンパイル使用メモリ 174,728 KB
実行使用メモリ 57,316 KB
最終ジャッジ日時 2024-12-15 06:38:01
合計ジャッジ時間 6,570 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 62
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#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 std::size_t SIZE = 1e5 + 10;
mdarray<pll, 32, SIZE> dp;

int main() {
    ll n, k;
    std::cin >> n >> k;
    vec<ll> p(n);
    for (ll i = 0; i < n; i++) {
        std::cin >> p[i];
        p[i]--;
    }

    for (ll i = 0; i < n; i++) {
        ll nxt = i + p[i] + 1;
        dp[0][i] = pll(nxt % n, nxt / n);
    }
    
    for (ll i = 0; i < 32 - 1; i++) {
        for (ll j = 0; j < n; j++) {
            ll a, b, c, d;
            std::tie(a, b) = dp[i][j];
            std::tie(c, d) = dp[i][a];
            dp[i + 1][j] = pll(c, b + d);
        }
    }

    vec<pll> ans(n);
    for (ll i = 0; i < n; i++) ans[i] = pll(i, 0);
    for (ll i = 0; k; i++, k /= 2) {
        if (!(k & 1)) continue;
        auto tmp = std::move(ans);
        ans.resize(n);
        for (ll j = 0; j < n; j++) {
            ll a, b, c, d;
            std::tie(a, b) = tmp[j];
            std::tie(c, d) = dp[i][a];
            ans[j] = pll(c, b + d);
        }
    }

    for (ll i = 0; i < n; i++) {
        ll a, b;
        std::tie(a, b) = ans[i];
        std::cout << b * n + a + 1 << '\n';
    }
    return 0;
}
0