結果

問題 No.77 レンガのピラミッド
コンテスト
ユーザー nanophoto12
提出日時 2015-12-27 23:26:02
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,785 bytes
コンパイル時間 1,150 ms
コンパイル使用メモリ 72,852 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-19 07:32:41
合計ジャッジ時間 1,180 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 2
other AC * 2 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <map>
#include <algorithm>
#include <cmath>

using namespace std;

#define FOR(x,y) for(int x = 0;x < y;x++)
#define LLI long long int

template<int um> class UF {
public:
    vector<int> par,rank,cnt;
    UF() {par=rank=vector<int>(um,0); cnt=vector<int>(um,1); for(int i=0;i<um;i++) par[i]=i;}
    void reinit() {FOR(i,um) rank[i]=0,cnt[i]=1,par[i]=i;}
    int operator[](int x) {return (par[x]==x)?(x):(par[x] = operator[](par[x]));}
    int count(int x) { return cnt[operator[](x)];}
    int operator()(int x,int y) {
        if((x=operator[](x))==(y=operator[](y))) return x;
        cnt[y]=cnt[x]=cnt[x]+cnt[y];
        if(rank[x]>rank[y]) return par[x]=y;
        rank[x]+=rank[x]==rank[y]; return par[y]=x;
    }
};

int main() {
    int n;
    cin >> n;
    vector<int> data(n);
    int sum = 0;
    FOR(i, n)
    {
        cin >> data[i];
        sum += data[i];
    }
    int maxSize = sqrt(sum) + 1;
    //cout << "maxSize:" << maxSize << endl;
    int minOperation = 1000000000;
    for(int i = 1; i < maxSize*2;i += 2)
    {
        if((i/2) * (i/2) > sum)
        {
            continue;
        }
        //cout << "i:" << i << endl;
        int middleIndex = i / 2;
        int count = 0;
        for(int x = 0;x < i;x++)
        {
            int height = middleIndex - abs(middleIndex-x) + 1;
            //cout << "height:" << height << endl;
            if(x < n)
            {
                count += abs(height - data[x]);
            }
            else
            {
                count += height;
            }
        }
        for(int x = i;x < n;x++)
        {
            count += data[x];
        }
        minOperation = min(minOperation, count);
    }
    cout << minOperation << endl;
    return 0;
}
0