이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
//#define int long long
#define ll long long
#define ld long double
#define pb push_back
#define S second
#define F first
//const int inf = 1e18;
const int N = 100005;
const ll inf = 1e18;
int n, y, k, par[N][205];
ll a[N], b[N], pref[N], suma, dp[N][205];
bool cut[N][205];
vector<int> ans;
ll f(int i, int j, int m){
// cout << x << ' ' << y << ' ' << pref[x] << ' ' << suma << '\n';
// cout << i << ' ' << j << ' ' << m << '\n';
// cout << dp[m][j - 1] << ' ' << (pref[i] - pref[m]) << ' ' << (pref[n] - pref[i]) << '\n';
return dp[m][j - 1] + (pref[i] - pref[m]) * (pref[n] - pref[i]);
}
main(){
// ios_base::sync_with_stdio(0);
// cin.tie(0);
// cout.tie(0);
cin >> n >> k;
k++;
for(int i = 1; i <= n; ++i){
cin >> a[i];
pref[i] = pref[i - 1] + a[i];
}
for(int i = 0; i <= n; ++i){
for(int j = 2; j <= k; ++j){
dp[i][j] = -inf;
}
}
for(int i = 1; i <= n; ++i){
for(int j = 2; j <= min(i + 1, k); ++j){
int l = 0, r = i - 1;
while(r - l > 2){
int m1 = l + (r - l) / 3, m2 = r - (r - l) / 3;
if(f(i, j, m1) < f(i, j, m2))
l = m1;
else
r = m2;
}
// cout << l << ' ' << r << '\n';
if(f(i, j, l) >= f(i, j, l + 1)) r = l;
else if(l + 1 < i) l++;
if(l + 1 < i && f(i, j, l) >= f(i, j, l + 1)) r = l;
else if(l + 1 < i) l++;
// cout << l << ' ' << r << '\n';
dp[i][j] = f(i, j, l);
if(dp[i][j] > dp[i - 1][j]) {
cut[i][j] = 1;
par[i][j] = l;
}
else{
dp[i][j] = dp[i - 1][j];
par[i][j] = i - 1;
}
// cout << i << ' ' << j << ' ' << dp[i][j] << '\n';
}
}
cout << dp[n][k] << '\n';
int ind = n;
while(ind != 0){
if(cut[ind][k]) {
ans.pb(ind);
ind = par[ind][k];
k--;
}
else ind = par[ind][k];
}
reverse(ans.begin(), ans.end());
for(auto to : ans){
cout << to << ' ';
}
}
컴파일 시 표준 에러 (stderr) 메시지
sequence.cpp:27:2: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
27 | main(){
| ^~~~
# | 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... |