在网上看到的题目,没事拿来练练手,比较简单。
强迫症,把前两题给PS掉了,哈哈~~
解题思路
咋一看这道题,总感觉在考察我们数据结构,比如构造一个多边形,算出面积,然后减去台阶所在空间,好像也不够巧妙吧。算了,用最直观的方式解一下(有大神看到,记得指点一二):
(1) 算出最高的台阶数;
// 找出最高的台阶
int maxValInSteps(uint32_t *stepArr, uint32_t nSize)
{
uint32_t maxVal = 0;
for (uint32_t i = 0; i < nSize; i++)
{
if (stepArr[i] > maxVal)
maxVal = stepArr[i];
}
return maxVal;
}
(2)计算第n层时,把此层第一个台阶,和最后一个不低于n的台阶,填充中间的凹陷部分;
// 计算第n层的积水量
int CulNthLevelWaters(uint32_t *stepArr, uint32_t nSize, uint32_t n)
{
uint32_t bPos = -1;
uint32_t ePos = -1;
uint32_t nWaters = 0;
// 找第一个台阶
for (uint32_t i = 0; i < nSize; i++)
{
if (stepArr[i] == n)
{
bPos = i;
break;
}
}
// 找最后一个台阶
for (uint32_t i = nSize - 1; i >= 0; i--)
{
if (stepArr[i] >= n)
{
ePos = i;
break;
}
}
// 填充中间的凹陷部分
for (uint32_t i = bPos; i <= ePos; i++)
{
if (stepArr[i] < n)
{
nWaters += (n - stepArr[i]);
stepArr[i] = n;
}
}
return nWaters;
}
(3) 从1开始,依次填充每一层,累加的积水量;
// 计算台阶积水量
int CulFootstepWaters(uint32_t *stepArr, uint32_t nSize)
{
uint32_t maxVal = maxValInSteps(stepArr, nSize);
uint32_t sumWaters = 0;
for (uint32_t iLevel = 1; iLevel <= maxVal; iLevel++)
{
// 找到两头的台阶,填充中间凹陷的部分
sumWaters += CulNthLevelWaters(stepArr, nSize, iLevel);
}
return sumWaters;
}
运行结果
这里偷下懒,直接构造数据运行了哈,运行结果:6
int main(int agrc, char **agrv)
{
uint32_t stepArr[] = { 0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1 };
printf("sum-water = %d\n", CulFootstepWaters(stepArr, sizeof(stepArr) / sizeof(uint32_t)));
return 0;
}