HDU 4721 Food and Productivity (二分+树状数组)

转载请注明出处，谢谢http://blog.csdn.net/ACM_cxlove?viewmode=contents    by---cxlove

题意 ：给出n * m的格子，每个格子有两个属性food , prod。对于每一个查询A，B，可以选择某个格子将food属性+A，prod+B，然后以这个格子为中心的正方形两个属性和的最小值最大。

http://acm.hdu.edu.cn/showproblem.php?pid=4721

其实很简单的题， 很早就会了，WA了好多天，结果是一个乘号打成了加号。。。不能更逗

由于所有属性值为正，所以没必要取边界上不完整的正方形，不过处理下也没事。。。

正方形的连长为2 * r + 1，预处理一下子矩阵和。

将所有的正方形两个属性和提取出来，离散化一下food，用BIT维护prod属性的前缀最大值。

对于每一次查询，二分答案ans , 两个属性和都要大于等于ans。

二分下food，查询一下prod的最值。

逆序后维护前缀最大值。。

 

#include <iostream>

#include <cstdio>

#include <cstring>

#include <cmath>

#include <algorithm>

#include <set>

#include <map>

#include <vector>

#include <queue>

#include <stack>

#define lson step << 1

#define rson step << 1 | 1

#define lowbit(x) (x & (-x))

#define Key_value ch[ch[root][1]][0] 

using namespace std;

typedef long long LL;

const int N = 505;

int n , m , r , q;

int food[N][N] , prod[N][N];

int a[N * N] , b[N * N] , tot , x[N * N];

int s[N * N] , cnt;

void add (int x , int v) {

    for (int i = x ; i <= cnt ; i += lowbit (i)) {

        s[i] = max (s[i] , v);

    }

}

int ask (int x) {

    int ret = 0;

    for (int i = x ; i > 0 ; i -= lowbit (i)) {

        ret = max (ret , s[i]);

    }

    return ret;

}

int main () {

    #ifndef ONLINE_JUDGE

        freopen ("input.txt" , "r" , stdin);

        // freopen ("output.txt" , "w" , stdout);

    #endif

    int t , cas = 0;

    scanf ("%d" , &t);

    while (t --) {

        memset (food , 0 , sizeof(food));

        memset (prod , 0 , sizeof(prod));

        memset (s , 0 , sizeof(s));

        scanf ("%d %d %d %d" , &n , &m , &r , &q);

        for (int i = 1 ; i <= n ; i ++) {

            for (int j = 1 ; j <= m ; j ++) {

                scanf ("%d" , &food[i][j]);

                food[i][j] += food[i - 1][j] + food[i][j - 1] - food[i - 1][j - 1];

            }

        }

        for (int i = 1 ; i <= n ; i ++) {

            for (int j = 1 ; j <= m ; j ++) {

                scanf ("%d" , &prod[i][j]);

                prod[i][j] += prod[i - 1][j] + prod[i][j - 1] - prod[i - 1][j - 1];

            }

        }

        tot = 0;

        r = 2 * r + 1;

        for (int i = r ; i <= n ; i ++) {

            for (int j = r ; j <= m ; j ++) {

                a[tot] = food[i][j] - food[i][j - r] - food[i - r][j] + food[i - r][j - r];

                b[tot] = prod[i][j] - prod[i][j - r] - prod[i - r][j] + prod[i - r][j - r];

                x[tot] = a[tot];

                tot ++;

            }

        }

        sort (x , x + tot);

        cnt = unique (x , x + tot) - x;

        for (int i = 0 ; i < tot ; i ++) {

            int y = cnt - (lower_bound (x , x + cnt , a[i]) - x);

            add (y , b[i]);

        }

        printf ("Case #%d:\n" , ++ cas);

        while (q --) {

            int A , B;

            scanf ("%d %d" , &A , &B);

            int low = min (A , B) + r * r * 1 , high = min (A , B) + r * r * 3 , mid , ans;

            while (low <= high) {

                mid = (low + high) >> 1;

                int y = cnt - (lower_bound (x , x + cnt , mid - A) - x);

                if (y < 1) {

                    high = mid - 1;

                    continue;

                }

                int tmp = ask (y);

                if (tmp + B >= mid) {

                    ans = mid;

                    low = mid + 1;

                }

                else high = mid - 1;

            }

            printf ("%d\n" , ans);

        }

        if (t) puts ("");

    }

    return 0;

}