Yukicoder No.77 レンガのピラミッド

http://yukicoder.me/problems/186

考察

移動とは

任意の列の一番上にある1つのレンガを持ち,別の列に移動するか,もしくは捨て置き場に移動すること.

なので,ピラミッドの長さを決めたら,それに対応する長さとのdiffを取る.もし超えている部分のほうが大きくても捨て置き場に置けばよいので,長さを決め打ちすれば,その長さにするための移動数が求まる.ピラミッドの長さはNよりも大きくてもよいので(特に考えず出してWAを生やした),十分な回数まで調べた.

Code

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#include <iostream>
#include <vector>
#include <string>
#include <cstring>
#include <algorithm>
#include <sstream>
#include <map>
#include <set>

#define REP(i,k,n) for(int i=k;i<n;i++)
#define rep(i,n) for(int i=0;i<n;i++)
#define INF 1<<30
#define pb push_back
#define mp make_pair

using namespace std;
typedef long long ll;
typedef pair<int,int> P;

vector<int> init(int n) {
    vector<int> ret;
    rep(i,n/2) {
        ret.push_back(i+1);
    }
    ret.push_back(n/2+1);
    for(int i=n/2; i>0; i--) {
        ret.push_back(i);
    }
    return ret;
}

int main() {
    int n;
    cin >> n;

    vector<int> v(1005);
    rep(i,n) cin >> v[i];

    int ans = INF;

    REP(size,1,1005) {
        if(size % 2 == 0) continue;
        vector<int> t = init(size);

        rep(i,n) {
            int a = 0, b = 0;

            rep(j,i) a += v[j];

            REP(j,i,size) {
                if(v[j] > t[j-i]) {
                    a += v[j] - t[j-i];
                }
                if(v[j] < t[j-i]) {
                    b += t[j-i] - v[j];
                }
            }

            REP(j,size,n) a += v[j];

            if(a >= b) {
                ans = min(ans,a);
            }
        }
    }

    cout << ans << endl;

    return 0;
}
Sep 3rd, 2015