/*under Implementation K回の試行で閉じたサイクルにたどり着かない場合があり、これをO(N)で計算する実装ができてない */ #include #include #define REP(i,n) for(int i=0,i##_len=(n);i P; class node_type { public: int v; bool s; node_type(const int v, const bool s) : v(v), s(s) {} }; ll N,K; vector p; vector

incycle;//(何番目の閉じたサイクル内にいるか,サイクルの何処に位置するか) vector

tocycle;//(閉じたサイクルのどの数字に最初に当たるか、そこまでの長さ) vector tocnt;//閉じたサイクルに行くまでの手数 vector> cycle;//(サイクルのk番目のP上での位置、スタート地点から何個P同士の境界を越えたか) vector outcycle; vector

ans; vector> tree; std::vector LA; std::vector path; std::vector stuck; void rsz(){ p.resize(N); incycle.resize(N); tocycle.resize(N); tocnt.resize(N); outcycle.resize(N,false); ans.resize(N); tree.resize(N); LA.resize(N,-1); stuck.reserve(N); path.resize(N); } void cycle_detection(){ vector used(N,false); vector prev(N,-1); int k=1; REP(i,N){ if(used[i]||incycle[i].first!=-1||prev[i]!=-1) continue; int j=i,pre=-1; bool out=false; while(!used[j]){ used[j]=true; prev[j]=pre; pre=j; j+=p[j]; j%=N; if(incycle[j].first!=-1||outcycle[j]){ out=true; break; } } if(out) {//サイクルに合流する数字は別に計算する for(int v=pre;v!=-1;v=prev[v]){ outcycle[v]=true; } continue; } cycle.resize(k+1); int u = pre, t = 0, num = 0; while(u!=j){ u=prev[u]; } for(int v=prev[u];v!=-1;v=prev[v]){ outcycle[v]=true; } do{ cycle[k].push_back(P(u,t)); incycle[u] = P(k,num); u+=p[u]; if(u>=N){ u-=N; t++; } num++; }while(u!=j); k++; } } P outcycle_calculate(int i){ if(incycle[i].first!=-1){ return P(i,0); } if(tocycle[i].first!=-1){ return tocycle[i]; } P res=outcycle_calculate((i+p[i])%N); tocnt[i]=tocnt[(i+p[i])%N]+1; tree[(i+p[i])%N].push_back(i); return tocycle[i]=P(res.first,res.second+p[i]); } P Ans_calculate(int start){ ll res=start; ll remain=K; if(incycle[res].first==-1){ remain-=tocnt[res]; res+=tocycle[res].second; } if(remain>0){ int C = incycle[res%N].first, num=incycle[res%N].second; ll LtoF = (cycle[C].back().first>=cycle[C][0].first); ll clen = N*(cycle[C].back().second + LtoF ); ll csz = cycle[C].size(); res += clen*(remain/csz); remain%=csz; int last=(num+remain)%csz; if(last>N>>K; rsz(); REP(i,N){ cin>>p[i]; incycle[i] = P(-1,-1); tocycle[i] = P(-1,-1); if(p[i]==N){ incycle[i] = P(0,0); cycle.push_back({P(i,0)}); } } cycle_detection(); //REP(i,cycle.size()) REP(j,cycle[i].size()) cerr<