Submission #37819

#TimeUsernameProblemLanguageResultExecution timeMemory
37819MatheusLealV수열 (APIO14_sequence)C++14
21 / 100
149 ms37340 KiB
#include <bits/stdc++.h>
#define int long long
#define N 10005
#define K 220
#define f first
#define s second
using namespace std;
typedef pair<int, int> pii;

int n, k, v[N], dp[N][K], sum[N], prox[N][K], r;

vector< pair<pii, int> > line;

int F(int x, int id)
{
	return line[id].f.f*x + line[id].f.s;
}
double intersect(pii r, pii s)
{
  return ((double)(r.s - s.s))/((double)(s.f - r.f));
}

bool bad(pii A, pii B, pii C)
{
  return intersect(A, C) >= intersect(B, A);
}

bool paral(pii i, pii j)
{
	if(i.f == j.f) return 1;
	return 0;
}

void addline(pair<pii, int> l)
{
	while(line.size() >= 2 && bad( line[line.size() - 2].f, line.back().f, l.f )) line.pop_back();

	line.push_back(l);
}

pii query(int x)
{
	int p = line.size();
	if(r >= p) r = p - 1;
	while(r < p - 1 && F(x, r) > F(x, r + 1))r++;
	return pii(F(x, r), line[r].s);
}


void print(int ini, int k)
{
	if(!prox[ini][k]) return;

	cout<<prox[ini][k]<<" ";

	print(prox[ini][k] + 1, k - 1);
}

int32_t main()
{
	ios::sync_with_stdio(false); cin.tie(0);

	cin>>n>>k;

	for(int i = 1; i <= n; i++) cin>>v[i], sum[i] = sum[i - 1] + v[i];

	//memset(dp, -1, sizeof dp);

	for(int i = 0; i <= n; i++) dp[i][0] = 0;

	for(int i = 0; i <= k; i++) dp[n][i] = 0;

	for(int q = 1; q <= k ; q++)
	{
		for(int i = n - 1; i >= 1; i--)
		{

			addline(make_pair(pii( -sum[i], -(sum[i]*sum[n] - sum[i]*sum[i] + dp[i + 1][q - 1] )), i));

			pii best = query(sum[i - 1]);

			dp[i][q] = -best.f - sum[i - 1]*sum[n];

			prox[i][q] = best.s;
		}

		line.clear();

		r = 0;
	}

	cout<<dp[1][k]<<"\n";

	print(1, k);

	cout<<"\n";
}
#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...