Submission #390029

#	Time	Username	Problem	Language	Result	Execution time	Memory
390029		vonatlus	Split the sequence (APIO14_sequence)	C++17	60 / 100	2091 ms	86344 KiB

sequence

/// adil sultanov | vonat1us 

#pragma GCC optimize("O3")
//#pragma comment(linker, "/STACK:36777216")

#include<bits/stdc++.h>

#define x first
#define y second
#define pb push_back
#define sz(x) (int) x.size()
#define all(z) (z).begin(), (z).end()
 
using namespace std;

using ll = long long;
using pii = pair<int, int>;                                   

const int MOD = 1e9 + 7; 
const int INF = 1e9 + 1e2;
  
void fin() {
#ifdef AM
    freopen(".in", "r", stdin);
#endif        
}                   

const bool flag = 0;

const int N = 1e5+10;
const int K = 210;

int p[N];

ll dp[N][2];

int pr[N][K];

void f(int r, int k) {
    if (k == 1) return;
    int cut = pr[r][k];
    if (cut == -1) cout << "what the fuck", exit(0);
    cout << cut+1 << " ";
    f(cut, k-1);
}

struct Line {
    int k, id;
    ll b;
    Line() {}
    
    Line(int k, ll b, int id): k(k), b(b), id(id) {}
    
    ll get(int x) {
        return x*1ll*k + b;
    }

    ll cross(Line l) {
        if (k == l.k) return 0ll;
        return (b-l.b)/(l.k-k);
    }
};

vector<Line> v;

void add(Line l) {
    while (sz(v) > 1 && v.back().cross(l) <= v[sz(v)-2].cross(v.back())) {
        v.pop_back();
    }
    v.pb(l);
}

int get(int x) {
    int l = 0;
    int r = sz(v)-2;
    int ans = sz(v)-1;
    while (l <= r) {
        int m = (l+r)/2;
        if (v[m].cross(v[m+1]) >= x) {
            ans = m, r = m-1;
        } else {
            l = m+1;
        }
    }
    return ans;//v[ans].get(x);
}

int n;

void solve(int k) {
    v.clear();
    int h = (k&1);      
    for (int i = 0; i < n; ++i) {
        if (i >= k-1) {
            int pos = get(p[i]);
            dp[i][h] = v[pos].get(p[i]);
            pr[i][k] = v[pos].id;
        }
        add(Line(p[i], dp[i][1-h]-p[i]*1ll*p[i], i));
    }
}

void ma1n() {
    int k;
    cin >> n >> k;
    k++;
    for (int i = 0, x; i < n; ++i) {
        cin >> x;
        p[i] = (i ? p[i-1] : 0)+x;
    }
    memset(pr, -1, sizeof(pr));
    for (int l = 2; l <= k; ++l) {
        solve(l);
    }
    cout << dp[n-1][k&1] << "\n";
    f(n-1, k);
} 

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(nullptr), fin();
    int ts = 1;
    if (flag) {
        cin >> ts;
    }
    while (ts--) {
        ma1n(); 
    }
    return 0;
}

Compilation message (stderr)

sequence.cpp: In constructor 'Line::Line(int, ll, int)':
sequence.cpp:49:8: warning: 'Line::b' will be initialized after [-Wreorder]
   49 |     ll b;
      |        ^
sequence.cpp:48:12: warning:   'int Line::id' [-Wreorder]
   48 |     int k, id;
      |            ^~
sequence.cpp:52:5: warning:   when initialized here [-Wreorder]
   52 |     Line(int k, ll b, int id): k(k), b(b), id(id) {}
      |     ^~~~

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