Advertisement
MinhNGUYEN2k4

Untitled

Sep 4th, 2021
888
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.57 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 5005
  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.  
  29. void readfile()
  30. {
  31.     ios_base::sync_with_stdio(false);
  32.     cin.tie(0);cout.tie(0);
  33.     if (fopen(Task".inp","r"))
  34.     {
  35.         freopen(Task".inp","r",stdin);
  36.         freopen(Task".out","w",stdout);
  37.     }
  38.     cin >> t;
  39. }
  40.  
  41. bool cmp(ii a, ii b){
  42.     return a.fi > b.fi;
  43. }
  44.  
  45. void solve(){
  46.     int n; cin >> n;
  47.     vector<ii> a(n);
  48.     for(int i=0; i<n; i++) cin >> a[i].se;
  49.     for(int i=0; i<n; i++) cin >> a[i].fi;
  50.     sort(a.begin(),a.end(),cmp);
  51.     vector<vector<int>> dp(n+1,vector<int>(n+1,0));
  52.     dp[1][1] = a[1].se;
  53.     for(int i=1; i<=n; i++){
  54.         for(int j=1; j<=i; j++){
  55.             dp[i][j] = dp[i-1][j-1] + a[i-1].se + a[i-1].fi*(j-1);
  56.             if (i>j) dp[i][j] = max(dp[i-1][j],dp[i][j]);
  57.         }
  58.     }
  59.     for(int i=1; i<=n; i++){
  60.         cout << dp[n][i] << ' ';
  61.     }
  62.     cout << endl;
  63. }
  64.  
  65. void proc()
  66. {
  67.     while (t--){
  68.         solve();
  69.     }
  70. }
  71.  
  72. signed main()
  73. {
  74.     readfile();
  75.     proc();
  76.     return 0;
  77. }
  78.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement