Problem p02678 (C++) — buggy submission, verdict: Time Limit Exceeded

// #define _CRT_SECURE_NO_WARNINGS
#define _USE_MATH_DEFINES // M_PI=3.1415...
#include <algorithm>
#include <bitset> // bitset<8> bs1(131uL); // 10000011 bs1[0]は1 01stringからビット集合生成可
#include <cctype>
#include <climits> // A/Bを計算するときA==LLONG_MIN && B==-1のみ(1ull<<63)で代用
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <functional>
#include <iomanip> // 20桁出力 cout << setprecision(20) << double;
#include <iostream>
#include <locale>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <vector>
using namespace std;

typedef long long LL;
typedef vector<LL> VL;
typedef vector<VL> VVL; // VVL v(100,VL(100,0));
typedef pair<LL, LL> P;
typedef vector<P> VP;
typedef vector<VP> VVP;
typedef vector<string> VS;
typedef map<LL, LL> MLL;
typedef pair<LL, pair<P, P>> PP;

#define INF 999999999999999997 // 少し加算したらオーバーフローする
#define MP make_pair
#define FAST_IO                                                                \\
  cin.tie(0);                                                                  \\
  ios::sync_with_stdio(false);
#define FOR(i, a, b) for (LL i = (a); i < (b); i++)
#define FOR_REV(i, a, b) for (int i = (a); i >= (b); i--)
#define FOR_ITR(d)                                                             \\
  for (auto itr = d.begin(), d_end = d.end(); itr != d_end; ++itr)
#define O(s) cout << s << endl;

#define DUMP_VVI(b)                                                            \\
  if (dbgF) {                                                                  \\
    FOR(i, 0, b.size()) {                                                      \\
      FOR(j, 0, b[i].size()) printf("%d ", b[i][j]);                           \\
      puts("");                                                                \\
    }                                                                          \\
  }
#define D_OUT(str, value)                                                      \\
  if (dbgF) {                                                                  \\
    cout << str << " : " << value << endl;                                     \\
  }

template <class T> T IN() {
  T d;
  cin >> d;
  return d;
}
// vector<T>をソートする TはVLならLL VPならP
template <class T> void SORT(vector<T> &v) { sort(v.begin(), v.end()); }
template <class T> void SORTGL(vector<T> &v) {
  sort(v.begin(), v.end(), greater<T>());
}

// vectorの総和  第3引数をLLにしないと大きい数でオーバーフローする
LL Sum(VL v) { return accumulate(v.begin(), v.end(), 0LL); }
// 最大公約数（Greatest Common Divisor）
LL gcd(LL a, LL b) { return (b > 0) ? gcd(b, a % b) : a; }
// 最小公倍数（Least Common Multiple）
LL lcm(LL a, LL b) { return a / gcd(a, b) * b; }
// うるう年判定
bool uruu(LL Y) {
  return (((Y % 4 == 0 && Y % 100 != 0) || Y % 400 == 0) ? true : false);
}
// 文字列Sからfromを検索してすべてtoに置き換える
// from,toがcharの場合はstring(1,c)しておくこと、to_string()は使えない
string Replace(string S, string from, string to) {
  auto p = S.find(from);
  while (p != string::npos) {
    S.replace(p, from.size(), to);
    p = S.find(from, p + to.size());
  }
  return S;
}
// 桁和
LL Ketawa_s(string s) {
  LL a = 0;
  FOR(i, 0, s.length()) a += s[i] - '0';
  return a;
}
// 末尾数字で切り上げ
LL CeilLastNumber(LL n) { return ((n + (10 - 1)) / 10) * 10; }
// (x,y)がそれぞれ0以上で高さh,幅wに収まるか x,yは0-indexed、h,wは1-indexed
bool InMap(LL x, LL y, LL h, LL w) {
  if (x >= 0 && x < h && y >= 0 && y < w)
    return true;
  return false;
}
bool InMap(P p, LL h, LL w) { return InMap(p.first, p.second, h, w); }
// 二項係数 nCk mod pを求める  1<=k<=n<=10^7, p>n
const LL MAX = 510000, p = 1000000007 /*998244353*/;
LL fac[MAX], finv[MAX], inv[MAX];
/* 前処理 */ void COM_init() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  FOR(i, 2, MAX) {
    fac[i] = fac[i - 1] * i % p;
    inv[i] = p - inv[p % i] * (p / i) % p;
    finv[i] = finv[i - 1] * inv[i] % p;
  }
}
/* 主処理 */ LL COM(int n, int k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % p) % p;
}

int dx[4] = {0, 1, 0, -1};
int dy[4] = {1, 0, -1, 0};
int dxx[8] = {0, 1, 1, 1, 0, -1, -1, -1};
int dyy[8] = {1, 1, 0, -1, -1, -1, 0, 1};

// -------------------------------------------------------------------------------

// 頂点iをスタートとしたダイクストラ
VL inline Dijkstra(VVP &graph, LL i, VL &prev) {
  VL distance(graph.size(), INF);

  // Pは現在見ている確定済み頂点の(コスト, 頂点番号)
  priority_queue<P, VP, greater<P>> q;

  distance[i] = 0;
  q.push(MP(0, i));

  while (!q.empty()) {
    P nowVertex = q.top(); // この頂点のコストが確定された
    q.pop();
    LL nowVertexNumber = nowVertex.second;
    LL nowVertexCost = nowVertex.first;

    // 接続している頂点
    FOR(j, 0, graph[nowVertexNumber].size()) {
      LL nextVertexNumber = graph[nowVertexNumber][j].first;
      // 現在の頂点から次の頂点へ伸びる辺のコスト
      LL fromNowToNextCost = graph[nowVertexNumber][j].second;

      if (distance[nextVertexNumber] >
          distance[nowVertexNumber] + fromNowToNextCost) {
        // 接続先のコスト更新
        distance[nextVertexNumber] =
            distance[nowVertexNumber] + fromNowToNextCost;
        prev[nextVertexNumber] = nowVertexNumber;
        q.push(MP(distance[nextVertexNumber], nextVertexNumber));
      }
    }
  }
  return distance;
}

VL get_path(const VL &prev, LL t) {
  VL path;
  for (LL cur = t; cur != -1; cur = prev[cur]) {
    path.push_back(cur);
  }
  // reverse(path.begin(), path.end()); // 逆順なのでひっくり返す
  return path;
}

int main() {
  FAST_IO;
  bool dbgF =
      true; // for D_OUT(str, value)  ... cout<< str <<" : "<< value <<endl;

  LL N, M;
  cin >> N >> M;
  VVP graph(N);
  VL distance_0(N, INF);
  VL prev_0(N, -1);
  FOR(i, 0, M) {
    LL a, b;
    cin >> a >> b;

    a--;
    b--;
    graph[a].push_back(MP(b, 1));
    graph[b].push_back(MP(a, 1));
  }

  /*FOR(i, 0, N) {
          distance[i] = Dijkstra(graph, i, prev[i]);
  }*/
  distance_0 = Dijkstra(graph, 0, prev_0);

  VL ansList;

  FOR(i, 1, N) {
    if (distance_0[i] == INF) {
      O("No");
      return 0;
    } else {
      // VL path = get_path(prev_0, i);
      // FOR(i, 0, prev_0.size()) {
      //	cout << prev_0[i] << " ";
      // }
      // cout << endl;
      ansList.push_back(prev_0[i] + 1);
    }
  }

  // FOR(i, 1, N) {
  //	if (distance[i][0] == INF) {
  //		O("No");
  //		return 0;
  //	}
  //	else {
  //		VL path = get_path(prev[i], 0);
  //		//FOR(i, 0, path.size()) {
  //		//	cout << path[i] << " ";
  //		//}
  //		ansList.push_back(path[1] + 1);
  //		//cout << endl;
  //	}
  // }

  O("Yes");
  FOR(i, 0, ansList.size()) { O(ansList[i]); }

  return 0;
}

Error: TLE

Fixed submission:

// #define _CRT_SECURE_NO_WARNINGS
#define _USE_MATH_DEFINES // M_PI=3.1415...
#include <algorithm>
#include <bitset> // bitset<8> bs1(131uL); // 10000011 bs1[0]は1 01stringからビット集合生成可
#include <cctype>
#include <climits> // A/Bを計算するときA==LLONG_MIN && B==-1のみ(1ull<<63)で代用
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <functional>
#include <iomanip> // 20桁出力 cout << setprecision(20) << double;
#include <iostream>
#include <locale>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <vector>
using namespace std;

typedef long long LL;
typedef vector<LL> VL;
typedef vector<VL> VVL; // VVL v(100,VL(100,0));
typedef pair<LL, LL> P;
typedef vector<P> VP;
typedef vector<VP> VVP;
typedef vector<string> VS;
typedef map<LL, LL> MLL;
typedef pair<LL, pair<P, P>> PP;

#define INF 999999999999999997 // 少し加算したらオーバーフローする
#define MP make_pair
#define FAST_IO                                                                \\
  cin.tie(0);                                                                  \\
  ios::sync_with_stdio(false);
#define FOR(i, a, b) for (LL i = (a); i < (b); i++)
#define FOR_REV(i, a, b) for (int i = (a); i >= (b); i--)
#define FOR_ITR(d)                                                             \\
  for (auto itr = d.begin(), d_end = d.end(); itr != d_end; ++itr)
#define O(s) cout << s << endl;

#define DUMP_VVI(b)                                                            \\
  if (dbgF) {                                                                  \\
    FOR(i, 0, b.size()) {                                                      \\
      FOR(j, 0, b[i].size()) printf("%d ", b[i][j]);                           \\
      puts("");                                                                \\
    }                                                                          \\
  }
#define D_OUT(str, value)                                                      \\
  if (dbgF) {                                                                  \\
    cout << str << " : " << value << endl;                                     \\
  }

template <class T> T IN() {
  T d;
  cin >> d;
  return d;
}
// vector<T>をソートする TはVLならLL VPならP
template <class T> void SORT(vector<T> &v) { sort(v.begin(), v.end()); }
template <class T> void SORTGL(vector<T> &v) {
  sort(v.begin(), v.end(), greater<T>());
}

// vectorの総和  第3引数をLLにしないと大きい数でオーバーフローする
LL Sum(VL v) { return accumulate(v.begin(), v.end(), 0LL); }
// 最大公約数（Greatest Common Divisor）
LL gcd(LL a, LL b) { return (b > 0) ? gcd(b, a % b) : a; }
// 最小公倍数（Least Common Multiple）
LL lcm(LL a, LL b) { return a / gcd(a, b) * b; }
// うるう年判定
bool uruu(LL Y) {
  return (((Y % 4 == 0 && Y % 100 != 0) || Y % 400 == 0) ? true : false);
}
// 文字列Sからfromを検索してすべてtoに置き換える
// from,toがcharの場合はstring(1,c)しておくこと、to_string()は使えない
string Replace(string S, string from, string to) {
  auto p = S.find(from);
  while (p != string::npos) {
    S.replace(p, from.size(), to);
    p = S.find(from, p + to.size());
  }
  return S;
}
// 桁和
LL Ketawa_s(string s) {
  LL a = 0;
  FOR(i, 0, s.length()) a += s[i] - '0';
  return a;
}
// 末尾数字で切り上げ
LL CeilLastNumber(LL n) { return ((n + (10 - 1)) / 10) * 10; }
// (x,y)がそれぞれ0以上で高さh,幅wに収まるか x,yは0-indexed、h,wは1-indexed
bool InMap(LL x, LL y, LL h, LL w) {
  if (x >= 0 && x < h && y >= 0 && y < w)
    return true;
  return false;
}
bool InMap(P p, LL h, LL w) { return InMap(p.first, p.second, h, w); }
// 二項係数 nCk mod pを求める  1<=k<=n<=10^7, p>n
const LL MAX = 510000, p = 1000000007 /*998244353*/;
LL fac[MAX], finv[MAX], inv[MAX];
/* 前処理 */ void COM_init() {
  fac[0] = fac[1] = 1;
  finv[0] = finv[1] = 1;
  inv[1] = 1;
  FOR(i, 2, MAX) {
    fac[i] = fac[i - 1] * i % p;
    inv[i] = p - inv[p % i] * (p / i) % p;
    finv[i] = finv[i - 1] * inv[i] % p;
  }
}
/* 主処理 */ LL COM(int n, int k) {
  if (n < k)
    return 0;
  if (n < 0 || k < 0)
    return 0;
  return fac[n] * (finv[k] * finv[n - k] % p) % p;
}

int dx[4] = {0, 1, 0, -1};
int dy[4] = {1, 0, -1, 0};
int dxx[8] = {0, 1, 1, 1, 0, -1, -1, -1};
int dyy[8] = {1, 1, 0, -1, -1, -1, 0, 1};

// -------------------------------------------------------------------------------

// 頂点iをスタートとしたダイクストラ
VL inline Dijkstra(VVP &graph, LL i, VL &prev) {
  VL distance(graph.size(), INF);

  // Pは現在見ている確定済み頂点の(コスト, 頂点番号)
  priority_queue<P, VP, greater<P>> q;

  distance[i] = 0;
  q.push(MP(0, i));

  while (!q.empty()) {
    P nowVertex = q.top(); // この頂点のコストが確定された
    q.pop();
    LL nowVertexNumber = nowVertex.second;
    LL nowVertexCost = nowVertex.first;

    // 接続している頂点
    FOR(j, 0, graph[nowVertexNumber].size()) {
      LL nextVertexNumber = graph[nowVertexNumber][j].first;
      // 現在の頂点から次の頂点へ伸びる辺のコスト
      LL fromNowToNextCost = graph[nowVertexNumber][j].second;

      if (distance[nextVertexNumber] >
          distance[nowVertexNumber] + fromNowToNextCost) {
        // 接続先のコスト更新
        distance[nextVertexNumber] =
            distance[nowVertexNumber] + fromNowToNextCost;
        prev[nextVertexNumber] = nowVertexNumber;
        q.push(MP(distance[nextVertexNumber], nextVertexNumber));
      }
    }
  }
  return distance;
}

VL get_path(const VL &prev, LL t) {
  VL path;
  for (LL cur = t; cur != -1; cur = prev[cur]) {
    path.push_back(cur);
  }
  // reverse(path.begin(), path.end()); // 逆順なのでひっくり返す
  return path;
}

int main() {
  FAST_IO;
  bool dbgF =
      true; // for D_OUT(str, value)  ... cout<< str <<" : "<< value <<endl;

  LL N, M;
  cin >> N >> M;
  VVP graph(N);
  VL distance_0(N, INF);
  VL prev_0(N, -1);
  FOR(i, 0, M) {
    LL a, b;
    cin >> a >> b;

    a--;
    b--;
    graph[a].push_back(MP(b, 1));
    graph[b].push_back(MP(a, 1));
  }

  /*FOR(i, 0, N) {
          distance[i] = Dijkstra(graph, i, prev[i]);
  }*/
  distance_0 = Dijkstra(graph, 0, prev_0);

  VL ansList;

  FOR(i, 1, N) {
    if (distance_0[i] == INF) {
      O("No");
      return 0;
    } else {
      VL path = get_path(prev_0, i);
      // FOR(i, 0, path.size()) {
      //	cout << path[i] << " ";
      // }
      // cout << endl;
      ansList.push_back(path[1] + 1);
    }
  }

  // FOR(i, 1, N) {
  //	if (distance[i][0] == INF) {
  //		O("No");
  //		return 0;
  //	}
  //	else {
  //		VL path = get_path(prev[i], 0);
  //		//FOR(i, 0, path.size()) {
  //		//	cout << path[i] << " ";
  //		//}
  //		ansList.push_back(path[1] + 1);
  //		//cout << endl;
  //	}
  // }

  O("Yes");
  FOR(i, 0, ansList.size()) { O(ansList[i]); }

  return 0;
}