Algoogle

Algorithm for Programming Contest

JOI 春合宿 2007 Mall

Category: JOI Tag: imos-method

Mall

問題概要


h*wのマスに使用可能な場所とそうでない場所がある.
購入可能ならそのコストが, そうでないならー1が入力で与えられる.
今b*aのエリアを買いたい.
その最小のコストを求めよ. ただし必ず買えるようなエリアが1つは存在するとする.

解法


imos法で2次元累積和をとって最小値を求めれば良い.
購入不可能な場所は適当に大きい値を入れておく.

コード


(Mall.cpp) download
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
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int inf = 1e9;

int w, h, a, b, f[1024][1024], g[1024][1024];

int solve()
{
        for (int i = 0; i < h; i++) {
                for (int j = 0; j < w; j++) {
                        if(f[i][j]<0) f[i][j] = inf;
                        g[i][j] += f[i][j];
                        g[i][min(w,j+a)] -= f[i][j];
                        g[min(h,i+b)][j] -= f[i][j];
                        g[min(h,i+b)][min(w,j+a)] += f[i][j];
                }
        }
        for (int i = 0; i < h; i++)
                for (int j = 1; j < w; j++)
                        g[i][j] += g[i][j-1];

        for (int i = 1; i < h; i++)
                for (int j = 0; j < w; j++)
                        g[i][j] += g[i-1][j];

        int ans = inf;
        for (int i = b-1; i < h; i++)
                for (int j = a-1; j < w; j++)
                        ans = min(ans, g[i][j]);
        return ans;
}

void input()
{
        cin >> w >> h >> a >> b;
        for (int i = 0; i < h; i++)
                for (int j = 0; j < w; j++)
                        cin >> f[i][j];
}

signed main()
{
        cin.tie(0);
        cin.sync_with_stdio(0);
        input();
        cout << solve() << endl;
        return 0;
}

Comments