제출 #470597

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
470597		ymm	수열 (APIO14_sequence)	C++17	0 / 100	11 ms	2848 KiB

sequence

///
///   ♪ Pizza mozzarella rella rella rella... ♪
///

#define _USE_MATH_DEFINES
#define FAST ios::sync_with_stdio(false),cin.tie(0);
#include <bits/stdc++.h>
#define Loop(x, l, r) for(int x = (l); x < (r); ++x)
#define LoopR(x, l, r) for(int x = (r)-1; x >= (l); --x)
#define all(x) x.begin(), x.end()
#define Kill(x) exit((cout << (x) << '\n', 0))
#define YN(flag) ((flag)? "YES": "NO")
#define F first
#define S second
typedef          long long   ll;
typedef unsigned long long  ull;
typedef std::pair<int,int>  pii;
typedef std::pair<ll ,ll >  pll;
using namespace std;

const int N = 100'010;
const int K = 210;
ll dp0[N], dp1[N];
ll sum[K], pro[K];
int pre[K];
ll a[N];
int n, k;

int main()
{
    FAST;
    cin >> n >> k;
    Loop(i,0,n) cin >> a[i];
    Loop(i,1,N) dp0[i] = 1e15;
    Loop(j,1,k+2)
    {
        dp1[0] = 0;
        Loop(i,0,n)
        {
            pro[j] += sum[j]*a[i], sum[j] += a[i];
            while(pre[j] < i && dp0[pre[j]] + pro[j] > dp0[pre[j]+1] + (pro[j] - a[pre[j]]*(sum[j]-a[pre[j]])))
            {
                sum[j] -= a[pre[j]];
                pro[j] -= a[pre[j]]*sum[j];
                pre[j]++;
            }
            dp1[i+1] = dp0[pre[j]]+pro[j];
        }
        Loop(i,0,N) dp0[i] = dp1[i];
    }
    ll sum = 0, sum2 = 0;
    Loop(i,0,n) sum += a[i], sum2 += a[i]*a[i];
    cout << (sum*sum-sum2)/2 - dp0[n] << '\n';
}

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