제출 #263214

#제출 시각아이디문제언어결과실행 시간메모리
263214kaage수열 (APIO14_sequence)C++17
71 / 100
2081 ms56616 KiB
#line 2 "/Users/kaage/Desktop/ProgrammingWorkspace/library/other/template.hpp"
#define _CRT_SECURE_NO_WARNINGS
#pragma target("avx2")
#pragma optimize("O3")
#pragma optimize("unroll-loops")
#include <algorithm>
#include <bitset>
#include <cassert>
#include <cfloat>
#include <climits>
#include <cmath>
#include <complex>
#include <ctime>
#include <deque>
#include <fstream>
#include <functional>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <list>
#include <map>
#include <memory>
#include <queue>
#include <random>
#include <set>
#include <stack>
#include <string>
#include <string.h>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
#define rep(i,n) for(int i=0;i<(n);i++)
#define REP(i,n) for(int i=1;i<=(n);i++)
#define all(V) V.begin(),V.end()
typedef long long lint;
typedef unsigned long long ulint;
typedef std::pair<int, int> P;
typedef std::pair<lint, lint> LP;
constexpr int INF = INT_MAX/2;
constexpr lint LINF = LLONG_MAX/2;
constexpr double eps = DBL_EPSILON;
constexpr double PI=3.141592653589793238462643383279;
template<class T>
class prique :public std::priority_queue<T, std::vector<T>, std::greater<T>> {};
template <class T, class U>
inline bool chmax(T& lhs, const U& rhs) {
	if (lhs < rhs) {
		lhs = rhs;
		return 1;
	}
	return 0;
}
template <class T, class U>
inline bool chmin(T& lhs, const U& rhs) {
	if (lhs > rhs) {
		lhs = rhs;
		return 1;
	}
	return 0;
}
inline lint gcd(lint a, lint b) {
	while (b) {
		lint c = a;
		a = b; b = c % b;
	}
	return a;
}
inline lint lcm(lint a, lint b) {
	return a / gcd(a, b) * b;
}
bool isprime(lint n) {
	if (n == 1)return false;
	for (int i = 2; i * i <= n; i++) {
		if (n % i == 0)return false;
	}
	return true;
}
template<typename T>
T mypow(T a, lint b) {
	T res(1);
	while(b){
		if(b&1)res*=a;
		a*=a;
		b>>=1;
	}
	return res;
}
lint modpow(lint a, lint b, lint m) {
	lint res(1);
	while(b){
		if(b&1){
			res*=a;res%=m;
		}
		a*=a;a%=m;
		b>>=1;
	}
	return res;
}
template<typename T>
void printArray(std::vector<T>& vec) {
	rep(i, vec.size()){
		std::cout << vec[i];
		std::cout<<(i==(int)vec.size()-1?"\n":" ");
	}
}
template<typename T>
void printArray(T l, T r) {
	T rprev = std::prev(r);
	for (T i = l; i != rprev; i++) {
		std::cout << *i << " ";
	}
	std::cout << *rprev << std::endl;
}
LP extGcd(lint a,lint b) {
	if(b==0)return {1,0};
	LP s=extGcd(b,a%b);
	std::swap(s.first,s.second);
	s.second-=a/b*s.first;
	return s;
}
LP ChineseRem(const lint& b1,const lint& m1,const lint& b2,const lint& m2) {
	lint p=extGcd(m1,m2).first;
	lint tmp=(b2-b1)*p%m2;
	lint r=(b1+m1*tmp+m1*m2)%(m1*m2);
	return std::make_pair(r,m1*m2);
}
/*template<typename F>
inline constexpr decltype(auto) lambda_fix(F&& f){
	return [f=std::forward<F>(f)](auto&&... args){
		return f(f,std::forward<decltype(args)>(args)...);
	};
}*/
#line 3 "/Users/kaage/Desktop/ProgrammingWorkspace/library/data-structure/LiChaoTree.hpp"
class LiChaoTree{
    int n=1,id=0;
    std::vector<std::tuple<lint,lint,lint>> interval;
    std::vector<std::pair<LP,int>> node;
    std::vector<lint> cord;
    lint calc(std::pair<LP,int> l,lint x){
        return l.first.first*x+l.first.second;
    }
public:
    LiChaoTree(std::vector<lint> vec){
        vec.emplace_back(vec.back()+1);
        while(n<(int)vec.size())n*=2;
        while((int)vec.size()<n+1)vec.emplace_back(vec.back()+1);
        node.assign(2*n,{{0,LINF},-1});
        interval.emplace_back(0,0,0);
        for(int range=n;range;range>>=1){
            for(int i=0;i<n;i+=range){
                if(range==1)interval.emplace_back(vec[i],0,vec[i+range]);
                else interval.emplace_back(vec[i],vec[i+range/2],vec[i+range]);
            }
        }
        cord=std::move(vec);
    }
    void addLine(lint a,lint b){
        int cnt=1;
        std::pair<LP,int> newLine={{a,b},id};
        while(true){
            lint l=std::get<0>(interval[cnt]),m=std::get<1>(interval[cnt]),r=std::get<2>(interval[cnt]);
            if(n<=cnt){
                if(calc(node[cnt],l)>calc(newLine,l))node[cnt]=newLine;
                break;
            }
            if(calc(node[cnt],l)<calc(newLine,l)&&calc(node[cnt],r)<calc(newLine,r))break;
            if(calc(node[cnt],l)>calc(newLine,l)&&calc(node[cnt],r)>calc(newLine,r)){
                node[cnt]=newLine;
                break;
            }
            if(calc(node[cnt],m)>calc(newLine,m))std::swap(node[cnt],newLine);
            if(calc(node[cnt],l)>calc(newLine,l))cnt=cnt<<1;
            else cnt=cnt<<1|1;
        }
        id++;
    }
    std::pair<lint,int> query(int idx){
        lint x=cord[idx];
        idx+=n;
        std::pair<lint,int> res={LINF,-1};
        while(idx){
            if(chmin(res.first,calc(node[idx],x)))res.second=node[idx].second;
            idx>>=1;
        }
        return res;
    }
};
#line 3 "main.cpp"
int n,k,a[100010];
ulint b[100010],dp[2][100010];
int from[210][100010];
int main(){
	std::cin>>n>>k;
	rep(i,n){
		std::cin>>a[i];
		b[i]=a[i];
		if(i)b[i]+=b[i-1];
	}
	rep(i,k){
		std::vector<lint> vec;
		for(int j=i+1;j<n;j++)vec.push_back(b[j]);
		LiChaoTree lct(vec);
		for(int j=i+1;j<n;j++){
			lct.addLine(-b[j-1],-(dp[0][j-1]-b[j-1]*b[j-1]));
			auto p=lct.query(j-(i+1));
			from[i][j]=p.second+i;
			dp[1][j]=-p.first;
		}
		rep(j,n){
			dp[0][j]=dp[1][j];
			dp[1][j]=0;
		}
	}
	std::vector<int> vec;
	int now=n-1;
	for(int i=k-1;i>=0;i--){
		now=from[i][now];
		vec.push_back(now+1);
	}
	std::reverse(all(vec));
	std::cout<<dp[0][n-1]<<std::endl;
	printArray(vec);
}

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

/Users/kaage/Desktop/ProgrammingWorkspace/library/other/template.hpp:3: warning: ignoring #pragma target  [-Wunknown-pragmas]
/Users/kaage/Desktop/ProgrammingWorkspace/library/other/template.hpp:4: warning: ignoring #pragma optimize  [-Wunknown-pragmas]
/Users/kaage/Desktop/ProgrammingWorkspace/library/other/template.hpp:5: warning: ignoring #pragma optimize  [-Wunknown-pragmas]
/Users/kaage/Desktop/ProgrammingWorkspace/library/other/template.hpp: In instantiation of 'void printArray(std::vector<_Tp>&) [with T = int]':
main.cpp:36:16:   required from here
/Users/kaage/Desktop/ProgrammingWorkspace/library/other/template.hpp:33:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
/Users/kaage/Desktop/ProgrammingWorkspace/library/other/template.hpp:102:2: note: in expansion of macro 'rep'
#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...