Advertisement
pb_jiang

luogu P2014 WA

Jan 23rd, 2023
92
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.17 KB | None | 0 0
  1. // Problem: P2014 [CTSC1997] 选课
  2. // Contest: Luogu
  3. // URL: https://www.luogu.com.cn/problem/P2014
  4. // Memory Limit: 125 MB
  5. // Time Limit: 1000 ms
  6. //
  7. // Powered by CP Editor (https://cpeditor.org)
  8.  
  9. #include <assert.h>
  10. #include <bits/stdc++.h>
  11. using namespace std;
  12. #define dbg(...) logger(#__VA_ARGS__, __VA_ARGS__)
  13. template <typename... Args> void logger(string vars, Args &&... values)
  14. {
  15.     cerr << vars << " = ";
  16.     string delim = "";
  17.     (..., (cerr << delim << values, delim = ", "));
  18.     cerr << endl;
  19. }
  20.  
  21. template <class T> inline auto vv(int m) { return vector<vector<T>>(m, vector<T>(m)); }
  22. template <class T> inline auto vv(int m, int n) { return vector<vector<T>>(m, vector<T>(n)); }
  23. template <class T, T init> inline auto vv(int m) { return vector<vector<T>>(m, vector<T>(m, init)); }
  24. template <class T, T init> inline auto vv(int m, int n) { return vector<vector<T>>(m, vector<T>(n, init)); }
  25.  
  26. template <class T> using mpq = priority_queue<T, vector<T>, greater<T>>;
  27.  
  28. using ll = long long;
  29. using pii = pair<int, int>;
  30. using vl = vector<ll>;
  31. using vi = vector<int>;
  32.  
  33. vi g[512];
  34. vi score;
  35. int n, m;
  36. vector<vector<int>> dp;
  37.  
  38. int dfs(int u, int fa, int c)
  39. {
  40.     if (dp[u][c] != INT_MIN / 2)
  41.         return dp[u][c];
  42.     dp[u][0] = 0;
  43.     dp[u][1] = score[u];
  44.     for (auto v : g[u]) {
  45.         for (int i = c - 1; i > 0; --i) {
  46.             int val = dfs(v, u, i);
  47.  
  48.             // dbg(v, u, i, val);
  49.             for (int j = c; j > i; --j) {
  50.                 int old = dp[u][j];
  51.                 int opt = dp[u][j - i] + val;
  52.                 dp[u][j] = max(dp[u][j], dp[u][j - i] + val);
  53.                 if (old != opt && u == 0 && j == m + 1) {
  54.                     fprintf(stderr, "old:%d,opt:%d[dp:%d,val:%d],u:0,v:%d,i:%d\n", old, opt, dp[u][j - i], val, v, i);
  55.                 }
  56.             }
  57.         }
  58.     }
  59.  
  60.     return dp[u][c];
  61. }
  62.  
  63. int main(int argc, char **argv)
  64. {
  65.     cin >> n >> m;
  66.     score = vi(n + 1);
  67.     for (int i = 1; i <= n; ++i) {
  68.         int k;
  69.         cin >> k >> score[i];
  70.         g[k].push_back(i);
  71.     }
  72.     dp = vv<int, INT_MIN / 2>(n + 1, m + 2);
  73.     int ans = dfs(0, -1, m + 1);
  74.     cout << ans << endl;
  75.     return 0;
  76. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement