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

 1 /*
 2 By:OhYee
 3 Github:OhYee
 4 HomePage:http://www.oyohyee.com
 5 Email:[email protected]
 6 Blog:http://www.cnblogs.com/ohyee/
 7
 8 かしこいかわいい？
 9 エリーチカ！
10 要写出来Хорошо的代码哦~
11 */
12
13 #include <cstdio>
14 #include <algorithm>
15 #include <cstring>
16 #include <cmath>
17 #include <string>
18 #include <iostream>
19 #include <vector>
20 #include <list>
21 #include <queue>
22 #include <stack>
23 #include <map>
24 using namespace std;
25
26 //DEBUG MODE
27 #define debug 0
28
29 //循环
30 #define REP(n) for(int o=0;o<n;o++)
31
32 const int maxh = 1000005;
33 int cnt[maxh];
34 int ans[maxh];
35
36 int Max = -1;
37
38 int Ans(int h) {
39     if(h > Max)
40         return 0;
41     if(ans[h] == -1)
42         ans[h] = cnt[h] + Ans(h + 1);
43
44     return ans[h];
45 }
46
47 void Do() {
48     memset(cnt,0,sizeof(cnt));
49     memset(ans,-1,sizeof(cnt));
50     Max = -1;
51
52
53     int n,m,q;
54     scanf("%d%d%d",&n,&m,&q);
55
56     for(int i = 0;i < n;i++)
57         for(int j = 0;j < m;j++) {
58             int temp;
59             scanf("%d",&temp);
60             Max = max(Max,temp);
61             cnt[temp]++;
62         }
63
64     REP(q) {
65         int h;
66         scanf("%d",&h);
67         printf("%d\n",Ans(h)-cnt[h]);
68     }
69 }
70
71 int main() {
72     int T;
73     scanf("%d",&T);
74     while(T--)
75         Do();
76     return 0;
77 }