結果

問題 No.1013 〇マス進む
コンテスト
ユーザー bayashiko
提出日時 2021-02-11 18:51:31
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 164 ms / 2,000 ms
コード長 2,672 bytes
コンパイル時間 3,161 ms
コンパイル使用メモリ 217,132 KB
最終ジャッジ日時 2025-01-18 17:37:35
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 62
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("Ofast")
#include<bits/stdc++.h>
using namespace std;
//#include<boost/multiprecision/cpp_int.hpp>
//#include<boost/multiprecision/cpp_dec_float.hpp>
//namespace mp=boost::multiprecision;
//#define mulint mp::cpp_int
//#define mulfloat mp::cpp_dec_float_100
//#include<atcoder/all>
//using namespace atcoder;
struct __INIT{__INIT(){cin.tie(0);ios::sync_with_stdio(false);cout<<fixed<<setprecision(15);}} __init;
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))
#define INF (1<<30)
#define LINF (lint)(1LL<<56)
#define endl "\n"
#define rep(i,n) for(lint (i)=0;(i)<(n);(i)++)
#define reprev(i,n) for(lint (i)=(n-1);(i)>=0;(i)--)
#define flc(x) __builtin_popcountll(x)
#define pint pair<int,int>
#define pdouble pair<double,double>
#define plint pair<lint,lint>
#define fi first
#define se second
#define all(x) x.begin(),x.end()
#define vec vector<lint>
#define nep(x) next_permutation(all(x))
typedef long long lint;
typedef __int128_t llint;
int dx[8]={1,1,0,-1,-1,-1,0,1};
int dy[8]={0,1,1,1,0,-1,-1,-1};
const int MAX_N=3e5+5;
//vector<int> bucket[MAX_N/1000];
constexpr int MOD=1000000007;
//constexpr int MOD=998244353;

struct doubling{
    vector<vector<lint>> to,sum;
    doubling(vector<lint> in,int log){
        to.resize(log);
        rep(i,log) to[i].resize(in.size());
        rep(i,in.size()) to[0][i]=in[i];
        for(int i=1;i<log;i++) rep(j,in.size()) to[i][j]=to[i-1][to[i-1][j]];
        sum.resize(log);
        rep(i,log) sum[i].resize(in.size());
        rep(i,in.size()) sum[0][i]=i;
        for(int i=1;i<log;i++) rep(j,in.size()) sum[i][j]=sum[i-1][j]+sum[i-1][to[i-1][j]];
    }
    doubling(vector<lint> in,vector<lint> in2, int log){
        to.resize(log);
        rep(i,log) to[i].resize(in.size());
        rep(i,in.size()) to[0][i]=in[i];
        for(int i=1;i<log;i++) rep(j,in.size()) to[i][j]=to[i-1][to[i-1][j]];
        sum.resize(log);
        rep(i,log) sum[i].resize(in.size());
        rep(i,in.size()) sum[0][i]=in2[i];
        for(int i=1;i<log;i++) rep(j,in.size()) sum[i][j]=sum[i-1][j]+sum[i-1][to[i-1][j]];
    }
    lint dest(lint now,lint k){//nowのk個先
        rep(i,to.size()){
            if(k&(1LL<<i)) now=to[i][now];
        }
        return now;
    }
    lint dsum(lint now,lint k){//Σ[i=0..k-1]
        lint res=0;
        rep(i,to.size()){
            if(k&(1LL<<i)) res+=sum[i][now],now=to[i][now];
        }
        return res;
    }
};

int main(void){
    int N,K;
    cin >> N >> K;
    vector<lint> P(N);
    rep(i,N) cin >> P[i];
    vector<lint> A(N);
    rep(i,N) A[i]=(i+P[i])%N;
    doubling D(A,P,35);
    rep(i,N) cout << D.dsum(i,K)+i+1 << endl;
}
0