Submission #298674

#TimeUsernameProblemLanguageResultExecution timeMemory
298674errorgornSplit the sequence (APIO14_sequence)C++14
71 / 100
119 ms131076 KiB
//雪花飄飄北風嘯嘯
//天地一片蒼茫

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/rope>
using namespace std;
using namespace __gnu_pbds;
using namespace __gnu_cxx;
#define ll long long
#define ii pair<ll,ll>
#define iii pair<ii,ll>
#define fi first
#define se second
#define endl '\n'
#define debug(x) cout << #x << " is " << x << endl

#define rep(x,start,end) for(auto x=(start)-((start)>(end));x!=(end)-((start)>(end));((start)<(end)?x++:x--))
#define all(x) (x).begin(),(x).end()
#define sz(x) (int)(x).size()

#define indexed_set tree<ll,null_type,less<ll>,rb_tree_tag,tree_order_statistics_node_update>
//change less to less_equal for non distinct pbds, but erase will bug

mt19937 rng(chrono::system_clock::now().time_since_epoch().count());

struct line{
	ll m,c;
	int idx;
	
	ii get(ll x){
		return ii(m*x+c,idx);
	}
};

struct hull{
	deque<line> dq={line({0,(ll)-1e18,-1})};
	
	double POI(line i,line j){
		return (double)(i.c-j.c)/(j.m-i.m);
	}
	
	void add(line i){
		if (dq.back().m==i.m){
			if (dq.back().c<i.c) dq.pop_back();
			else return;
		}
		while (sz(dq)>=2 && POI(dq[sz(dq)-2],dq.back())>POI(dq.back(),i)) dq.pop_back();
		dq.push_back(i);
	}
	
	ii query(ll x){
		while (sz(dq)>=2 && POI(dq[0],dq[1])<x) dq.pop_front();
		return dq.front().get(x);
	}
};
hull hulls[205];

int n,k;
ll arr[100005];
ll memo[100005][205];
int p[100005][205];

int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin.exceptions(ios::badbit | ios::failbit);
	
	cin>>n>>k;
	rep(x,1,n+1) cin>>arr[x],arr[x]+=arr[x-1];
	hulls[0].add(line({0,0,0}));
	
	k++;
	
	rep(x,1,n+1){
		rep(y,k+1,1){
			auto temp=hulls[y-1].query(arr[x]-arr[n]);
			temp.fi+=arr[x]*(arr[n]-arr[x]);
			tie(memo[x][y],p[x][y])=temp;
			
			hulls[y].add(line({arr[x],memo[x][y],x}));
		}
	}
	
	cout<<memo[n][k]<<endl;
	int px=n,py=k;
	rep(x,0,k-1){
		cout<<p[px][py]<<" ";
		px=p[px][py];
		py--;
	}
}
#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...