Advertisement
7oSkaaa

B

May 27th, 2022
94
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.40 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4.  
  5. #define cin_2d(vec, n, m) for(int i = 0; i < n; i++) for(int j = 0; j < m && cin >> vec[i][j]; j++);
  6. #define cout_2d(vec, n, m) for(int i = 0; i < n; i++, cout << "\n") for(int j = 0; j < m && cout << vec[i][j] << " "; j++);
  7. #define cout_map(mp) for(auto& [f, s] : mp) cout << f << "  " << s << "\n";
  8. #define Time cerr << "Time Taken: " << (float)clock() / CLOCKS_PER_SEC << " Secs" << "\n";
  9. #define fixed(n) fixed << setprecision(n)
  10. #define ceil(n, m) (((n) / (m)) + ((n) % (m) ? 1 : 0))
  11. #define fill(vec, value) memset(vec, value, sizeof(vec));
  12. #define Num_of_Digits(n) ((int)log10(n) + 1)
  13. #define mod_combine(a, b, m) (((a % m) * (b % m)) % m)
  14. #define all(vec) vec.begin(), vec.end()
  15. #define rall(vec) vec.rbegin(), vec.rend()
  16. #define sz(x) int(x.size())
  17. #define debug(x) cout << #x << ": " << (x) << "\n";
  18. #define fi first
  19. #define se second
  20. #define Pair pair < int, int >
  21. #define ll long long
  22. #define ull unsigned long long
  23. #define Mod  1'000'000'007
  24. #define OO 2'000'000'000
  25. #define EPS 1e-9
  26. #define PI acos(-1)
  27.  
  28. template < typename T = int > istream& operator >> (istream &in, vector < T > &v) {
  29.     for (auto &x: v) in >> x;
  30.     return in;
  31. }
  32.  
  33. template < typename T = int > ostream& operator << (ostream &out, const vector < T > &v) {
  34.     for (const T &x: v) out << x << ' ';
  35.     return out;
  36. }
  37.  
  38. void AhMeD_HoSSaM(){
  39.     ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  40.     freopen("cals.in", "r", stdin);
  41.     /*#ifndef ONLINE_JUDGE
  42.         freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  43.     #endif*/
  44. }
  45.  
  46. struct Edge {
  47.  
  48.     int v, w;
  49.  
  50.     Edge(int V = 0, int W = 0) : v(V), w(W) {}
  51.  
  52. };
  53.  
  54. int n, m, k;
  55. vector < vector < Edge > > adj;
  56. vector < vector < ll > > dp;
  57.  
  58. ll dfs(int u, int cal){
  59.     if(cal >= k) return cal == k;
  60.     ll& ways = dp[u][cal];
  61.     if(~ways) return ways;
  62.     ways = 0;
  63.     for(auto& [v, w] : adj[u])
  64.         ways += dfs(v, cal + w);
  65.     return ways;    
  66. }
  67.  
  68. void Solve(){
  69.     cin >> n >> m >> k;
  70.     adj.resize(n + 1);
  71.     dp.assign(n + 1, vector < ll > (k + 1, 0));
  72.     for(int i = 0, u, v, w; i < m && cin >> u >> v >> w; i++){
  73.         adj[u].push_back(Edge(v, w));
  74.         adj[v].push_back(Edge(u, w));
  75.     }
  76.     cout << dfs(1, 0) << '\n';
  77. }
  78.  
  79. int main(){
  80.     AhMeD_HoSSaM();
  81.     int t = 1;
  82.     cin >> t;
  83.     while(t--)
  84.         Solve();
  85.     return 0;
  86. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement