Submission #703718

#TimeUsernameProblemLanguageResultExecution timeMemory
703718600Mihnea수열 (APIO14_sequence)C++17
50 / 100
2080 ms3992 KiB
#include <cmath>
#include <functional>
#include <fstream>
#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
#include <set>
#include <map>
#include <list>
#include <time.h>
#include <math.h>
#include <random>
#include <deque>
#include <queue>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <cassert>
#include <bitset>
#include <sstream>
#include <chrono>
#include <cstring>
#include <numeric>

using namespace std;

typedef long long ll;

void minup(ll& a, ll b)
{
	a = min(a, b);
}

const int N = 100000 + 7;
const int K = 200 + 7;
const ll INF = (ll)1e18 + 7;
int n;
int k;
ll a[N];
ll pref[N];
ll dp[N];
ll ndp[N];
int par[K][N];

ll get(int l, int r)
{
	return pref[r] - pref[l - 1];
}

void clr()
{
	for (int i = 0; i <= n; i++)
	{
		dp[i] = ndp[i] = INF;
	}
	dp[0] = 0;
}

void swp()
{
	for (int i = 0; i <= n; i++)
	{
		dp[i] = ndp[i];
		ndp[i] = INF;
	}
}

signed main()
{
#ifdef ONPC	
	FILE* stream;
	freopen_s(&stream, "input.txt", "r", stdin);
#else
	ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#endif

	cin >> n >> k;
	k++;
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
		pref[i] = pref[i - 1] + a[i];
	}
	clr();

	for (int step = 1; step <= k; step++)
	{
		for (int i = 1; i <= n; i++)
		{
			for (int j = 1; j <= i; j++)
			{
				minup(ndp[i], dp[j - 1] + get(j, i) * get(j, i));
				if (dp[j - 1] + get(j, i) * get(j, i) == ndp[i])
				{
					par[step][i] = j - 1;
				}
			}
		}
		swp();
	}
	ll sol = (get(1, n) * get(1, n) - dp[n]) / 2;
	cout << sol << "\n";
	int p = n;
	for (int step = k; step >= 2; step--)
	{
		cout << par[step][p] << " ";
		p = par[step][p];
	}
	cout << "\n";
	
	return 0;
}
#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...