Problem p02684 (C++) — buggy submission, verdict: Runtime Error

#include <bits/stdc++.h>
using namespace std;
//=====================================================
// マクロ定義部
//=====================================================

#define FOR(i, a, b) for (ll i = (a); i < (b); ++i)
#define FOR_EQ(i, a, b) for (ll i = (a); i <= (b); ++i)
#define rep(i, n) FOR(i, 0, n)
#define rep_eq(i, n) FOR_EQ(i, 0, n)
#define DEBUG_VARIABLE(variable) cout << #variable << ":" << variable << endl
#define INFTY 1000000000000000000 // 10^18
#define WHITE 1                   // （未訪問）
#define GRAY                                                                   \\
  2 // （訪問）未だに訪問していない頂点への辺を持っている。スタックに退避。
#define BLACK 3 // （訪問完了）未訪問への頂点への辺を持たない
//=====================================================
// 型定義部
//=====================================================

typedef unsigned long long ull;
typedef long long ll;
typedef pair<ll, ll> ll_ll;
typedef vector<ll> Vector;
typedef vector<Vector> DVector;
typedef vector<pair<ll, ll>> Vec_Pair;
template <class T> bool chmax(T &a, const T &b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <class T> bool chmin(T &a, const T &b) {
  if (b < a) {
    a = b;
    return 1;
  }
  return 0;
}
//===================================================
// 関数のプロトタイプ宣言・関数に必要な配列の宣言
//===================================================

template <class X> void printArray(X *array, ll);
template <class X> void printVector(vector<X> &v);
template <class X> void printDimention(vector<vector<X>> &dv);
template <class X> void printMap(X);
template <class X> void printVecPair(vector<X> &v);
void printDV_Pair(vector<vector<pair<ll, ll>>> &v);
void printSet(set<ll> &S);
// 繰返し自乗法
ll RepeatSquaring(ll a, ll index);
// 繰返し自乗法 MOD
ll RepeatSquaring_MOD(ll a, ll index, ll M);
// 最大公約数
ll GCD(ll, ll);
// 最小公倍数
ll LCM(ll, ll);
// 割り算した時に余りが有る時だけ、+1の商を返す。
ll div_one(ll res, ll div);

//===================================================
// グローバル
//===================================================

const int MOD = 1000000007;
// 1個上から時計周り
const int ddx[8] = {0, 1, 1, 1, 0, -1, -1, -1};
const int ddy[8] = {1, 1, 0, -1, -1, -1, 0, 1};
// 上右下左
const int dx[4] = {0, 1, 0, -1}; // dj
const int dy[4] = {-1, 0, 1, 0}; // di
static const int NIL = -1;

ll N, k, K, V, E;
Vector A;
DVector doubling;
//===================================================
// 関数
//===================================================

//===================================================
// main 問題処理部
//===================================================
int main(int argc, char const *argv[]) {
  cin.tie(0);
  ios::sync_with_stdio(false);

  cin >> N >> K;
  A.resize(N);
  rep(i, N) {
    cin >> A[i];
    A[i]--;
  }
  doubling.resize(61, Vector(N));

  rep(i, N) doubling[0][i] = A[i];

  FOR(i, 1, 61) {
    rep(j, N) { doubling[i][j] = doubling[i - 1][doubling[i - 1][j]]; }
  }

  // bitで
  ll from = 0, to;
  for (ll bit = 0; bit < 61; bit++) {
    if ((K >> bit) & 1) {
      to = doubling[bit][from];
    }
    from = to;
  }
  cout << to + 1 << endl;

  return 0;
}

//=====================================================
// 関数群
//=====================================================

template <class X> void printArray(X array[], ll n) {
  rep(i, n) {
    if (i)
      cout << " ";
    cout << array[i];
  }
  cout << endl;
}

template <class X> void printVector(vector<X> &v) {
  rep(i, (int)v.size()) {
    if (i)
      cout << " ";
    cout << v[i];
  }
  cout << endl;
}

// 二次元配列の表示
template <class X> void printDimention(vector<vector<X>> &dv) {
  rep(i, (ll)dv.size()) {
    rep(j, (ll)dv[i].size()) {
      if (j)
        cout << " ";
      cout << dv[i][j];
    }
    cout << endl;
  }
  cout << endl;
}

// mapの表示
template <class X> void printMap(X mp) {
  for (auto it : mp) {
    cout << it.first << " " << it.second << endl;
  }
}

// 内部にpairを内包したvectorの表示
template <class X> void printVecPair(vector<X> &v) {
  rep(i, (int)v.size()) { cout << v[i].first << " " << v[i].second << endl; }
  cout << endl;
}

// 内部にVec_Pairを内包したvectorの表示
void printDV_Pair(vector<vector<pair<ll, ll>>> &v) {
  rep(i, (int)v.size()) {
    cout << i << endl;
    rep(j, (int)v[i].size()) {
      cout << v[i][j].first << " " << v[i][j].second << endl;
    }
    cout << endl;
  }
  cout << endl;
}

void printSet(set<ll> &S) {
  for (set<ll>::iterator it = S.begin(); it != S.end(); it++) {
    if (it != --S.end())
      cout << *it << " ";
    else
      cout << *it << endl;
  }
}

ll RepeatSquaring(ll a, ll index) {
  if (index == 0)
    return 1;
  else if (index % 2 == 0) {
    ll tmp = RepeatSquaring(a, index / 2);
    return tmp * tmp;
  } else {
    return a * RepeatSquaring(a, index - 1);
  }
}

/**
 * MODを使用した繰返し自乗法
 * a: 基数, index: 指数, M: MOD
 */
ll RepeatSquaring_MOD(ll a, ll index, ll M) {
  // Nは基数、Pは指数、MはMOD
  if (index == 0)
    return 1;
  if (index % 2 == 0) {
    ll tmp = RepeatSquaring_MOD(a, index / 2, M);
    return (tmp * tmp) % M;
  } else {
    return (a * RepeatSquaring_MOD(a, index - 1, M)) % M;
  }
}

// 最大公約数(Greatest Common Divisor)
//  ユークリッドの互除法使用。 0・負の数に対応
ll GCD(ll a, ll b) {
  if (b == 0)
    return abs(a);
  return GCD(b, a % b);
}

// 最小公倍数(Least Common Multiple)
ll LCM(ll m, ll n) {
  // 引数に0がある場合は0を返す
  if (m == 0 || n == 0)
    return 0;

  // lcm = m * n / gcd(m, n)
  return ((m / GCD(m, n)) * n);
}

// 割り算した時に余りが有る時だけ、+1の商を返す。
ll div_one(ll res, ll div) {
  // res 割られる数、div 割る数
  return (res + div - 1) / div;
}

Error: 0

Fixed submission:

#include <bits/stdc++.h>
using namespace std;
//=====================================================
// マクロ定義部
//=====================================================

#define FOR(i, a, b) for (ll i = (a); i < (b); ++i)
#define FOR_EQ(i, a, b) for (ll i = (a); i <= (b); ++i)
#define rep(i, n) FOR(i, 0, n)
#define rep_eq(i, n) FOR_EQ(i, 0, n)
#define DEBUG_VARIABLE(variable) cout << #variable << ":" << variable << endl
#define INFTY 1000000000000
#define WHITE 1 // （未訪問）
#define GRAY                                                                   \\
  2 // （訪問）未だに訪問していない頂点への辺を持っている。スタックに退避。
#define BLACK 3 // （訪問完了）未訪問への頂点への辺を持たない
//=====================================================
// 型定義部
//=====================================================

typedef unsigned long long ull;
typedef long long ll;
typedef pair<ll, ll> ll_ll;
typedef vector<ll> Vector;
typedef vector<Vector> DVector;
typedef vector<pair<ll, ll>> Vec_Pair;
template <class T> inline bool chmax(T &a, T b) {
  if (a < b) {
    a = b;
    return true;
  }
  return false;
}
//===================================================
// 関数のプロトタイプ宣言・関数に必要な配列の宣言
//===================================================

template <class X> void printArray(X *array, ll);
template <class X> void printVector(vector<X> &v);
template <class X> void printDimention(vector<vector<X>> &dv);
template <class X> void printMap(X);
template <class X> void printVecPair(vector<X> &v);
void printDV_Pair(vector<vector<pair<ll, ll>>> &v);
void printSet(set<ll> &S);
// 繰返し自乗法
ll RepeatSquaring(ll a, ll index);
// 最大公約数
ll GCD(ll, ll);
// 最小公倍数
ll LCM(ll, ll);
// 割り算した時に余りが有る時だけ、+1の商を返す。
ll div_one(ll res, ll div);

//===================================================
// グローバル
//===================================================

const int MOD = 1000000007;
// 1個上から時計周り
const int ddx[8] = {0, 1, 1, 1, 0, -1, -1, -1};
const int ddy[8] = {1, 1, 0, -1, -1, -1, 0, 1};
// 上右下左
const int dx[4] = {0, 1, 0, -1}; // dj
const int dy[4] = {-1, 0, 1, 0}; // di
static const int NIL = -1;

ll n, N, k, K, V, E;

//===================================================
// 関数
//===================================================

//===================================================
// main 問題処理部
//===================================================
int main(int argc, char const *argv[]) {
  cin.tie(0);
  ios::sync_with_stdio(false);

  cin >> N >> K;
  Vector A(N + 1, -1);
  FOR_EQ(i, 1, N) { cin >> A[i]; }

  // 循環を調べる
  ll times = 0;
  ll nxt = 1;
  Vector cir; // 添字はワープの回数に対応する
  set<ll> record;
  while (record.find(nxt) == record.end()) {
    record.insert(nxt);
    cir.push_back(nxt);
    times++;
    nxt = A[nxt];
  }
  // printVector(cir);
  // it には循環の最初の数値のイテレータが入っている
  Vector::iterator it = find(cir.begin(), cir.end(), nxt);
  ll until_cir = it - cir.begin(); // 循環に入るまでのワープの回数
  // DEBUG_VARIABLE(times);
  // DEBUG_VARIABLE(until_cir);
  times -= until_cir;
  K -= until_cir;
  // 循環までのcirを削除
  rep(i, until_cir) { cir.erase(cir.begin()); }
  ll rem = K % times;
  cout << cir[rem] << endl;

  return 0;
}

//=====================================================
// 関数群
//=====================================================

template <class X> void printArray(X array[], ll n) {
  rep(i, n) {
    if (i)
      cout << " ";
    cout << array[i];
  }
  cout << endl;
}

template <class X> void printVector(vector<X> &v) {
  rep(i, (int)v.size()) {
    if (i)
      cout << " ";
    cout << v[i];
  }
  cout << endl;
}

// 二次元配列の表示
template <class X> void printDimention(vector<vector<X>> &dv) {
  rep(i, (ll)dv.size()) {
    rep(j, (ll)dv[i].size()) {
      if (j)
        cout << " ";
      cout << dv[i][j];
    }
    cout << endl;
  }
  cout << endl;
}

// mapの表示
template <class X> void printMap(X mp) {
  for (auto it : mp) {
    cout << it.first << " " << it.second << endl;
  }
}

// 内部にpairを内包したvectorの表示
template <class X> void printVecPair(vector<X> &v) {
  rep(i, (int)v.size()) { cout << v[i].first << " " << v[i].second << endl; }
  cout << endl;
}

// 内部にVec_Pairを内包したvectorの表示
void printDV_Pair(vector<vector<pair<ll, ll>>> &v) {
  rep(i, (int)v.size()) {
    cout << i << endl;
    rep(j, (int)v[i].size()) {
      cout << v[i][j].first << " " << v[i][j].second << endl;
    }
    cout << endl;
  }
  cout << endl;
}

void printSet(set<ll> &S) {
  for (set<ll>::iterator it = S.begin(); it != S.end(); it++) {
    if (it != --S.end())
      cout << *it << " ";
    else
      cout << *it << endl;
  }
}

ll RepeatSquaring(ll a, ll index) {
  if (index == 0)
    return 1;
  else if (index % 2 == 0) {
    ll tmp = RepeatSquaring(a, index / 2);
    return tmp * tmp;
  } else {
    return a * RepeatSquaring(a, index - 1);
  }
}

// 最大公約数(Greatest Common Divisor)
//  ユークリッドの互除法使用
ll GCD(ll a, ll b) { return b ? GCD(b, a % b) : a; }

// 最小公倍数(Least Common Multiple)
ll LCM(ll m, ll n) {
  // 引数に0がある場合は0を返す
  if (m == 0 || n == 0)
    return 0;

  // lcm = m * n / gcd(m, n)
  return ((m / GCD(m, n)) * n);
}

// 割り算した時に余りが有る時だけ、+1の商を返す。
ll div_one(ll res, ll div) {
  // res 割られる数、div 割る数
  return (res + div - 1) / div;
}