HFUT 1357.梯田AGAIN(安徽省2016“京胜杯”程序设计大赛 D)

梯田AGAIN

Time Limit: 1000 MS	Memory Limit: 65536 KB
Total Submissions: 23	Accepted: 8

Description

大家还记得去年的梯田吗？土豪YZK在一块小岛上有着一大片n*m的梯田，每块1*1的田地都有它的高度。奴隶们不甘被YZK剥削，他们联合起来决定发动一场海啸淹掉YZK的部分梯田。

奴隶们去年尝试了一下，结果发现，由于土质太过松软，水能够透过土地渗入到相邻的梯田，即对于海啸高度h，梯田中所有小于等于h的土地都会由于土质松软而被被淹没。

现在给你一个n*m的矩阵，代表梯田中每块田地的高度。然后给定q个询问，每个询问给定一个海啸高度h，问在此高度下，不被淹没的梯田数量是多少。

Input

第一行一个整数T，表示测试数据组数。

对于每组测试数据：

第一行三个数字n，m，q，表示梯田的行数，列数和询问数。

之后n行，每行m个数字，表示每块田地的高度，梯田高度不大于1000000。

之后q行，每行给出一个海啸高度h，问大于这个高度的梯田有多少块。

0<T<20。

0<n，m<=200。

0<=q<1000。

0<=h<=1000000.

Output

对于每个询问，给出一个整数，表示大于这个海啸高度的梯田数量。

Sample Input

2
2 2 2
1 2
3 4
2
3
2 3 3
1 2 3
3 4 5
0
4
5

Sample Output

​2
1
6
1
0

 

题目比较容易理解，可以看出，整个梯田的大小可能是200*200，而询问的数量是1000，h的最大高度为1000000

因此，可以知道，在极限数据情况下，如果保存梯田中每块田的高度的话，会有大量的重复计算，会导致超时。

 

可以将每个高度的梯田的数目记录下来，采用记忆化搜索的方式来查询。

（比赛时的数据貌似比这个容易过）

 

AC代码：GitHub

/*
By:OhYee
Github:OhYee
HomePage:http://www.oyohyee.com
Email:oyohyee@oyohyee.com
Blog:http://www.cnblogs.com/ohyee/

かしこいかわいい？
エリーチカ！
要写出来Хорошо的代码哦~
*/

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <string>
#include <iostream>
#include <vector>
#include <list>
#include <queue>
#include <stack>
#include <map>
using namespace std;

//DEBUG MODE
#define debug 0

//循环
#define REP(n) for(int o=0;o<n;o++)

const int maxh = 1000005;
int cnt[maxh];
int ans[maxh];

int Max = -1;

int Ans(int h) {
    if(h > Max)
        return 0;
    if(ans[h] == -1)
        ans[h] = cnt[h] + Ans(h + 1);

    return ans[h];
}

void Do() {
    memset(cnt,0,sizeof(cnt));
    memset(ans,-1,sizeof(cnt));
    Max = -1;


    int n,m,q;
    scanf("%d%d%d",&n,&m,&q);
    
    for(int i = 0;i < n;i++)
        for(int j = 0;j < m;j++) {
            int temp;
            scanf("%d",&temp);
            Max = max(Max,temp);
            cnt[temp]++;
        }

    REP(q) {
        int h;
        scanf("%d",&h);
        printf("%d\n",Ans(h)-cnt[h]);
    }
}

int main() {
    int T;
    scanf("%d",&T);
    while(T--)
        Do();
    return 0;
}