Submission #6903664

---

Source Code Expand

#include <algorithm>
#include <iostream>
#include <vector>
#include <string>
#include <cmath>
using namespace std;

vector<int> get_binary_number(int n){
  vector<int> b;
  while(n != 0){
    b.emplace_back(n%2);
    n /=2;
  }
  return b;
}


int main() {
//  int n = 256;
//  while(n != 0){
//    cout << n%2 << endl;
//    n /=2;
//  }


  // 入力
  long long N, M, D;
  cin >> N >> M >> D;
  vector<int> a(M);
  for (int i = 0; i < M; ++i) {
    cin >> a[i];
    a[i]--;
  }

  //
  int c, lim = 33;
  int results[lim][N];
  for (int i = 0; i < N; ++i) {
    c = i;
//    printf("%d, (%d, %d)\n", c, left, right);

    for (int j = 0; j < M; ++j) {
      if(a[j] == c){
//        printf("%d, (%d -> %d)\n", a[j], c, a[j]+1);
        c = a[j]+1;

      }else if(a[j]+1 == c){
//        printf("%d, (%d -> %d)\n", a[j], c, a[j]);
        c = a[j];
      }
    }

//    cout << c+1 << endl;
    results[0][i] = c;
//    break;
  }

  // ダブリング
  for (int i = 1; i < lim; ++i) {
    for (int j = 0; j < N; ++j) {
      int k = results[i-1][j];
      results[i][j] = results[i-1][k];
//      cout << results[i][j] << endl;
    }
  }

  //
  vector<int> ans(N);
  for (int i = 0; i < N; ++i) {
    ans[i] = i;
  }

  vector<int> b = get_binary_number(D);
  for (int i = 0; i < b.size(); ++i) {
    if(b[i] == 0) continue;
//    printf("i: %d\n", i);
    for (int j = 0; j < N; ++j) {
      ans[j] = results[i][ans[j]];
    }
  }

  for (int a: ans) {
    cout << a+1 << endl;
  }

  return 0;
}

Submission Info

Judge Result