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

#include <bits/stdc++.h>
using namespace std;
#define REP(i, m, n) for (int i = m; i < n; i++)
#define rep(i, n) REP(i, 0, n)
#define ALL(a) (a).begin(), (a).end()
#define RALL(a) (a).rbegin(), (a).rend()
#define dup(x, y) (((x) + (y)-1) / (y))
#define PI 3.14159265359
typedef long long LL;
const LL MOD = 1e9 + 7;
const LL LLINF = 1LL << 60;
const int INF = 1 << 30;

class UnionFind {
public:
  // rank:それぞれの木の深さ
  vector<int> rank;
  // p:ポインタ的な配列
  vector<int> p;
  UnionFind() {}
  UnionFind(int size) {
    rank.resize(size, 0);
    p.resize(size, 0);

    // 最初は、全ての値は別の木に属するようにする
    //{0},{1},{2},...といった数字一つのみを持つ木が作られる
    for (int i = 0; i < size; i++) {
      makeSet(i);
    }
  };

  // ただ一つの要素xを持つ木を作成する
  void makeSet(int x) {
    p[x] = x;
    rank[x] = 0;
  }

  // 要素xが属する木のルートノードの値を求める
  int findSet(int x) {
    // xがルートノードでないならポインタがさしている先を再帰的に探索する
    // ルートノードならreturnする
    if (x != p[x]) {
      // ここで最終的に返ってくるのはルートノードの値であるため、探索中に経由したすべてのノードのポインタはルートノードへとつながる。
      p[x] = findSet(p[x]);
    }
    return p[x];
  }

  // xとyが同じ木に属しているか判定する
  bool isSame(int x, int y) { return findSet(x) == findSet(y); }

  // xを要素に持つ木とyを要素に持つ木を統合する
  void unite(int x, int y) { link(findSet(x), findSet(y)); }

private:
  // xを根に持つ木とyを根に持つ木を統合する
  void link(int x, int y) {
    // なるべく木が深くならないように、rankが小さい方をrankが大きい方へ統合する
    if (rank[x] > rank[y]) {
      p[y] = x;
    } else {
      p[x] = y;
      // 木の深さが同じだったらどうしても一段深くなってしまう
      if (rank[x] == rank[y]) {
        rank[y]++;
      }
    }
  }
};

const int MOD_NUM = 2;
// base : 9973
class RollingHash {
  vector<LL> hash[MOD_NUM];
  vector<LL> mod = {999999937LL, 1000000007LL};
  LL base = 9973;

public:
  RollingHash(string s) { initialize(s); }

  vector<LL> getHashNum(int left, int right) {
    vector<LL> res(MOD_NUM);
    for (int i = 0; i < MOD_NUM; i++) {
      res[i] = ((hash[i][right] -
                 hash[i][left] * this->mpower(base, right - left, mod[i])) %
                    mod[i] +
                mod[i]) %
               mod[i];
    }

    return res;
  }

private:
  void initialize(string s) {
    for (int i = 0; i < MOD_NUM; i++) {
      hash[i].assign(s.size() + 1, 0);
      for (int j = 0; j < s.size(); j++) {
        hash[i][j + 1] = (hash[i][j] * base + s[j]) % mod[i];
      }
    }
  }

  LL mpower(LL x, LL n, LL mod) {
    if (n == 0) {
      return 1;
    } else if (n % 2 == 0) {
      return mpower(x * x % mod, n / 2, mod);
    } else {
      return x * mpower(x, n - 1, mod) % mod;
    }
  }
};

template <class T> inline void chmax(T &a, T b) {
  if (a < b) {
    a = b;
  }
}
template <class T> inline void chmin(T &a, T b) {
  if (a > b) {
    a = b;
  }
}
template <class T> inline T lcm(T x, T y) { return x / __gcd(x, y) * y; }
template <class T> inline void print_vector(vector<T> vec) {
  for (int i = 0; i < vec.size(); i++) {
    cout << vec[i] << " ";
  }
  cout << endl;
}
template <class T> inline T mpower(T x, T n) {
  if (n == 0) {
    return 1;
  } else if (n % 2 == 0) {
    return mpower(x * x % MOD, n / 2);
  } else {
    return x * mpower(x, n - 1) % MOD;
  }
}

template <class T> inline T q_power(T x, T n) {
  if (n == 0) {
    return 1;
  } else if (n % 2 == 0) {
    return q_power(x * x, n / 2);
  } else {
    return x * q_power(x, n - 1);
  }
}

LL mfrac(LL x) {
  LL res = 1;
  for (LL i = x; i >= 1; i--) {
    res *= i;
    res %= MOD;
  }
  return res;
}

// ノード数の上限値　問題によって変更する
#define MAX_NODE_NUM 300000

// 辺の構造体
struct edge {
  int next_node;
  int cost;
};

// O(ElogV) E:辺の数 V:頂点の数
class Dijkstra {
public:
  vector<int> reach_cost;
  vector<int> path;

  Dijkstra(int start, vector<vector<edge>> graph) {
    reach_cost.resize(MAX_NODE_NUM, INF);
    path.resize(MAX_NODE_NUM, -1);
    reach_cost[start] = 0;
    p_queue.push(pair<int, int>(0, start));
    this->graph = graph;
    search_route();
  }

  vector<int> get_reach_cost() { return reach_cost; }

  vector<int> get_path() { return path; }

private:
  // 優先度付きキューを使ってコストが一番低くなる経路をpopしながら探索する
  priority_queue<pair<int, int>, vector<pair<int, int>>,
                 greater<pair<int, int>>>
      p_queue;
  vector<vector<edge>> graph;

  void search_route() {
    while (!p_queue.empty()) {
      pair<int, int> p = p_queue.top();
      p_queue.pop();
      int node = p.second;
      if (reach_cost[node] < p.first) {
        continue;
      }

      for (int i = 0; i < graph[node].size(); i++) {
        edge e = graph[node][i];
        if (reach_cost[e.next_node] > reach_cost[node] + e.cost) {
          reach_cost[e.next_node] = reach_cost[node] + e.cost;
          path[e.next_node] = node;
          p_queue.push(pair<int, int>(reach_cost[e.next_node], e.next_node));
        }
      }
    }
  }
};

int main() {
  int N, M;
  cin >> N >> M;

  vector<vector<edge>> graph(MAX_NODE_NUM);
  for (int i = 0; i < M; i++) {
    int a, b;
    cin >> a >> b;
    a--;
    b--;

    edge e_1 = {b, 1};
    graph[a].push_back(e_1);

    edge e_2 = {a, 1};
    graph[b].push_back(e_2);
  }

  Dijkstra dijkstra(0, graph);

  vector<int> ans = dijkstra.get_path();

  cout << "Yes" << endl;
  for (int i = 1; i < N; i++) {
    printf("%d\\n", ans[i] + 1);
  }
  return 0;
}


Error: 0

Fixed submission:

#include <bits/stdc++.h>
using namespace std;
#define REP(i, m, n) for (int i = m; i < n; i++)
#define rep(i, n) REP(i, 0, n)
#define ALL(a) (a).begin(), (a).end()
#define RALL(a) (a).rbegin(), (a).rend()
#define dup(x, y) (((x) + (y)-1) / (y))
#define PI 3.14159265359
typedef long long LL;
const LL MOD = 1e9 + 7;
const LL LLINF = 1LL << 60;
const int INF = 1 << 30;

class UnionFind {
public:
  // rank:それぞれの木の深さ
  vector<int> rank;
  // p:ポインタ的な配列
  vector<int> p;
  UnionFind() {}
  UnionFind(int size) {
    rank.resize(size, 0);
    p.resize(size, 0);

    // 最初は、全ての値は別の木に属するようにする
    //{0},{1},{2},...といった数字一つのみを持つ木が作られる
    for (int i = 0; i < size; i++) {
      makeSet(i);
    }
  };

  // ただ一つの要素xを持つ木を作成する
  void makeSet(int x) {
    p[x] = x;
    rank[x] = 0;
  }

  // 要素xが属する木のルートノードの値を求める
  int findSet(int x) {
    // xがルートノードでないならポインタがさしている先を再帰的に探索する
    // ルートノードならreturnする
    if (x != p[x]) {
      // ここで最終的に返ってくるのはルートノードの値であるため、探索中に経由したすべてのノードのポインタはルートノードへとつながる。
      p[x] = findSet(p[x]);
    }
    return p[x];
  }

  // xとyが同じ木に属しているか判定する
  bool isSame(int x, int y) { return findSet(x) == findSet(y); }

  // xを要素に持つ木とyを要素に持つ木を統合する
  void unite(int x, int y) { link(findSet(x), findSet(y)); }

private:
  // xを根に持つ木とyを根に持つ木を統合する
  void link(int x, int y) {
    // なるべく木が深くならないように、rankが小さい方をrankが大きい方へ統合する
    if (rank[x] > rank[y]) {
      p[y] = x;
    } else {
      p[x] = y;
      // 木の深さが同じだったらどうしても一段深くなってしまう
      if (rank[x] == rank[y]) {
        rank[y]++;
      }
    }
  }
};

const int MOD_NUM = 2;
// base : 9973
class RollingHash {
  vector<LL> hash[MOD_NUM];
  vector<LL> mod = {999999937LL, 1000000007LL};
  LL base = 9973;

public:
  RollingHash(string s) { initialize(s); }

  vector<LL> getHashNum(int left, int right) {
    vector<LL> res(MOD_NUM);
    for (int i = 0; i < MOD_NUM; i++) {
      res[i] = ((hash[i][right] -
                 hash[i][left] * this->mpower(base, right - left, mod[i])) %
                    mod[i] +
                mod[i]) %
               mod[i];
    }

    return res;
  }

private:
  void initialize(string s) {
    for (int i = 0; i < MOD_NUM; i++) {
      hash[i].assign(s.size() + 1, 0);
      for (int j = 0; j < s.size(); j++) {
        hash[i][j + 1] = (hash[i][j] * base + s[j]) % mod[i];
      }
    }
  }

  LL mpower(LL x, LL n, LL mod) {
    if (n == 0) {
      return 1;
    } else if (n % 2 == 0) {
      return mpower(x * x % mod, n / 2, mod);
    } else {
      return x * mpower(x, n - 1, mod) % mod;
    }
  }
};

template <class T> inline void chmax(T &a, T b) {
  if (a < b) {
    a = b;
  }
}
template <class T> inline void chmin(T &a, T b) {
  if (a > b) {
    a = b;
  }
}
template <class T> inline T lcm(T x, T y) { return x / __gcd(x, y) * y; }
template <class T> inline void print_vector(vector<T> vec) {
  for (int i = 0; i < vec.size(); i++) {
    cout << vec[i] << " ";
  }
  cout << endl;
}
template <class T> inline T mpower(T x, T n) {
  if (n == 0) {
    return 1;
  } else if (n % 2 == 0) {
    return mpower(x * x % MOD, n / 2);
  } else {
    return x * mpower(x, n - 1) % MOD;
  }
}

template <class T> inline T q_power(T x, T n) {
  if (n == 0) {
    return 1;
  } else if (n % 2 == 0) {
    return q_power(x * x, n / 2);
  } else {
    return x * q_power(x, n - 1);
  }
}

LL mfrac(LL x) {
  LL res = 1;
  for (LL i = x; i >= 1; i--) {
    res *= i;
    res %= MOD;
  }
  return res;
}

// ノード数の上限値　問題によって変更する
#define MAX_NODE_NUM 300

// 辺の構造体
struct edge {
  int next_node;
  int cost;
};

// O(ElogV) E:辺の数 V:頂点の数
class Dijkstra {
public:
  vector<int> reach_cost;
  vector<int> path;

  Dijkstra(int start, vector<vector<edge>> graph) {
    reach_cost.resize(MAX_NODE_NUM, 10000);
    path.resize(MAX_NODE_NUM, 0);
    reach_cost[start] = 0;
    p_queue.push(pair<int, int>(0, start));
    this->graph = graph;
    search_route();
  }

  vector<int> get_reach_cost() { return reach_cost; }

  vector<int> get_path() { return path; }

private:
  // 優先度付きキューを使ってコストが一番低くなる経路をpopしながら探索する
  priority_queue<pair<int, int>, vector<pair<int, int>>,
                 greater<pair<int, int>>>
      p_queue;
  vector<vector<edge>> graph;

  void search_route() {
    while (!p_queue.empty()) {
      pair<int, int> p = p_queue.top();
      p_queue.pop();
      int node = p.second;
      if (reach_cost[node] < p.first) {
        continue;
      }

      for (int i = 0; i < graph[node].size(); i++) {
        edge e = graph[node][i];
        if (reach_cost[e.next_node] > reach_cost[node] + e.cost) {
          reach_cost[e.next_node] = reach_cost[node] + e.cost;
          path[e.next_node] = node;
          p_queue.push(pair<int, int>(reach_cost[e.next_node], e.next_node));
        }
      }
    }
  }
};

int main() {
  int N, M;
  cin >> N >> M;

  if (N - 1 > M) {
    cout << "No" << endl;
    return 0;
  }

  vector<vector<edge>> graph(N);
  for (int i = 0; i < M; i++) {
    int a, b;
    cin >> a >> b;
    a--;
    b--;

    edge e_1 = {b, 1};
    graph[a].push_back(e_1);

    edge e_2 = {a, 1};
    graph[b].push_back(e_2);
  }

  Dijkstra dijkstra(0, graph);

  vector<int> ans = dijkstra.get_path();

  cout << "Yes" << endl;
  for (int i = 1; i < N; i++) {
    printf("%d\\n", ans[i] + 1);
  }
  return 0;
}