結果

問題 No.1013 〇マス進む
コンテスト
ユーザー T1610
提出日時 2020-03-20 22:48:34
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 216 ms / 2,000 ms
コード長 3,514 bytes
コンパイル時間 1,699 ms
コンパイル使用メモリ 172,596 KB
実行使用メモリ 42,752 KB
最終ジャッジ日時 2024-12-15 07:29:24
合計ジャッジ時間 9,250 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 62
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
#define rep(i,n) REP(i,0,n)
#define REP(i,s,e) for(int i=(s); i<(int)(e); i++)
#define repr(i, n) REPR(i, n, 0)
#define REPR(i, s, e) for(int i=(int)(s-1); i>=(int)(e); i--)
#define pb push_back
#define all(r) r.begin(),r.end()
#define rall(r) r.rbegin(),r.rend()
#define fi first
#define se second

typedef long long ll;
typedef vector<int> vi;
typedef vector<ll> vl;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;

const ll INF = 1e18;
const ll MOD = 1e9 + 7;
const double EPS = 1e-8;

template<typename T> T chmax(T& a, const T& b){return a = (a > b ? a : b);}
template<typename T> T chmin(T& a, const T& b){return a = (a < b ? a : b);}

#define DEBUG_MODE
#ifdef DEBUG_MODE
#define dump(x) cout << #x << " : " << x << " "
#define dumpL(x) cout << #x << " : " << x << '\n'
#define LINE cout << "line : " << __LINE__ << " "
#define LINEL cout << "line : " << __LINE__ << '\n'
#define dumpV(v) cout << #v << " : ["; for(auto& t : v) cout << t << ", "; cout<<"]" << " "
#define dumpVL(v) cout << #v << " : ["; for(auto& t : v) cout << t << ", "; cout<<"]" << endl
#define STOP assert(false)
#else
#define dump(x) 
#define dumpL(x) 
#define LINE 
#define LINEL 
#define dumpV(v) 
#define dumpVL(v) 
#define STOP assert(false)
#endif
#define mp make_pair
 
namespace std {
template<class S, class T>
ostream &operator <<(ostream& out, const pair<S, T>& a) {
    out << '(' << a.fi << ", " << a.se << ')';
    return out;
}
}

namespace Doubling {
    using NODE_TYPE = int;
    using COST_TYPE = long long;
    // d[i][0]はあらかじめ計算しておく！
    void preCalc(vector<vector<NODE_TYPE>>& d, vector<vector<NODE_TYPE>>& over) {
        int n = d.size();
        int m = d[0].size();
        REP(j, 1, m) rep(i, n) {
            int cur = d[i][j-1];
            d[i][j] = d[d[i][j-1]][j-1];
            over[i][j] = over[i][j-1] + over[cur][j-1];
        }
        
    }
    // 頂点uからcostでいける頂点を求める
    ll getNode(NODE_TYPE u, COST_TYPE cost, const vector<vector<NODE_TYPE>>& d, vector<vector<NODE_TYPE>>& over) {
        int n = d.size();
        int m = d[0].size();
        NODE_TYPE cur = u;
        ll x = 0;
        repr(j, m) {
            if(cost&(COST_TYPE(1)<<j)) {
                x += over[cur][j];
                cur = d[cur][j];
                
            }
        }
        return cur + x * n;
    }
    // 頂点uから頂点vまでのコストを求める(iはトポロジカルソート後のidx)
    COST_TYPE getCost(NODE_TYPE u, NODE_TYPE v, const vector<vector<NODE_TYPE>>& d) {
        if(u >= v) assert(false);
        int n = d.size();
        int m = d[0].size();
        int cur = u;
        COST_TYPE ret = 0;
        while(cur < v) {
            if(d[cur][0] >= v) {
                ret += 1;
                break;
            }
            int j = lower_bound(all(d[cur]), v) - d[cur].begin() - 1;
            cur = d[cur][j];
            ret+= COST_TYPE(1) << j;
        }
        return ret;
    }
};

int main(){
    cin.tie(0);
    ios::sync_with_stdio(false);
    int n, k;
    cin >> n >> k;
    vi p(n);
    rep(i, n) cin >> p[i];
    vector<vi> d(n, vi(40, -1));
    vector<vi> over(n, vi(40, -1));
    rep(i, n) {
        d[i][0] = (i+p[i]) %n;
        over[i][0] = (i >= d[i][0] ? 1 : 0);
        // dump(i);dump(d[i][0]);dumpL(over[i][0]);
    }
    Doubling::preCalc(d, over);
    rep(i, n) {
        cout << Doubling::getNode(i, k, d, over)+1 << '\n';
    }
    return 0;
}
0