Advertisement
esraa_syam

Untitled

Aug 6th, 2023
80
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 3.13 KB | None | 0 0
  1. #include<bits/stdc++.h>
  2. #define nl "\n"
  3. #define fi first
  4. #define se second
  5. #define pi 3.14159
  6. #define ll long long
  7. #define odd(a) (a&1)
  8. #define even(a) !(a&1)
  9. #define Mod 1'000'000'007
  10. #define INF 2'000'000'000 // 2e9
  11. #define sz(x) int(x.size())
  12. #define charToInt(s) (s - '0')
  13. #define ull unsigned long long
  14. #define number_line iota(all(vec) , 1)
  15. #define all(s) s.begin(), s.end()
  16. #define rall(v) v.rbegin() , v.rend()
  17. #define fixed(n) fixed << setprecision(n)
  18. #define Num_of_Digits(n) ((int)log10(n) + 1)
  19. #define to_decimal(bin) stoll(bin, nullptr, 2)
  20. #define Ceil(n, m) (((n) / (m)) + ((n) % (m) ? 1 : 0))
  21. #define Floor(n, m) (((n) / (m)) - ((n) % (m) ? 0 : 1))
  22. #define Upper(s) transform(all(s), s.begin(), ::toupper);
  23. #define Lower(s) transform(all(s), s.begin(), ::tolower);
  24. #define cout_map(mp) for(auto& [f, s] : mp) cout << f << " " << s << "\n";
  25. // ----- bits-------
  26. #define pcnt(n) __builtin_popcount(n)
  27. #define pcntll(n) __builtin_popcountll(n)
  28. #define clz(n) __builtin_clz(n) // <---100
  29. #define clzll(n) __builtin_clzll(n)
  30. #define ctz(n) __builtin_ctz(n) // 0001---->
  31. #define ctzll(n) __builtin_ctzll(n)
  32.  
  33. using namespace std;
  34.  
  35. template < typename T = int > istream& operator >> (istream &in, vector < T > & v){
  36. for(auto & x : v) in >> x;
  37. return in;
  38. }
  39.  
  40. template < typename T = int > ostream& operator << (ostream &out, const vector < T > & v){
  41. for(const T & x : v) out << x << " ";
  42. return out;
  43. }
  44.  
  45. void esraa()
  46. {
  47. //freopen("shuffle.in", "r", stdin), freopen("shuffle.out", "w", stdout);
  48. ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  49. #ifndef ONLINE_JUDGE
  50. freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  51. #endif
  52. }
  53. vector < vector < ll > > adj;
  54. vector < ll > powers;
  55. int n , m , k;
  56. ll dp[1004][1004][1004];
  57. vector < ll > vis;
  58. // void dfs(int node , int cnt){
  59. // vis[node] = true;
  60. // if(cnt > m) return;
  61. // for(auto & child : adj[node]){
  62. // if(!vis[child]){
  63. // dfs(child , cnt + 1);
  64. // }
  65. // }
  66. // }
  67. ll rec(ll node , ll cnt , ll crd){
  68. if(crd >= k) return (crd == k ? 0 : -INF);
  69. if(cnt >= m) return (cnt == m ? 0 : -INF);
  70. //if(vis[node]) return 0;
  71. // rec case
  72. //if(sz(adj[node]) == 0) return powers[node];
  73. ll ret = -INF;
  74. for(auto & child : adj[node]){
  75. if(!vis[child]){
  76. vis[child] = 1;
  77. ret = rec(child , cnt + 1 , crd + 1) + powers[node];
  78. max(ret , rec(child , cnt + 1 , crd));
  79. }
  80. }
  81.  
  82. return ret;
  83. }
  84. void solve(){
  85. cin >> n >> m >> k;
  86. adj.assign(n + 5 , vector < ll > ());
  87. powers.assign(n , 0);
  88. cin >> powers;
  89. for(int i = 0 ; i < n - 1 ; i++){
  90. int u , v;
  91. cin >> u >> v;
  92. adj[u].push_back(v);
  93. adj[v].push_back(u);
  94. }
  95. ll maxi = -INF , ans;
  96. for(int i = 1 ; i <= n ; i++){
  97. vis.assign(n + 1 , 0);
  98. // ans = rec(i , 0 , 0);
  99. maxi = max(ans , maxi);
  100. }
  101. cout << maxi << nl;
  102.  
  103.  
  104. }
  105. int main()
  106. {
  107. esraa();
  108. int t = 1;
  109. cin >> t;
  110. //cin.ignore();
  111. while(t--)
  112. solve();
  113. return 0;
  114. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement