verify/AOJ-GRL_1_A.test.cpp

View this file on GitHub
Last update: 2022-12-30 04:33:07+09:00
Problem: https://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=GRL_1_A
Link: View error logs on GitHub Actions

Depends on

Code

#define PROBLEM "https://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=GRL_1_A"

#include "../template/template.hpp"
#include "../graph/dijkstra-with-cost.hpp"

int main(){
  int N, M, r;
  cin >> N >> M >> r;
  CostGraph G(N);
  REP(i,M){
    int s, t , d;
    cin >> s >> t >> d;
    G[s].push_back({t, d});
  }

  auto dist = dijkstraWithCost(G, r);
  REP(i,N) {
    if(dist[i] == LLINF) cout << "INF" << endl;
    else cout << dist[i] << endl;
  }
}

#line 1 "verify/AOJ-GRL_1_A.test.cpp"
#define PROBLEM "https://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=GRL_1_A"

#line 1 "template/template.hpp"
#include<bits/stdc++.h>
using ll = long long;
#define REP(i, n) for(ll i = 0; (i) < ll(n); ++ (i))
#define FOR(i, m, n) for(ll i = (m); (i) <= ll(n); ++ (i))
#define REPR(i, n) for(ll i = ll(n) - 1; (i) >= 0; -- (i))
#define FORR(i, m, n) for(ll i = ll(n); (i) >= ll(m); -- (i))
#define ALL(x) x.begin(),x.end()

#define INF (int)1e9
#define LLINF (long long)1e18
#define MOD (int)(1e9+7)
#define MOD9 (int)998244353
#define PI 3.141592653589
#define PB push_back
#define F first
#define S second

#define YESNO(T) if(T){cout<<"YES"<<endl;}else{cout<<"NO"<<endl;}
#define yesno(T) if(T){cout<<"yes"<<endl;}else{cout<<"no"<<endl;}
#define YesNo(T) if(T){cout<<"Yes"<<endl;}else{cout<<"No"<<endl;}
#define Yes(T) {cout<<"Yes"<<endl; if(T) return 0;}
#define No(T) {cout <<"No"<<endl; if(T) return 0;}
#define YES(T) {cout<<"YES"<<endl; if(T) return 0;}
#define NO(T) {cout <<"NO"<<endl; if(T) return 0;}

#define Graph vector<vector<int> >
#define CostGraph vector<vector<pair<int,ll> > >
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define VI vector<int>
#define VL vector<ll>
#define VVI vector<vector<int> >
#define VVL vector<vector<ll> >
#define VPII vector<pair<int,int> >
#define VPLL vector<pair<ll,ll> >

#define DDD fixed<<setprecision(10)
#define PAD setfill('0')<<right<<setw(8)

template <class T>
inline bool chmin(T &a, T b) {
  if(a > b){ a = b; return true;}
  return false;
}
template <class T>
inline bool chmax(T &a, T b) {
  if(a < b){a = b; return true;}
  return false;
}
struct input{
  int n;
  input() {}
  input(int n_) : n(n_){};
  template <class T>
  operator T(){
    T ret;
    std::cin >> ret;
    return ret;
  }
  template <class T>
  operator std::vector<T>() {
    std::vector<T> ret(n);
    REP(i,n) std::cin >> ret[i];
    return ret;
  }
};
template <class T>
inline void printVec(std::vector<T> v){
  REP(i,v.size()){
    if(i) std::cout << " ";
    std::cout << v[i];
  } std::cout << std::endl;
}

using namespace std;
#line 1 "graph/dijkstra-with-cost.hpp"
vector<ll> dijkstraWithCost(CostGraph& G, int s){
  int n = G.size();
  vector<ll> dist(n, LLINF);
  dist[s] = 0;
  priority_queue<pair<ll,ll>, vector<pair<ll,ll>>, greater<pair<ll,ll>>> pq;
  pq.emplace(0, s);
  while(!pq.empty()) {
    auto [nowCost, now] = pq.top(); pq.pop();
    if(dist[now] < nowCost) continue;
    for(auto [to, cost]: G[now]) {
      if(chmin(dist[to], nowCost + cost)) pq.emplace(nowCost + cost, to);
    }
  }
  return dist;
}
#line 5 "verify/AOJ-GRL_1_A.test.cpp"

int main(){
  int N, M, r;
  cin >> N >> M >> r;
  CostGraph G(N);
  REP(i,M){
    int s, t , d;
    cin >> s >> t >> d;
    G[s].push_back({t, d});
  }

  auto dist = dijkstraWithCost(G, r);
  REP(i,N) {
    if(dist[i] == LLINF) cout << "INF" << endl;
    else cout << dist[i] << endl;
  }
}