이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include <fstream>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
using namespace std;
template<class T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
struct custom_hash {static uint64_t splitmix64(uint64_t x) {x += 0x9e3779b97f4a7c15;x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;x = (x ^ (x >> 27)) * 0x94d049bb133111eb;return x ^ (x >> 31);}size_t operator()(uint64_t x) const {static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count();return splitmix64(x + FIXED_RANDOM);}};
// string __fname = ""; ifstream in (__fname + ".in"); ofstream out (__fname + ".out");
// #define cin in
// #define cout out
#define int64 long long
#define uint64 unsigned long long
#define x first
#define y second
#define pb push_back
#define pii pair <int, int>
#define pii64 pair <int64, int64>
#define db(x) cout << "> " << #x << ": " << (x) << "\n"
#define qr queries()
#define yn(x) if (x) {ctn("Yes");}else {ctn("No");}
void solve(int);
void queries(){int n;cin >> n;for (int i = 1; i <= n; i++) solve(i);}
template<class T>T ceildiv(T a, T b) {return a / b + !!(a % b);}
template<class T>T gcd (T a, T b){return (b ? gcd(b, a % b): a);}
template<class T>T lcm (T a, T b){return a * b / gcd(a, b);}
// // // // // // // // // // // // // // // // // // // // // //
/* TEMPLATE - VANILLA */
// // // // // // // // // // // // // // // // // // // // // //
const int ddx[] = {-1, -1, 0, 1, 1, 1, 0, -1};
const int ddy[] = {0, 1, 1, 1, 0, -1, -1, -1};
const int dx[] = {-1, 0, 1, 0};
const int dy[] = {0, 1, 0, -1};
const double pi = 3.14159265359;
const double eps = 1e-6;
const int64 hash_inv = 940594066;
const int64 hash_p = 101;
const int64 mod = 1e9 + 7;
const int maxn = 1e5 + 2;
const int maxk = 201;
int64 dp[maxn][maxk]; // dp[i][j] -> scor maxim folosind primele i elemente repartizate folosind j splits si ultimul split fiind la elementul i
int64 pref [maxn];
int64 dad[maxn][maxk];
void solve(int id){
return;
}
int64 calc (int64 i1, int64 i2, int64 i3) { // [i1;i3] -> [i1; i2 - 1] + [i2; i3]
return (pref[i3] - pref[i2 - 1]) * (pref[i2 - 1] - pref[i1 - 1]);
}
int main(){
ios_base::sync_with_stdio(0); cin.tie(0); cout << fixed; cout << setprecision(10);
int n,k;
cin >> n >> k;
for (int i = 1; i <= n; i++){
int x;
cin >> x;
pref[i] = pref[i-1] + x;
}
if (n == 1) {
cout << "0\n";
return 0;
}
for (int i = 0; i <= n; i++) for (int j = 1; j <= k; j++) dp[i][j] = -1e15;
for (int t = 1; t <= k; t++){
for (int i = 1; i < n; i++){
for (int j = 1; j <= i; j++){ // [j; n] -> [j; i] + [i + 1; n]
int64 nc = dp[j-1][t-1] + calc(j, i + 1, n);
if (nc > dp[i][t]) {
dp[i][t] = nc;
dad[i][t] = j-1;
}
}
}
}
int64 mx = -1, mxi = 0;
for (int i = 1; i <= n; i++){
if (dp[i][k] > mx) {
mx = dp[i][k];
mxi = i;
}
}
vector <int> v;
v.push_back(mxi);
while ((mxi = dad[mxi][k--]) > 0) {
v.push_back(mxi);
}
cout << mx << "\n";
reverse(v.begin(), v.end());
for (int i: v) {
cout << i << " ";
}
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |