제출 #1196765

#제출 시각아이디문제언어결과실행 시간메모리
1196765Muhammad_AneeqSplit the sequence (APIO14_sequence)C++20
100 / 100
1113 ms84316 KiB
#include <iostream>
#include <vector>
#include <map>
#include <algorithm>
using namespace std;
#define ll long long
#define pii pair<ll,ll>
#define fi first
#define se second
vector<pii>hull={};
vector<int>ind;
bool com(pii a,pii b)
{
    if (a.fi/a.se!=b.fi/b.se)
        return a.fi/a.se < b.fi/b.se;
    a.fi%=a.se;
    b.fi%=b.se;
    return (a.fi*b.se<b.fi*a.se);
}
bool cw(pii a,pii b,pii c)
{
    return com({b.fi-a.fi,a.se-b.se},{c.fi-b.fi,b.se-c.se});
}
void insert(pii x,int in)
{
    while (hull.size()&&hull.back()==x)
    {
        hull.pop_back();
        ind.pop_back();
    }
    while (hull.size()>1&&!cw(hull[hull.size()-2],hull.back(),x))
    {
        hull.pop_back();
        ind.pop_back();
    }
    hull.push_back(x);
    ind.push_back(in);
}
ll vl(pii a,ll x)
{
    return x*a.se+a.fi;
}
pair<ll,int> query(ll x)
{
    ll ans=vl(hull[0],x);
    int in=ind[0];
    int st=1,en=hull.size()-1;
    while (st<=en)
    {
        int mid=(st+en)/2;
        ll ans1=vl(hull[mid-1],x),ans2=vl(hull[mid],x);
        if (ans1>ans2)
        {
            if (ans1>ans)
                in=ind[mid-1];
            ans=max(ans,ans1);
            en=mid-1;
        }
        else
        {
            if (ans2>ans)
                in=ind[mid];
            ans=max(ans,ans2);
            st=mid+1;
        }
    }
    return {ans,in};
}
inline void solve()
{
    int n,k;
    cin>>n>>k;
    int a[n+1]={};
    ll p[n+1]={};
    for (int i=1;i<=n;i++)
    {
        cin>>a[i];
        p[i]=p[i-1]+a[i];
    }
    ll dp1[n+1]={};
    for (int i=1;i<=n;i++)
        dp1[i]=p[i]*(p[n]-p[i]);
    ll dp[n+1]={};
    int pre[k+1][n+1]={};
    for (int l=2;l<=k;l++)
    {
        hull={};
        ind={};
        insert({dp1[l-1]-p[n]*p[l-1],p[l-1]},l-1);
        for (int i=l;i<=n;i++)
        {
            pii y=query(p[i]);
            pre[l][i]=y.se;
            dp[i]=y.fi+(p[n]-p[i])*p[i];
            insert({dp1[i]-p[n]*p[i],p[i]},i);
        }
        for (int i=1;i<=n;i++)
        {
            dp1[i]=dp[i];
            dp[i]=0;
        }
    }
    ll ans=-1;
    int ind=1;
    for (int i=1;i<=n;i++)
    {
        if (dp1[i]>ans)
        {
            ans=dp1[i];
            ind=i;
        }
    }
    int ans1[k]={};
    for (int l=k;l>=1;l--)
    {
        ans1[l-1]=ind;
        ind=pre[l][ind];
    }
    cout<<ans<<endl;
    for (auto i:ans1)
        cout<<i<<' ';
    cout<<endl;
}
int main()
{
    ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
    int t=1;
    for (int i=1;i<=t;i++)
    {
        solve();
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...