题目
Description
大家还记得去年的梯田吗?土豪YZK在一块小岛上有着一大片nm的梯田,每块11的田地都有它的高度。奴隶们不甘被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
5Sample Output
2
1
6
1
0
题解
题目比较容易理解,可以看出,整个梯田的大小可能是200*200,而询问的数量是1000,h的最大高度为1000000
因此,可以知道,在极限数据情况下,如果保存梯田中每块田的高度的话,会有大量的重复计算,会导致超时。
可以将每个高度的梯田的数目记录下来,采用记忆化搜索的方式来查询。
(比赛时的数据貌似比这个容易过)
代码
/* 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; }