Morass

Hotel Booking - or something

Mar 1st, 2016
136
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.69 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. #define PB push_back
  4. #define ZERO (1e-10)
  5. #define INF (1<<29)
  6. #define CL(A,I) (memset(A,I,sizeof(A)))
  7. #define DEB printf("DEB!\n");
  8. #define D(X) cout<<"  "<<#X": "<<X<<endl;
  9. #define EQ(A,B) (A+ZERO>B&&A-ZERO<B)
  10. typedef long long ll;
  11. typedef long double ld;
  12. typedef pair<ll,ll> pll;
  13. typedef vector<int> vi;
  14. typedef pair<int,int> ii;
  15. typedef vector<ii> vii;
  16. #define IN(n) int n;scanf("%d",&n);
  17. #define FOR(i, m, n) for (int i(m); i < n; i++)
  18. #define REP(i, n) FOR(i, 0, n)
  19. #define F(n) REP(i, n)
  20. #define FF(n) REP(j, n)
  21. #define FT(m, n) FOR(k, m, n)
  22. #define aa first
  23. #define bb second
  24. void ga(int N,int *A){F(N)scanf("%d",A+i);}
  25. #define MX (1<<14)
  26. #define MT (601)
  27. int cn[MX][MT],H[MX];
  28. struct eg{
  29.     int v,t,m;
  30.     bool operator<(const eg&r)const{return v>r.v;}
  31. };
  32. vector<int> g[MX],v[MX];
  33. int dj(int f,int t){
  34.     priority_queue<eg> q;
  35.     q.push({0,f,0});
  36.     CL(cn,-1);
  37.     while(!q.empty()){
  38.         int a(q.top().t),s(q.top().m),w(q.top().v);
  39.         cn[a][s]=w;
  40.         if(a==t)return w;
  41.         F((ll)g[a].size())if(v[a][i]+s<=600)
  42.             q.push({w,g[a][i],v[a][i]+s});
  43.         if(H[a])F((ll)g[a].size())
  44.             q.push({w+1,g[a][i],v[a][i]});
  45.         while(!q.empty()&&~cn[q.top().t][q.top().m])
  46.             q.pop();
  47.     }
  48.     return -1;
  49. }
  50. #define ADD(A,B,V) (g[A].push_back(B),g[B].push_back(A),v[A].push_back(V),v[B].push_back(V))
  51. int N,a,b,c;
  52. int main(void){
  53.     while(scanf("%d",&N),N){
  54.         CL(H,0);
  55.         IN(h)F(h)scanf("%d",&a),H[--a]=1;
  56.         IN(Q)F(Q)scanf("%d%d%d",&a,&b,&c),--a,--b,ADD(a,b,c);
  57.         printf("%d\n",dj(0,N-1));
  58.         F(N)g[i].clear(),v[i].clear();
  59.     }
  60.     return 0;
  61. }
Advertisement
Add Comment
Please, Sign In to add comment