結果

問題 No.1013 〇マス進む
ユーザー snow39snow39
提出日時 2020-03-20 23:05:47
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 204 ms / 2,000 ms
コード長 3,515 bytes
コンパイル時間 1,336 ms
コンパイル使用メモリ 110,920 KB
実行使用メモリ 21,632 KB
最終ジャッジ日時 2024-05-09 00:18:00
合計ジャッジ時間 9,770 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 3 ms
5,376 KB
testcase_06 AC 3 ms
5,376 KB
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 3 ms
5,376 KB
testcase_11 AC 3 ms
5,376 KB
testcase_12 AC 3 ms
5,376 KB
testcase_13 AC 8 ms
5,376 KB
testcase_14 AC 90 ms
11,904 KB
testcase_15 AC 151 ms
17,408 KB
testcase_16 AC 137 ms
16,128 KB
testcase_17 AC 77 ms
11,008 KB
testcase_18 AC 96 ms
12,672 KB
testcase_19 AC 56 ms
8,832 KB
testcase_20 AC 183 ms
20,528 KB
testcase_21 AC 70 ms
10,368 KB
testcase_22 AC 97 ms
12,928 KB
testcase_23 AC 32 ms
6,400 KB
testcase_24 AC 188 ms
21,144 KB
testcase_25 AC 188 ms
20,736 KB
testcase_26 AC 31 ms
6,528 KB
testcase_27 AC 67 ms
9,856 KB
testcase_28 AC 32 ms
6,400 KB
testcase_29 AC 179 ms
20,224 KB
testcase_30 AC 62 ms
9,216 KB
testcase_31 AC 118 ms
14,592 KB
testcase_32 AC 88 ms
11,776 KB
testcase_33 AC 87 ms
11,648 KB
testcase_34 AC 143 ms
16,640 KB
testcase_35 AC 131 ms
16,000 KB
testcase_36 AC 10 ms
5,376 KB
testcase_37 AC 10 ms
5,376 KB
testcase_38 AC 155 ms
18,048 KB
testcase_39 AC 48 ms
7,936 KB
testcase_40 AC 138 ms
16,640 KB
testcase_41 AC 33 ms
6,528 KB
testcase_42 AC 82 ms
11,392 KB
testcase_43 AC 52 ms
8,448 KB
testcase_44 AC 90 ms
11,776 KB
testcase_45 AC 76 ms
10,624 KB
testcase_46 AC 36 ms
6,784 KB
testcase_47 AC 82 ms
11,520 KB
testcase_48 AC 99 ms
12,672 KB
testcase_49 AC 145 ms
16,512 KB
testcase_50 AC 118 ms
14,592 KB
testcase_51 AC 104 ms
13,312 KB
testcase_52 AC 21 ms
5,376 KB
testcase_53 AC 31 ms
6,272 KB
testcase_54 AC 131 ms
15,872 KB
testcase_55 AC 39 ms
6,912 KB
testcase_56 AC 173 ms
19,144 KB
testcase_57 AC 111 ms
14,080 KB
testcase_58 AC 204 ms
21,368 KB
testcase_59 AC 200 ms
21,632 KB
testcase_60 AC 202 ms
21,376 KB
testcase_61 AC 199 ms
21,376 KB
testcase_62 AC 194 ms
21,544 KB
testcase_63 AC 2 ms
5,376 KB
testcase_64 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <cmath>
#include <map>
#include <queue>
#include <iomanip>
#include <set>
#include <tuple>
#define mkp make_pair
#define mkt make_tuple
#define rep(i,n) for(int i = 0; i < (n); ++i)
using namespace std;
typedef long long ll;
const ll MOD=1e9+7;
template<class T> void chmin(T &a,const T &b){if(a>b) a=b;}
template<class T> void chmax(T &a,const T &b){if(a<b) a=b;}

class DisjointSet{
public:
  vector<int> rank,p;
  vector<int> sz;
 
  DisjointSet(){}
  DisjointSet(int size){
    rank.resize(size,0);
    p.resize(size,0);
    sz.resize(size,0);
    for(int i=0;i<size;i++) makeSet(i);
  }
 
  void makeSet(int x){
    p[x]=x;
    rank[x]=0;
    sz[x]=1;
  }
 
  bool same(int x,int y){
    return findSet(x)==findSet(y);
  }
 
  void unite(int x,int y){
    if(same(x,y)) return;
    link(findSet(x),findSet(y));
  }
 
  void link(int x,int y){
    if(rank[x]>rank[y]){
      p[y]=x;
      sz[x]+=sz[y];
    }else{
      p[x]=y;
      sz[y]+=sz[x];
      if(rank[x]==rank[y]){
        rank[y]++;
      }
    }
  }
 
  int findSet(int x){
    if(x!=p[x]){
      p[x]=findSet(p[x]);
    }
    return p[x];
  }
 
  int findSize(int x){
    return sz[findSet(x)];
  }
};

int N,K;
vector<int> P;

vector<vector<int>> g;
vector<vector<int>> rg;
vector<ll> last,used,in,dist;
vector<ll> ans;

void dfs(int now,vector<ll> &v,vector<ll> &sum,int M,int i){
    v.push_back(last[now]);
    for(auto nex:rg[now]){
        if(used[nex]){
            dist[nex]=dist[now]+1;
            last[nex]=last[now]+P[nex];

            {
                if(dist[nex]>=K){
                    ans[nex]=last[nex]-v[dist[nex]-K];
                }else{
                    ans[nex]+=last[nex];
                    ll rest=K-dist[nex];
                    ans[nex]+=sum[M]*(rest/M);
                    ans[nex]+=sum[i+rest%M]-sum[i];
                }
            }
            dfs(nex,v,sum,M,i);
        }
    }
    v.pop_back();
}

void solve(vector<int> &nodes){
  int n=nodes.size();
  queue<int> Q;
  for(auto p:nodes) if(in[p]==0) Q.push(p);

  while(!Q.empty()){
    int now=Q.front();Q.pop();
    used[now]=1;
    for(auto nex:g[now]){
      in[nex]--;
      if(in[nex]==0) Q.push(nex);
    }
  }

  vector<int> circuit;
  int st=0;
  for(auto p:nodes) if(used[p]==0) st=p;
  Q.push(st);
  while(!Q.empty()){
    int now=Q.front();
    Q.pop();
    circuit.push_back(now);
    for(auto nex:g[now]){
      if(nex!=st) Q.push(nex);
    }
  }

  int M=circuit.size();
  vector<ll> sum(2*M+1,0);
  for(int i=0;i<2*M;i++) sum[i+1]=sum[i]+P[circuit[i%M]];

  for(int i=0;i<M;i++){
    int p=circuit[i];
    dist[p]=0;
    last[p]=0;
    vector<ll> v;

    dfs(p,v,sum,M,i);
  }


  for(int i=0;i<M;i++){
    int p=circuit[i];
    ans[p]+=sum[M]*(K/M);
    ans[p]+=sum[i+K%M]-sum[i];
  }
}

int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);

  cin>>N>>K;
  P.resize(N);
  rep(i,N) cin>>P[i];

  g.resize(N);
  rg.resize(N);
  last.resize(N,0);
  used.resize(N,0);
  in.resize(N,0);
  dist.resize(N,MOD);
  ans.resize(N,0);

  DisjointSet us(N);
  for(int i=0;i<N;i++){
    int tar=(i+P[i])%N;
    us.unite(i,tar);
    g[i].push_back(tar);
    rg[tar].push_back(i);
  }

  rep(i,N) in[i]=rg[i].size();

  vector<vector<int>> nodes(N);
  for(int i=0;i<N;i++) nodes[us.findSet(i)].push_back(i);

  for(int i=0;i<N;i++){
    if(nodes[i].size()==0) continue;
    solve(nodes[i]);
  }

  for(int i=0;i<N;i++) cout<<ans[i]+i+1<<endl;

  return 0;
}
0