# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
983000 | Hugo1729 | 수열 (APIO14_sequence) | C++11 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll dp[100001][202]={0};
ll pref[100001]={0};
ll p[100001][202];
int main(){
// cin.tie(0)->sync_with_stdio(0);
int n, k; cin >> n >> k;
vector<ll> a(n);
for(int i=0;i<n;i++)cin >> a[i];
for(int i=0;i<n;i++)pref[i+1]=pref[i]+a[i];
for(int j=1;j<=k+1;j++){
for(int i=1;i<=n;i++){ //maak i = 0
for(int x=0;x<i;x++){
if(dp[x][j-1]+(pref[i]-pref[x])*(pref[n]-pref[i])=>dp[i][j]){
dp[i][j]=dp[x][j-1]+(pref[i]-pref[x])*(pref[n]-pref[i]);
p[i][j]=x;
}
}
cout << '(' << i << ' ' << j << ')' << dp[i][j] << ' ' << p[i][j] << ' ';
}
cout << '\n';
}
cout << dp[n][++k] << '\n';
int sus=p[n][k--];
while(sus>0){
cout << sus << ' ';
sus=p[sus][k--];
}
return 0;
// 7 3
// 4 1 3 4 0 2 3
}