360笔试题（2019年8月）：求表面积

题目类似于初中堆小方块求三视图，这里求的是表面积。

输入：

第一行是M、N，代表小方块占地面积是M*N的二维区域。

以下M行分别输入每个位置小方块的个数（即高度），也就是逐行输入，

输出：表面积（刚考完，无法保存图）

测试用例：

输入：
    2 2
    2 1
    1 1
输出：20

C++代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

int main()
{
	int N, M;
	while (cin>>N>>M)
	{
		vector<vector<int>> A(N, vector<int>(M, 0));
		int S = 0;
		for (int i = 0; i < N; i++)
		{
			for (int j = 0; j < M; j++)
			{
				cin >> A[i][j];
				if (A[i][j] > 0)
				{
					S = S + 4 * A[i][j] + 2;//4个侧面+上下两个面
				}
			}
		}
		for (int i = 0; i < N; i++)
		{
			for (int j = 0; j < M-1; j++)
			{
				int mW = min(A[i][j], A[i][j + 1]);
				S = S - 2 * mW;//减去左右方向重合的面
			}
		}
		for (int i = 0; i < N - 1; i++)
		{
			for (int j = 0; j < M; j++)
			{
				int mH = min(A[i][j], A[i + 1][j]);
				S = S - 2 * mH;//减去上下方向重合的面
			}
		}
		cout << S << endl;
	}

	return 0;
}