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 | tourist | 4009 |
2 | jiangly | 3839 |
3 | Radewoosh | 3646 |
4 | jqdai0815 | 3620 |
4 | Benq | 3620 |
6 | orzdevinwang | 3612 |
7 | Geothermal | 3569 |
8 | ecnerwala | 3494 |
9 | Um_nik | 3396 |
10 | gamegame | 3386 |
# | User | Contrib. |
---|---|---|
1 | Um_nik | 164 |
2 | -is-this-fft- | 160 |
2 | maomao90 | 160 |
4 | atcoder_official | 158 |
4 | cry | 158 |
4 | awoo | 158 |
7 | adamant | 155 |
8 | nor | 154 |
9 | TheScrasse | 153 |
10 | Dominater069 | 152 |
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 |
---|