結果
問題 | No.788 トラックの移動 |
ユーザー |
|
提出日時 | 2022-03-18 15:11:20 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 420 ms / 2,000 ms |
コード長 | 2,736 bytes |
コンパイル時間 | 2,395 ms |
コンパイル使用メモリ | 209,364 KB |
最終ジャッジ日時 | 2025-01-28 10:07:43 |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 14 |
ソースコード
#include<bits/stdc++.h>using namespace std;#define rep(i,n) for(ll i=0;i<n;i++)#define repl(i,l,r) for(ll i=(l);i<(r);i++)#define per(i,n) for(ll i=(n)-1;i>=0;i--)#define perl(i,r,l) for(ll i=r-1;i>=l;i--)#define fi first#define se second#define pb push_back#define ins insert#define pqueue(x) priority_queue<x,vector<x>,greater<x>>#define all(x) (x).begin(),(x).end()#define CST(x) cout<<fixed<<setprecision(x)#define vtpl(x,y,z) vector<tuple<x,y,z>>#define rev(x) reverse(x);using ll=long long;using vl=vector<ll>;using vvl=vector<vector<ll>>;using pl=pair<ll,ll>;using vpl=vector<pl>;using vvpl=vector<vpl>;const ll MOD=1000000007;const ll MOD9=998244353;const int inf=1e9+10;const ll INF=4e18;const ll dy[9]={0,-1,0,1,1,1,-1,-1,0};const ll dx[9]={-1,0,1,0,1,-1,1,-1,0};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 Edge{ll to,cost;};template<typename T>struct Dijkstra{vector<T> dist;vector<int> prev;Dijkstra(vector<vector<Edge>> &g,int s){dist=vector<T>(g.size(),numeric_limits<T>::max()/5);prev=vector<int>(g.size(),-1);using pi=pair<T,int>;priority_queue<pi,vector<pi>,greater<pi>> que;dist[s]=0;que.emplace(dist[s],s);while(!que.empty()){T cost;int idx;tie(cost,idx)=que.top();que.pop();if(dist[idx]<cost)continue;for(auto &e:g[idx]){T ncost=cost+e.cost;if(dist[e.to]<=ncost)continue;prev[e.to]=idx;dist[e.to]=ncost;que.emplace(dist[e.to],e.to);}}}vector<int> get_path(int t){//到達できない場合、return=t;vector<int> path;for (int cur = t; cur != -1; cur = prev[cur]) {path.push_back(cur);}reverse(path.begin(), path.end());return path;}};int main(){ll n,m,l;cin >> n >> m >> l;l--;vl t(n);rep(i,n)cin >> t[i];vector<vector<Edge>> g(n);rep(i,m){ll a,b,c;cin >> a >> b >> c;a--;b--;g[a].push_back({b,c});g[b].push_back({a,c});}ll ans=INF;Dijkstra<ll> dl(g,l);rep(i,n){Dijkstra<ll> dk(g,i);ll ret=0;rep(j,n)ret+=dk.dist[j]*t[j]*2;chmin(ans,ret+dl.dist[i]);rep(j,n){//最初に立ち寄る頂点if(!t[j])chmin(ans,ret+dl.dist[j]);else chmin(ans,ret-dk.dist[j]+dl.dist[j]);}}cout << ans << endl;}