字节笔试题：台阶积水问题

在网上看到的题目，没事拿来练练手，比较简单。

强迫症，把前两题给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;
}