I was solving Book Shop of CSES DP section.
- I am getting TLE even after multiple attempts, Can you tell me how to optimize my TOP-DOWN approach.
- I have also replace 2-d array with 2-d vector but that is also getting TLE.
my code
# | User | Rating |
---|---|---|
1 | jiangly | 3898 |
2 | tourist | 3840 |
3 | orzdevinwang | 3706 |
4 | ksun48 | 3691 |
5 | jqdai0815 | 3682 |
6 | ecnerwala | 3525 |
7 | gamegame | 3477 |
8 | Benq | 3468 |
9 | Ormlis | 3381 |
10 | maroonrk | 3379 |
# | User | Contrib. |
---|---|---|
1 | cry | 168 |
2 | -is-this-fft- | 165 |
3 | Dominater069 | 161 |
4 | atcoder_official | 160 |
4 | Um_nik | 160 |
6 | djm03178 | 157 |
7 | adamant | 153 |
8 | luogu_official | 151 |
9 | awoo | 149 |
10 | TheScrasse | 146 |
CSES DP SECTION — Book Shop
#include "bits/stdc++.h"
using namespace std;
int dp[1001][100001];
vector <int >price, page;
int state(int index, int total)
{
if (index < 0) return 0;
if (dp[index][total] != -1) return dp[index][total];
int &ans = dp[index][total];
if (total >= price[index]) {
ans = page[index] + state(index - 1, total - price[index]);
}
ans = max(ans, state(index - 1, total));
return ans;
}
int main() {
ios::sync_with_stdio(false); cin.tie(nullptr);
int n, req; cin >> n >> req;
price.resize(n); page.resize(n);
for (auto &x : price) cin >> x;
for (auto &x : page) cin >> x;
memset(dp, -1, sizeof(dp));
cout << state(n - 1, req);
Name |
---|