結果

問題 No.160 最短経路のうち辞書順最小
ユーザー purple_jwl
提出日時 2015-03-03 00:09:07
言語 C++11(廃止可能性あり)
(gcc 13.3.0)
結果
AC  
実行時間 15 ms / 5,000 ms
コード長 1,771 bytes
コンパイル時間 2,100 ms
コンパイル使用メモリ 168,680 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-24 01:13:27
合計ジャッジ時間 2,526 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

#define REP(i, x, n) for(int i = x; i < (int)(n); i++)
#define rep(i, n) REP(i, 0, n)
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define F first
#define S second
#define mp make_pair

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> P;

const int INF = 1 << 30;

struct State {
  int pos, dist;
  State(int p, int d): pos(p), dist(d) {}
  bool operator < (const State &stt) const {
    return stt.dist < dist;
  }
};

struct Edge {
  int to, dist;
  Edge(int t, int d): to(t), dist(d) {}
};

int N, M, S, G;
vector<Edge> edge[200];
int prevv[200];
int minDist[200];

void solve() {
  rep(i, N) {
    prevv[i] = INF;
    minDist[i] = INF;
  }

  priority_queue<State> pq;
  pq.push(State(G, 0));
  minDist[G] = 0;
  prevv[G] = -1;
  
  while(!pq.empty()) {
    State stt = pq.top();
    pq.pop();

    rep(i, edge[stt.pos].size()) {
      int to = edge[stt.pos][i].to;
      int dist = edge[stt.pos][i].dist;
      if(minDist[to] > minDist[stt.pos] + dist) {
        prevv[to] = stt.pos;
        minDist[to] = minDist[stt.pos] + dist;
        pq.push(State(to, minDist[to]));
      }
      else if(minDist[to] == minDist[stt.pos] + dist) {
        prevv[to] = min(prevv[to], stt.pos);
      }
    }
  }

  vector<int> ans;
  ans.push_back(S);
  for(int p = prevv[S]; p != G; p = prevv[p]) {
    ans.push_back(p);
  }
  ans.push_back(G);

  rep(i, ans.size()) {
    if(i) cout << ' ';
    cout << ans[i];
  }
  cout << endl;
}

int main() {
  // ios_base::sync_with_stdio(false);
  cin >> N >> M >> S >> G;
  rep(i, M) {
    int a, b, c;
    cin >> a >> b >> c;
    edge[a].push_back(Edge(b, c));
    edge[b].push_back(Edge(a, c));
  }
  solve();
  return 0;
}
0