Submission #908250

#	Time	Username	Problem	Language	Result	Execution time	Memory
908250		sofija6	Split the sequence (APIO14_sequence)	C++14	11 / 100	99 ms	17500 KiB

sequence

#include <bits/stdc++.h>
#define ll long long
#define MAXN 10010
#define MAXK 210
using namespace std;
ll a[MAXN],dp[MAXN][2],prevv[MAXN][MAXK],pref[MAXN];
struct line
{
    long double k,m;
    long double Calc(ll x)
    {
        return k*x+m;
    }
    long double Intersect(line l)
    {
        return (m-l.m)/(l.k-k);
    }
};
int main()
{
    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    ll n,k;
    cin >> n >> k;
    for (ll i=1;i<=n;i++)
    {
        cin >> a[i];
        pref[i]=pref[i-1]+a[i];
    }
    deque<pair<line,ll> > d;
    for (ll i=1;i<n;i++)
        dp[i][0]=pref[i]*(pref[n]-pref[i]);
    for (ll j=2;j<=k;j++)
    {
        d.clear();
        for (ll i=1;i<j;i++)
        {
            line cur={-pref[i],dp[i][0]};
            while (d.size()>1 && cur.Intersect(d[0].first)>=d[0].first.Intersect(d[1].first))
                d.pop_front();
            d.push_front({cur,i});
        }
        for (ll i=j;i<n;i++)
        {
            while (d.size()>1)
            {
                ll sz=d.size();
                if (d[sz-1].first.Calc(pref[n]-pref[i])<=d[sz-2].first.Calc(pref[n]-pref[i]))
                    d.pop_back();
                else
                    break;
            }
            dp[i][1]=d.back().first.Calc(pref[n]-pref[i])+pref[i]*(pref[n]-pref[i]);
            prevv[i][j]=d.back().second;
            line cur={-pref[i],dp[i][0]};
            while (d.size()>1 && cur.Intersect(d[0].first)>=d[0].first.Intersect(d[1].first))
                d.pop_front();
            d.push_front({cur,i});
        }
        d.clear();
        for (ll i=1;i<n;i++)
        {
            dp[i][0]=dp[i][1];
            dp[i][1]=0;
        }
    }
    ll maxx=-1,pos=0;
    for (ll i=k;i<n;i++)
    {
        if (dp[i][0]>=maxx)
        {
            maxx=dp[i][0];
            pos=i;
        }
    }
    cout << maxx << "\n";
    for (ll i=k;i>0;i--)
    {
        cout << pos << " ";
        pos=prevv[pos][i];
    }
    return 0;
}
/**
7 3
4 1 3 4 0 2 3
**/

Compilation message (stderr)

sequence.cpp: In function 'int main()':
sequence.cpp:37:23: warning: narrowing conversion of '- pref[i]' from 'long long int' to 'long double' [-Wnarrowing]
   37 |             line cur={-pref[i],dp[i][0]};
      |                       ^~~~~~~~
sequence.cpp:37:39: warning: narrowing conversion of 'dp[i][0]' from 'long long int' to 'long double' [-Wnarrowing]
   37 |             line cur={-pref[i],dp[i][0]};
      |                                ~~~~~~~^
sequence.cpp:54:23: warning: narrowing conversion of '- pref[i]' from 'long long int' to 'long double' [-Wnarrowing]
   54 |             line cur={-pref[i],dp[i][0]};
      |                       ^~~~~~~~
sequence.cpp:54:39: warning: narrowing conversion of 'dp[i][0]' from 'long long int' to 'long double' [-Wnarrowing]
   54 |             line cur={-pref[i],dp[i][0]};
      |                                ~~~~~~~^

• Subtask 1: 11 / 11
• Subtask 2: 0 / 11
• Subtask 3: 0 / 11
• Subtask 4: 0 / 17
• Subtask 5: 0 / 21
• Subtask 6: 0 / 29