oj.uz

답안 #982979

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
982979 2024-05-15T06:23:05 Z Hugo1729 수열 (APIO14_sequence) C++11
0 / 100
 2000 ms 101696 KB 
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

ll pref[100001]={0};
ll dp[100001][202]={0};
ll p[100001][202]={0};

struct line{
    ll m,c;

    line(ll _m, ll _c){m=_m;c=_c;}

    ll intersect(line l){
        return m-l.m==0 ? LLONG_MAX/4 : (l.c-c)/(m-l.m);
    }

    ll value(ll x){
        return m*x+c;
    }
};

deque<pair<line,ll>> q;
deque<ll> pos;

void insert(ll m, ll c){
    line l(m,c);
    while(!q.empty()&&q.front().second>=l.intersect(q.front().first)){
        // cout << "irem" << ' ' << q.front().first.m << ' ' << q.front().first.c << ' ' << q.front().second << '\n';
        q.pop_front();
        pos.pop_front();
    }

    if(q.empty()){
        q.push_front({l,0});
    }else{
        q.push_front({l,l.intersect(q.front().first)});
    }
    // cout << "iadd" << ' ' << q.front().first.m << ' ' << q.front().first.c << ' ' << q.front().second << '\n';
}

ll query(ll x){
    while(q.size()>1&&q[q.size()-2].second<=x){
        // cout << "qrem" << ' ' << q.back().first.m << ' ' << q.back().first.c << ' ' << q.back().second << '\n';
        q.pop_back();
        pos.pop_back();
    }

    // cout << "query" << ' ' << q.back().first.m << ' ' << q.back().first.c << ' ' << q.back().second <<  ' ' << q.back().first.value(x) <<'\n';

    return q.back().first.value(x);
}


int main(){
    cin.tie(0)->sync_with_stdio(0);

    ll n,k; cin >> n >> k;

    vector<ll> a(n);
    for(ll i=0;i<n;i++)cin >> a[i];
    for(ll i=0;i<n;i++)pref[i+1]=pref[i]+a[i];

    for(ll j=1;j<=k+1;j++){
        // q.clear();
        // pos.clear();
        // insert(0,0);
        // pos.push_front(0);
        for(ll i=1;i<=n;i++){
            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;
                }
            }

            // dp[i][j]=query(pref[i]);
            // p[i][j]=pos.back();

            // insert(pref[i],dp[i][j-1]-pref[i]*pref[i]);
            // pos.push_front(i);
            // cout << '(' << i << j <<')' <<dp[i][j] << ' ' << p[i][j] << ' ';
        }
        // cout << '\n';
    }

    cout << dp[n][++k] << '\n';

    ll sus=n;
    vector<ll> ans;
    while(p[sus][k]>0){
        ans.push_back(p[sus][k--]);
        sus=p[sus][k];
    }

    for(ll i=ans.size()-1;i>=0;i--)cout << ans[i] << ' ';

    return 0;
// 7 3
// 4 1 3 4 0 2 3
// (11)0 (21)4 (31)16 (41)35 (51)35 (61)48 (71)72 
// (12)0 (22)4 (32)19 (42)48 (52)48 (62)64 (72)95 
// (13)0 (23)4 (33)19 (43)51 (53)51 (63)72 (73)108 
// 108
}

Subtask #1
0 / 11.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 348 KB contestant found the optimal answer: 108 == 108
2 Correct 0 ms 348 KB contestant found the optimal answer: 999 == 999
3 Correct 0 ms 348 KB contestant found the optimal answer: 0 == 0
4 Correct 1 ms 348 KB contestant found the optimal answer: 1542524 == 1542524
5 Correct 1 ms 344 KB contestant found the optimal answer: 4500000000 == 4500000000
6 Incorrect 0 ms 348 KB declared answer doesn't correspond to the split scheme: declared = 1, real = 0
7 Halted 0 ms 0 KB -

Subtask #2
0 / 11.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 1 ms 600 KB declared answer doesn't correspond to the split scheme: declared = 1093956, real = 1093884
2 Halted 0 ms 0 KB -

Subtask #3
0 / 11.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 1 ms 1112 KB declared answer doesn't correspond to the split scheme: declared = 610590000, real = 514120000
2 Halted 0 ms 0 KB -

Subtask #4
0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 4 ms 3420 KB declared answer doesn't correspond to the split scheme: declared = 21503404, real = 21475184
2 Halted 0 ms 0 KB -

Subtask #5
0 / 21.0

# 결과 실행 시간 메모리 Grader output
1 Correct 593 ms 32244 KB contestant found the optimal answer: 1818678304 == 1818678304
2 Correct 505 ms 32012 KB contestant found the optimal answer: 1326260195 == 1326260195
3 Execution timed out 2054 ms 32244 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #6
0 / 29.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 2067 ms 101696 KB Time limit exceeded
2 Halted 0 ms 0 KB -