木板存水问题
空地上竖立着n个从左到右排列的木板,它们可以把水挡住,但溢出最边上木板的水将会流到空地上。已知木板间距都是单位1,现给定每个木板的高度,请求出总共能接住的水量?说明一点,这里只考虑间距(宽度)和高度,不考虑第三个维度,因此水量是平方单位。
比如下面的木板长度就是2,1,3,可以容下4单位水
|
|- -|
|-|-|
输入描述:
第一行一个正整数T,表示T个测试样例;
对于每个测试样例,
输入正整数n (n<=1e5) ,表示n块木板;
接下来输入n个正整数,表示木板高度 h (h<1e4)。
输出描述:
输出行,每行一个正整数,表示每个样例能接住的最大平方单
位的水量。
示例1输入输出示例仅供调试,后台判题数据一般不包含示例
输入
2
3
2 1 3
2 4 3
输出
4
5
思路分析:
从两端的木板开始比较(不断向中间移动),比较得出那个较短的板的高度,求出其的存水量。逻辑上可以理解成不断降低所有木板高度 。
代码:
#include <stdio.h>
#define N 100005
int main(void)
{
int t,n,j=0;
int result[1000] = {0};
scanf("%d", &t);//t表示有t组数据
scanf("%d", &n);//n表示有n块木板
while (j<t) {
int i, num[N];//数组 num[N] 用来存每块木板长度
for (i = 0; i < n; i++)
scanf("%d", num + i);
int left = 0, right = n - 1, res = 0, counted = 0;
//从最右边的木板和最左边的木板开始比较,然后向中间移动
while (right > left) {
//左右木板中最小的那个记为min
int min = num[left] < num[right] ? num[left] : num[right];
//不断降低木板高度
if (min > counted) {
res += (min - counted) * (right - left);
counted = min;
}
if (num[left] < num[right])
left++;
else
right--;
}
//存入结果的数组
result[j] = res;
j++;
}
//打印结果
for(int i=0; i<t; i++)
printf("%d\n", result[i]);
return 0;
}