#1198697 (C++23) No.913 木の燃やし方

• 問題: No.913 木の燃やし方
• ユーザー: askr58
• 提出日時: 2026-11-13 19:06:13
• 言語: C++23 (gcc 15.3.0 + boost 1.92.0 + ACL)
• 結果: AC
• 実行時間: 1,002 ms / 3,000 ms
• コード長: 5,366 bytes
• コンパイル時間: 4,558 ms
• コンパイル使用メモリ: 339,920 KB
• 実行使用メモリ: 11,232 KB
• 最終ジャッジ日時: 2026-11-13 19:06:56
• 合計ジャッジ時間: 40,400 ms
• ジャッジサーバーID （参考情報）: judge2_0 / judge5_0

ファイルパターン	結果
sample	AC * 3
other	AC * 35

ソースコード

// SPDX-License-Identifier: CC0-1.0
//
// [English]
// To the extent possible under law, the author(s) have dedicated all copyright
// and related and neighboring rights to this software to the public domain
// worldwide. This software is distributed without any warranty.
// See <https://creativecommons.org/publicdomain/zero/1.0/> for details.
//
// [日本語]
// 法律上可能な範囲で、著作者はこのソフトウェアに関する著作権および
// 著作隣接権を含むすべての関連する権利を、全世界でパブリックドメインに
// 提供しています。本ソフトウェアは一切の保証なく配布されます。
// 自由にコピー・改変・再配布・商用利用できます（クレジット表記は不要です）。
// 詳細は <https://creativecommons.org/publicdomain/zero/1.0/deed.ja> を参照してください。
// ※ 英語の文面および CC0 1.0 の正式な条文（英語）が優先されます。

#include <iostream>
#include <bitset>
#include <random>
#include <chrono>
#include <iomanip>
#include <set>
#include <map>
#include <queue>
#include <deque>
#include <string>
#include <stack>
#include <ranges>
#include <algorithm>
#include <vector>
using namespace std;
using ll=long long;

#include <atcoder/all>
using mint=atcoder::modint998244353;

ostream& operator<<(ostream& os,const mint& x){
	os<<x.val();
	return os;
}
istream& operator>>(istream& is,mint& x){
	int t;
	is>>t;
	x=t;
	return is;
}

template <typename S,typename T>
ostream& operator<<(ostream& os,const pair<S,T>& p);
template <typename S,typename T>
istream& operator>>(istream& is,pair<S,T>& p);
template <typename T,size_t n>
ostream& operator<<(ostream& os,const array<T,n>& arr);
template <typename T,size_t n>
istream& operator>>(istream& is,array<T,n>& arr);
template <typename T>
ostream& operator<<(ostream& os,const vector<T>& vec);
template <typename T>
istream& operator>>(istream& is,vector<T>& vec);

template <typename S,typename T>
ostream& operator<<(ostream& os,const pair<S,T>& p){
	os<<p.first<<" "<<p.second;
	return os;
}
template <typename S,typename T>
istream& operator>>(istream& is,pair<S,T>& p){
	is>>p.first>>p.second;
	return is;
}

template <typename T,size_t n>
ostream& operator<<(ostream& os,const array<T,n>& arr){
	for(int i=0;i<n;i++)os<<arr[i]<<(i+1==n?"":" ");
	return os;
}
template <typename T,size_t n>
istream& operator>>(istream& is,array<T,n>& arr){
	for(int i=0;i<n;i++)is>>arr[i];
	return is;
}
template <typename T>
ostream& operator<<(ostream& os,const vector<T>& vec){
	for(int i=0;i<(int)vec.size();i++)os<<vec[i]<<(i+1==(int)vec.size()?"":" ");
	return os;
}
template <typename T>
istream& operator>>(istream& is,vector<T>& vec){
	for(int i=0;i<(int)vec.size();i++)is>>vec[i];
	return is;
}

template<class... Vecs>
void input_vec(Vecs&... vs) {
    const auto n = get<0>(tie(vs...)).size();

    for (size_t i = 0; i < n; ++i)
        ((cin >> vs[i]), ...);
}

template <class... Vecs>
void output_vec(const Vecs&... vs) {
    const auto n = get<0>(tie(vs...)).size();

    for (size_t i = 0; i < n; ++i) {
        bool first = true;
        (((cout << (exchange(first, false) ? "" : " ") << vs[i])), ...);
        cout << endl;
    }
}

template <typename T>
vector<T> make_unique(vector<T> vec){
	ranges::sort(vec);
	vec.erase(unique(vec.begin(),vec.end()),vec.end());
	return vec;
}

vector<int> Iota(int n,int s=0){
	vector<int> res(n);
	iota(res.begin(),res.end(),s);
	return res;
}

vector<ll> Iotall(int n,ll s=0){
	vector<ll> res(n);
	iota(res.begin(),res.end(),s);
	return res;
}
void YESNO(bool f){
	if(f)cout<<"Yes"<<endl;
	else cout<<"No"<<endl;
}

using vl=vector<ll>;
using vvl=vector<vector<ll>>;
using vvvl=vector<vector<vector<ll>>>; 
using vi=vector<int>;
using vvi=vector<vector<int>>;
using vvvi=vector<vector<vector<int>>>;
int main(){
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	cout<<fixed<<setprecision(10);
	int n;
	cin>>n;
	vl a(n);
	cin>>a;
	vl ans(n,1e18);
	vl s(n+1);
	for(int i=0;i<n;i++)s[i+1]=s[i]+a[i];
	auto f=[&](int l,int r)->ll{
		return (r-l)*(ll)(r-l)+s[r]-s[l];
	};
	auto chmin=[](ll& x,ll y)->bool{
		bool res=x>y;	
		x=min(x,y);
		return res;
	};
	auto monotone_minima=[&](auto rec,int h,int w,function<ll(int,int)> f)->vl{
		if(h==1){
			vl res(1,1e18);
			for(int j=0;j<w;j++)chmin(res[0],f(0,j));
			return res;
		}
		int m=h/2;
		vl res(h,1e18);
		int aminj;
		for(int j=0;j<w;j++){
			if(chmin(res[m],f(m,j)))aminj=j;
		}
		auto lres=rec(rec,m,aminj+1,f);
		for(int i=0;i<m;i++)res[i]=lres[i];
		if(m<h-1){
			auto rres=rec(rec,h-m-1,w-aminj,[&](int x,int y)->ll{
				return f(x+m+1,y+aminj);
			});
			for(int j=m+1;j<h;j++)res[j]=rres[j-m-1];
		}
		return res;
	};
	auto rec=[&](auto rec,int l,int r)->void{
		if(l==r)return;
		if(l+1==r){
			chmin(ans[l],f(l,l+1));
			return;
		}
		int m=(l+r)/2;
		auto lres=monotone_minima(monotone_minima,m-l+1,r-m,[&](int x,int y)->ll{
			return f(x+l,y+m+1);
		});
		for(int i=l;i<=m;i++){
			if(i>l)chmin(lres[i-l],lres[i-l-1]);
			chmin(ans[i],lres[i-l]);
		}
		auto rres=monotone_minima(monotone_minima,r-m,m-l+1,[&](int x,int y)->ll{
			return f(y+l,x+m+1);
		});
		for(int i=r-1;i>=m;i--){
			if(i<r-1)chmin(rres[i-m],rres[i-m+1]);
			chmin(ans[i],rres[i-m]);
		}
		rec(rec,l,m);
		rec(rec,m+1,r);
		return;
	};
	rec(rec,0,n);
	for(ll x:ans)cout<<x<<endl;
}