結果

問題 No.1013 〇マス進む
ユーザー snow39snow39
提出日時 2020-03-20 22:59:48
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,552 bytes
コンパイル時間 1,685 ms
コンパイル使用メモリ 111,296 KB
実行使用メモリ 19,968 KB
最終ジャッジ日時 2024-05-08 23:58:01
合計ジャッジ時間 10,331 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
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 3 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 3 ms
5,376 KB
testcase_06 WA -
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 4 ms
5,376 KB
testcase_10 WA -
testcase_11 AC 3 ms
5,376 KB
testcase_12 WA -
testcase_13 AC 8 ms
5,376 KB
testcase_14 AC 93 ms
11,136 KB
testcase_15 AC 162 ms
16,256 KB
testcase_16 AC 145 ms
15,232 KB
testcase_17 AC 81 ms
10,368 KB
testcase_18 AC 100 ms
11,776 KB
testcase_19 WA -
testcase_20 WA -
testcase_21 AC 73 ms
9,728 KB
testcase_22 AC 107 ms
12,160 KB
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 AC 33 ms
6,144 KB
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 AC 61 ms
8,704 KB
testcase_31 AC 123 ms
13,696 KB
testcase_32 AC 92 ms
10,880 KB
testcase_33 AC 90 ms
11,008 KB
testcase_34 AC 144 ms
15,488 KB
testcase_35 AC 141 ms
14,720 KB
testcase_36 AC 11 ms
5,376 KB
testcase_37 AC 9 ms
5,376 KB
testcase_38 AC 165 ms
16,896 KB
testcase_39 AC 50 ms
7,808 KB
testcase_40 AC 146 ms
15,232 KB
testcase_41 AC 34 ms
6,272 KB
testcase_42 AC 86 ms
10,624 KB
testcase_43 AC 54 ms
7,936 KB
testcase_44 AC 91 ms
11,008 KB
testcase_45 AC 80 ms
10,112 KB
testcase_46 AC 37 ms
6,528 KB
testcase_47 AC 84 ms
10,496 KB
testcase_48 AC 101 ms
11,904 KB
testcase_49 AC 152 ms
15,360 KB
testcase_50 AC 123 ms
13,568 KB
testcase_51 AC 110 ms
12,416 KB
testcase_52 AC 22 ms
5,376 KB
testcase_53 AC 32 ms
5,888 KB
testcase_54 AC 138 ms
14,592 KB
testcase_55 AC 39 ms
6,656 KB
testcase_56 AC 180 ms
17,780 KB
testcase_57 AC 122 ms
13,184 KB
testcase_58 AC 208 ms
19,840 KB
testcase_59 AC 208 ms
19,968 KB
testcase_60 AC 207 ms
19,840 KB
testcase_61 WA -
testcase_62 WA -
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<int> last,used,in,dist;
vector<ll> ans;

void dfs(int now,vector<ll> &v,int M,vector<ll> &sum,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[(int)v.size()-1-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,M,sum,i);
        }
    }
    v.pop_back();
}

void solve(int col,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;
    v.push_back(0);

    dfs(p,v,M,sum,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(i,nodes[i]);
  }

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

  return 0;
}
0