#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long ll;
void solve() {
int n, m, k;
if (!(cin >> n >> m >> k)) return;
// dp[j] - maksymalna suma dla j zjedzonych nalesnikow
// Uzywamy optymalizacji pamieciowej (tylko jeden wiersz DP)
vector<ll> dp(k + 1, 0);
for (int i = 0; i < n; ++i) {
vector<ll> current_stack(m);
vector<ll> pref(m + 1, 0);
for (int j = 0; j < m; ++j) {
cin >> current_stack[j];
pref[j + 1] = pref[j] + current_stack[j];
}
// Aktualizujemy DP od tylu, aby nie uzywac wynikow z tego samego stosu
for (int j = k; j >= 0; --j) {
// x to liczba nalesnikow brana z obecnego stosu
// Musimy wziac max z obecnego stanu lub nowego (dp z poprzednich stosow + prefiks obecnego)
for (int x = 1; x <= m && x <= j; ++x) {
dp[j] = max(dp[j], dp[j - x] + pref[x]);
}
}
}
cout << dp[k] << endl;
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
solve();
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | #include <iostream> #include <vector> #include <algorithm> using namespace std; typedef long long ll; void solve() { int n, m, k; if (!(cin >> n >> m >> k)) return; // dp[j] - maksymalna suma dla j zjedzonych nalesnikow // Uzywamy optymalizacji pamieciowej (tylko jeden wiersz DP) vector<ll> dp(k + 1, 0); for (int i = 0; i < n; ++i) { vector<ll> current_stack(m); vector<ll> pref(m + 1, 0); for (int j = 0; j < m; ++j) { cin >> current_stack[j]; pref[j + 1] = pref[j] + current_stack[j]; } // Aktualizujemy DP od tylu, aby nie uzywac wynikow z tego samego stosu for (int j = k; j >= 0; --j) { // x to liczba nalesnikow brana z obecnego stosu // Musimy wziac max z obecnego stanu lub nowego (dp z poprzednich stosow + prefiks obecnego) for (int x = 1; x <= m && x <= j; ++x) { dp[j] = max(dp[j], dp[j - x] + pref[x]); } } } cout << dp[k] << endl; } int main() { ios_base::sync_with_stdio(false); cin.tie(NULL); solve(); return 0; } |
English