MAGCARI

God Of War

Nov 11th, 2022
1,154
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.01 KB | None | 0 0
  1. /*
  2.     Task    : _example
  3.     Author  : Phumipat C. [MAGCARI]
  4.     Language: C++
  5.     Created : 12 November 2022 [11:50]
  6. */
  7. #include<bits/stdc++.h>
  8. using namespace std;
  9. struct A{
  10.     int v,w;
  11. };
  12. vector<A > g[100010];
  13. struct B{
  14.     int v,d,k;
  15.     bool operator < (const A&o) const{
  16.         return d > o.d;
  17.     }
  18. };
  19. priority_queue<B > heap;
  20. int dis[100010][2];
  21. int main(){
  22.     int n,m,s,t,u,v,w,sumW = 0;
  23.     scanf("%d %d %d %d",&n,&m,&s,&t);
  24.     for(int i=1;i<=m;i++){
  25.         scanf("%d %d %d",&u,&v,&w);
  26.         g[u].push_back({v,w});
  27.         g[v].push_back({u,w});
  28.         sumW+=w;
  29.     }
  30.     for(int i=0;i<n;i++)
  31.         dis[i][0] = dis[i][1] = 1e9;
  32.     dis[s][1] = 0;
  33.     heap.push({s,0,1});
  34.     while(!heap.empty()){
  35.         auto now = heap.top();
  36.         heap.pop();
  37.         if(now.v == t && now.k == 0){
  38.             printf("%d\n",sumW-now.d);
  39.             break;
  40.         }
  41.         for(auto x:g[now.v]){
  42.             //special
  43.             if(now.k && dis[x.v][0] > now.d){
  44.                 dis[x.v][0] = now.d;
  45.                 heap.push({x.v,now.d,0});
  46.             }
  47.             //normal
  48.             if(dis[x.v][now.k] > now.d + x.w){
  49.                 dis[x.v][now.k] = now.d + x.w;
  50.                 heap.push({x.v,dis[x.v][now.k],now.k});
  51.             }
  52.         }
  53.     }
  54.     return 0;
  55. }
Advertisement
Add Comment
Please, Sign In to add comment