#pragma GCC optimize("O3,unroll-loops")
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
// Odłączenie synchronizacji dla maksymalnej wydajności I/O
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n, m, k;
if (!(cin >> n >> m >> k)) return 0;
// dp[j] przechowuje maksymalny rozmiar dla dokładnie j zjedzonych naleśników
// Inicjalizujemy -1 jako stan niemożliwy do osiągnięcia
vector<long long> dp(k + 1, -1);
dp[0] = 0;
int current_items = 0;
// Tablica pomocnicza na sumy prefiksowe stosu
vector<long long> pref(m + 1, 0);
for (int i = 0; i < n; ++i) {
// Wczytywanie naleśników z danego stosu i liczenie sum prefiksowych
for (int x = 1; x <= m; ++x) {
long long val;
cin >> val;
pref[x] = pref[x - 1] + val;
}
// Limit optymalizuje pusty obieg pętli
int limit = min(k, current_items);
// Iterujemy OD TYŁU, aby nie nadpisać wyników z bieżącego stosu
for (int j = limit; j >= 0; --j) {
if (dp[j] == -1) continue;
long long base_val = dp[j];
int max_x = min(m, k - j);
for (int x = 1; x <= max_x; ++x) {
long long new_val = base_val + pref[x];
if (new_val > dp[j + x]) {
dp[j + x] = new_val;
}
}
}
current_items = min(k, current_items + m);
}
// Wypisujemy wynik dla dokładnie k naleśników
cout << dp[k] << "\n";
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 45 46 47 48 49 50 51 52 53 54 55 56 57 | #pragma GCC optimize("O3,unroll-loops") #include <iostream> #include <vector> #include <algorithm> using namespace std; int main() { // Odłączenie synchronizacji dla maksymalnej wydajności I/O ios_base::sync_with_stdio(false); cin.tie(NULL); int n, m, k; if (!(cin >> n >> m >> k)) return 0; // dp[j] przechowuje maksymalny rozmiar dla dokładnie j zjedzonych naleśników // Inicjalizujemy -1 jako stan niemożliwy do osiągnięcia vector<long long> dp(k + 1, -1); dp[0] = 0; int current_items = 0; // Tablica pomocnicza na sumy prefiksowe stosu vector<long long> pref(m + 1, 0); for (int i = 0; i < n; ++i) { // Wczytywanie naleśników z danego stosu i liczenie sum prefiksowych for (int x = 1; x <= m; ++x) { long long val; cin >> val; pref[x] = pref[x - 1] + val; } // Limit optymalizuje pusty obieg pętli int limit = min(k, current_items); // Iterujemy OD TYŁU, aby nie nadpisać wyników z bieżącego stosu for (int j = limit; j >= 0; --j) { if (dp[j] == -1) continue; long long base_val = dp[j]; int max_x = min(m, k - j); for (int x = 1; x <= max_x; ++x) { long long new_val = base_val + pref[x]; if (new_val > dp[j + x]) { dp[j + x] = new_val; } } } current_items = min(k, current_items + m); } // Wypisujemy wynik dla dokładnie k naleśników cout << dp[k] << "\n"; return 0; } |
English