Advertisement
MinhNGUYEN2k4

diving

Oct 2nd, 2021
1,167
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.73 KB | None | 0 0
  1. //Nguyen Huu Hoang Minh
  2. #include <bits/stdc++.h>
  3. #define sz(x) int(x.size())
  4. #define all(x) x.begin(),x.end()
  5. #define reset(x) memset(x, 0,sizeof(x))
  6. #define pb push_back
  7. #define mp make_pair
  8. #define fi first
  9. #define se second
  10. #define N 10005
  11. #define remain(x) if (x > MOD) x -= MOD
  12. #define ii pair<int, int>
  13. #define iiii pair< ii , ii >
  14. #define viiii vector< iiii >
  15. #define vi vector<int>
  16. #define vii vector< ii >
  17. #define bit(x, i) (((x) >> (i)) & 1)
  18. #define Task "test"
  19. #define int long long
  20.  
  21. using namespace std;
  22.  
  23. typedef long double ld;
  24. const int inf = 1e10;
  25. const int minf = -1e10;
  26.  
  27. int t;
  28. int n, m;
  29. vector<ii> g[N];
  30. int d[10][N];
  31. int bk, p[9];
  32. int limits;
  33.  
  34. void readfile()
  35. {
  36.     ios_base::sync_with_stdio(false);
  37.     cin.tie(0);cout.tie(0);
  38.     if (fopen(Task".inp","r"))
  39.     {
  40.         freopen(Task".inp","r",stdin);
  41.         //freopen(Task".out","w",stdout);
  42.     }
  43.     cin >> t;
  44. }
  45.  
  46. void djk(int st, int di){
  47.     d[di][st] = 0;
  48.     priority_queue<ii, vii, greater<ii>> q;
  49.     q.push(ii(d[di][st],st));
  50.     while (q.size()){
  51.         int u = q.top().se;
  52.         int du = q.top().fi;
  53.         q.pop();
  54.         if (du!=d[di][u]) continue;
  55.         for(auto i : g[u]){
  56.             int v = i.fi;
  57.             int uv = i.se;
  58.             if (d[di][v] > d[di][u]+uv){
  59.                 d[di][v] = d[di][u]+uv;
  60.                 q.push(ii(d[di][v],v));
  61.             }
  62.         }
  63.     }
  64. }
  65.  
  66. int dp[10][300];
  67.  
  68. void solve(){
  69.     cin >> n >> m;
  70.     for(int i=0; i<n; i++){
  71.         g[i].clear();
  72.     }
  73.     memset(d, 0x3f3f3f3f3f3f, sizeof d);
  74.     for(int i=1; i<=m; i++){
  75.         int u, v, z;
  76.         cin >> u >> v >> z;
  77.         g[u].pb(ii(v,z));
  78.         g[v].pb(ii(u,z));
  79.     }
  80.     cin >> bk;
  81.     for(int i=1; i<=bk; i++) cin >> p[i];
  82.     djk(0,0);
  83.     for(int i=1; i<=bk; i++) djk(p[i],i);
  84.     cin >> limits;
  85.     memset(dp, -1, sizeof dp);
  86.     int ans=0;
  87.     for(int i=1; i<=bk; i++){
  88.         int msk = (1<<(i-1));
  89.         if (d[0][p[i]] + d[i][0] <= limits){
  90.             dp[i][msk] = d[0][p[i]];
  91.             ans = 1;
  92.         }
  93.     }
  94.     int nn = (1<<bk);
  95.     for(int msk=1; msk < nn; msk++){
  96.         for(int i=1; i<=bk; i++){
  97.             if (!bit(msk,i-1)) continue;
  98.             for(int j=1; j<=bk; j++){
  99.                 if (bit(msk,j-1)) continue;
  100.                 int nmsk = msk|(1<<(j-1));
  101.                 if (dp[i][msk]!=-1 && dp[i][msk] + d[i][p[j]] + d[j][0] <= limits){
  102.                     dp[j][nmsk] = dp[i][msk] + d[i][p[j]];
  103.                     ans = max(ans,(int)__builtin_popcount(nmsk));
  104.                 }
  105.             }
  106.         }
  107.     }
  108.     cout << ans << '\n';
  109. }
  110.  
  111. void proc()
  112. {
  113.     while (t--){
  114.         solve();
  115.     }
  116. }
  117.  
  118. signed main()
  119. {
  120.     readfile();
  121.     proc();
  122.     return 0;
  123. }
  124.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement