ABC013D 阿弥陀

D: 阿弥陀 - AtCoder Beginner Contest 013 | AtCoder

(null)

まず一回アミダをシュミレーションする.すると,ゴールする場所 に変える関数が作れる.これを遷移行列にして行列類乗だ!と思ったが, に落ちるが結果となってしまい意味が無い.
そもそも行列で持つ必要がない.ダブリングで に落として,
resをmapで作ってしまい が余計にかかる.変に をつけて落ちることがあるので気をつけたい.

Code

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#include <iostream>
#include <vector>
#include <string>
#include <cstring>
#include <algorithm>
#include <sstream>
#include <map>
#include <set>

#define REP(i,k,n) for(int i=k;i<n;i++)
#define rep(i,n) for(int i=0;i<n;i++)
#define INF 1<<30
#define pb push_back
#define mp make_pair

using namespace std;
typedef long long ll;
typedef pair<int,int> P;

int main() {
  int n, m, d;
  cin >> n >> m >> d;

  vector<int> a(m);
  rep(i, m) cin >> a[i];

  vector<int> v(n);
  rep(i, n) v[i] = i;

  rep(i, m) {
      swap(v[a[i]-1], v[a[i]]);
  }

  map<int, int> res;
  rep(i, n) {
      res[v[i]] = i;
  }

  vector<int> next(n), cur(n);
  rep(i, n) cur[i] = i;
  while(d) {
      if(d & 1) {
          rep(i, n) next[i] = res[cur[i]];
          rep(i, n) cur[i] = next[i];
      }

      rep(i, n) next[i] = res[res[i]];
      rep(i, n) res[i] = next[i];

      d /= 2;
  }

  rep(i, n) {
      cout << cur[i] + 1 << endl;
  }

  return 0;
}
Apr 3rd, 2016