Submission #54286

#	Time	Username	Problem	Language	Result	Execution time	Memory
54286		updown1	Split the sequence (APIO14_sequence)	C++17	71 / 100	2069 ms	84924 KiB

sequence

/*
dp+convex hull trick
only add lines to the end of the hull bc slopes increase bc pre[i] increases
answer for i is >= answer for i-1
*/
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define For(i, a, b) for(int i=a; i<b; i++)
#define ffi For(i, 0, N)
#define ffj For(j, 0, K+1)
#define ffa ffi ffj
#define s <<" "<<
#define c <<" : "<<
#define w cout
#define e "\n"
#define pb push_back
#define mp make_pair
#define a first
#define b second
//#define int ll
//500,000,000 operations
const int MAXN = 100000;
//Global Variables
int from[MAXN][201], N, K;
ll pre[MAXN], BIG=1000000000, dp[MAXN][2];
deque<int> hull; ///(ind)

struct Line { ///Ax+B
    ll A, B, l, r;
    Line() {A=0; B=0;}
    Line(int j, int line) {
        A = pre[j];
        B = dp[j][line%2]-pre[j]*pre[j];
        //w<< j s ":" s A s B<<e;
        //w<< dp[j][line%2]-pre[j]*pre[j] s B<<e;
    }
}val[MAXN];

pair<ll, int> getit(ll X) {
    int j = hull.front();
    //w<< X s j<<e;
    return mp(val[j].A*X+val[j].B, j);
}
bool below(int ind1, int ind2, ll X) {///Is ind1 below ind2 at location
    //w<< "Comparing" s ind1 s ind2 s "at location" s X<<e;//<< val[ind1].A*X+val[ind1].B s val[ind2].A*X+val[ind2].B <<e;
    return (val[ind1].A*X+val[ind1].B <= val[ind2].A*X+val[ind2].B);
}
void put(int ind, int line, ll X) {
    val[ind] = Line(ind, line);

    while (!hull.empty()) {
        int j = hull.back();
        if (below(j, ind, val[j].l)) hull.pop_back();
        else break;
    }
    if (!hull.empty()) {
        int last = hull.back();
        if (below(last, ind, BIG)) {
            /// split the range
            ll x = (val[last].B - val[ind].B)/(val[ind].A - val[last].A);
            hull.pb(ind);
            val[last].r = x;
            val[ind].l = x+1;
            val[ind].r = BIG;
        }
    }
    while (!hull.empty()) {
        int j = hull.front();
        if (val[j].r < X) hull.pop_front();
        else break;
    }

    if (hull.empty()) {
        hull.pb(ind);
        ///range: 0-BIG
        val[ind].l = 0;
        val[ind].r = BIG;
    }
}

main() {
    //ifstream cin("test.in");
	ios_base::sync_with_stdio(0); cin.tie(0);
	cin >> N >> K;
	cin >> pre[0];
	For (i, 1, N) {ll a; cin >> a; pre[i] = a+pre[i-1];}
	For (j, 1, K+1) {
        hull.clear();
        put(j-1, j-1, pre[j]);
        For (i, j, N) {
            pair<ll, int> save = getit(pre[i]);
            dp[i][j%2] = save.a;
            //w<< "dp["<< i+1 <<"]["<<j<<"]:" s dp[i][j%2]<<e;
            from[i][j] = save.b;
            if (i != N-1) put(i, j-1, pre[i+1]);
            dp[i][(j-1)%2] = 0;
            //w<< "added" s i s j-1<<e; For(i, 0, hull.size()) {w<<hull[i].a s (i==hull.size()-1? BIG:hull[i+1].a-1) c hull[i].b <<e;}w<<e;
        }
	}
	//ffj {ffi w<< dp[i][j]<< " "; w<<e;}w<<e;
	//ffj {ffi w<< from[i][j]+1<< " "; w<<e;}w<<e;
	//ffi w<< pre[i]<<e;

	w<< dp[N-1][K%2]<<e; ll at = N-1; for (ll j=K; j>=1; j--) {w<< from[at][j]+1<< " ";at = from[at][j];}w<<e;
}

Compilation message (stderr)

sequence.cpp:82:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
 main() {
      ^

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