제출 #100221

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
100221		TadijaSebez	수열 (APIO14_sequence)	C++11	71 / 100	2061 ms	84472 KiB

sequence

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define mp make_pair
#define ldb long double
const int N=100050;
const int K=205;
const ll inf=1e18;
struct Line{ ll k,n;int id;Line(){}Line(ll a, ll b, int c):k(a),n(b),id(c){}pair<ll,int> Get(ll x){ return mp(k*x+n,id);}} Hull[N];
ldb sec(Line a, Line b){ return b.k==a.k?-1e18:(ldb)(a.n-b.n)/(b.k-a.k);}
int hsz,ptr=1;
ll dp[N],sum[N];
int go[N][K];
void Print(int n, int k)
{
	if(k==0) return;
	Print(go[n][k],k-1);
	printf("%i ",n);
}
int main()
{
	int n,k;
	scanf("%i %i",&n,&k);k++;
	for(int i=1;i<=n;i++) scanf("%lld",&sum[i]),sum[i]+=sum[i-1],dp[i]=-inf;
	for(int j=1;j<=k;j++)
	{
		hsz=0;ptr=1;
		int c=j&1,p=c^1;
		Hull[++hsz]=Line(sum[j-1],dp[j-1]-sum[j-1]*sum[j-1],j-1);
		for(int i=j;i<=n;i++)
		{
			Line l=Line(sum[i],dp[i]-sum[i]*sum[i],i);
			while(ptr<hsz && Hull[ptr].Get(sum[i])<Hull[ptr+1].Get(sum[i])) ptr++;
			dp[i]=Hull[ptr].k*sum[i]+Hull[ptr].n;
			go[i][j]=Hull[ptr].id;
			while(hsz>ptr && sec(Hull[hsz-1],Hull[hsz])>=sec(Hull[hsz],l)) hsz--;
			Hull[++hsz]=l;
		}
	}
	printf("%lld\n",dp[n]);
	Print(go[n][k],k-1);
	return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

sequence.cpp: In function 'int main()':
sequence.cpp:28:13: warning: unused variable 'p' [-Wunused-variable]
   int c=j&1,p=c^1;
             ^
sequence.cpp:23:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%i %i",&n,&k);k++;
  ~~~~~^~~~~~~~~~~~~~~
sequence.cpp:24:62: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  for(int i=1;i<=n;i++) scanf("%lld",&sum[i]),sum[i]+=sum[i-1],dp[i]=-inf;
                        ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~

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