Submission #697873

#	Time	Username	Problem	Language	Result	Execution time	Memory
697873		fdnfksd	Split the sequence (APIO14_sequence)	C++17	0 / 100	2073 ms	131072 KiB

sequence

#include<bits/stdc++.h>
#define TASKNAME "codeforce"
#define pb push_back
#define pli pair<int,int>
#define fastio ios_base::sync_with_stdio(false); cin.tie(NULL);
using namespace std;
using ll=long long;
const ll maxN=3e5+10;
const ll inf=1e18;
const ll mod=1e9+7;
ll n,k,a[maxN],sum=0,pre[maxN],pre2[maxN],dp[maxN][201];
ll cost(ll l,ll r)
{
    return (pre[r]-pre[l-1])*(pre[r]-pre[l-1])-(pre2[r]-pre2[l-1]);
}
ll trace[maxN][201]={0};
void solve()
{
    cin >> n >> k;
    k++;
    ll ss=0;
    for(int i=1;i<=n;i++) cin >> a[i],ss+=sum*a[i],sum+=a[i];
    for(int i=1;i<=n;i++)
    {
        pre[i]=pre[i-1]+a[i];
        pre2[i]=pre2[i-1]+a[i]*a[i];
    }
    for(int i=1;i<=n;i++)
    {
        dp[i][1]=pre[i]*pre[i]-pre2[i];
    }
    dp[0][1]=inf;
    for(int j=2;j<=k;j++)
    {
        for(int i=1;i<=n;i++)
        {
            dp[i][j]=inf;
            for(int q=1;q<i;q++)
            {
                dp[i][j]=min(dp[i][j],dp[q-1][j-1]+cost(q,i));
                if(dp[i][j]==dp[q-1][j-1]+cost(q,i))
                {
                    trace[i][j]=q-1;
                }
            }
        }
    }
    cout <<ss-dp[n][k]/2<<'\n';
    ll x=n;
    for(int i=k;i>1;i--)
    {
        cout << trace[x][i]<<' ';
        x=trace[x][i];
    }
}
int main()
{
    fastio
    //freopen(TASKNAME".INP","r",stdin);
    //freopen(TASKNAME".OUT","w",stdout);
    solve();
}

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