結果

問題 No.913 木の燃やし方
ユーザー treeone
提出日時 2019-10-18 20:16:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,053 bytes
コンパイル時間 2,246 ms
コンパイル使用メモリ 197,268 KB
最終ジャッジ日時 2025-01-07 22:32:16
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 5 TLE * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using int64 = long long;
const int64 INF = 1e18;

struct CumulativeSum{
    vector<int64> data;
    CumulativeSum(int sz) : data(sz, 0) {};
    void add(int k, int64 x){
        data[k] += x;
    }
    void build(){
        for(int i = 1; i < data.size(); i++){
            data[i] += data[i - 1];
        }
    }
    int64 query(int k){
        if(k < 0) return 0;
        return (data[min(k, (int)data.size() - 1)]);
    }
    int64 query(int l, int r){
        return query(r - 1) - query(l - 1);
    }
};

int main(){
    int N;
	cin >> N;
    vector<int64> A(N), ans(N, INF);
    CumulativeSum S(N);
	for(int i = 0; i < N; i++){
        cin >> A[i];
        S.add(i, A[i]);
    }
    S.build();
    for(int l = 0; l < N; l++){
        int64 MIN = INF;
        for(int r = N; r > l; r--){
            int64 tmp = (r - l) * (r - l) + S.query(l, r);
            MIN = min(MIN, tmp);
            ans[r - 1] = min(ans[r - 1], MIN);
        }
    }
	for(int i = 0; i < N; i++){
		cout << ans[i] << endl;
	}
}
0