【问题描述】
在 N * N 的网格上，我们放置一些 1 * 1 * 1 的立方体。
每个值 v = grid[i][j] 表示 v 个正方体叠放在对应单元格 (i, j) 上。
请你返回最终形体的表面积。
示例 1：
输入：[[2]]
输出：10
示例 2：
输入：[[1,2],[3,4]]
输出：34
示例 3：
输入：[[1,0],[0,2]]
输出：16
示例 4：
输入：[[1,1,1],[1,0,1],[1,1,1]]
输出：32
示例 5：
输入：[[2,2,2],[2,1,2],[2,2,2]]
输出：46
提示：
1 <= N <= 50
0 <= grid[i][j] <= 50
【解答思路】
1. 时间复杂度：O(N^2)
一个球六个面
当垂直时，重叠（垂直个数减1）
当水平时（行列），重叠（取矮的个数）
grid[i][j]表示在坐标(i, j)上有grid[i][j]个正方体。
例子：[[1, 2], [3, 4]]，
grid[0][0] = 1，表示坐标(0, 0)上有1个正方体。
grid[0][1] = 2，表示坐标(0, 1)上有2个正方体。
grid[1][0] = 3，表示坐标(1, 0)上有3个正方体。
grid[1][1] = 4，表示坐标(1, 1)上有4个正方体。
(中间两者与顺序无关)
public int surfaceArea(int[][] grid) {
// 习惯上应该做参数检查，但题目中给出了 N >= 1 ，故可以略去
int rows = grid.length;
// 题目保证了输入一定是 N * N，但为了使得程序适用性更强，还是单独把 cols 做赋值
int cols = grid[0].length;
int sum = 0;
// 垂直重叠
int verticalOverlap = 0;
// 行重叠
int rowOverlap = 0;
// 列重叠
int colOverlap = 0;
for (int i = 0; i < rows; i++) {
for (int j = 0; j < cols; j++) {
sum += grid[i][j];
if (grid[i][j] > 1) {
verticalOverlap += (grid[i][j] - 1);
}
if (j > 0) {
rowOverlap += Math.min(grid[i][j - 1], grid[i][j]);
}
if (i > 0) {
colOverlap += Math.min(grid[i - 1][j], grid[i][j]);
}
}
}
return sum * 6 - (verticalOverlap + rowOverlap + colOverlap) * 2;
}
//代码链接：https://leetcode-cn.com/problems/surface-area-of-3d-shapes/solution/hua-tu-ji-suan-san-ge-zhong-die-bu-fen-by-liweiwei/
【总结】
- 类似题目没有必要刷 题目比较难懂
- 数据分析（pandas）和数据可视化（matplotlib）的工作使用 python
- 多查google，多看官方文档
= 官方文档一般都会给出api的使用方法
from functools import reduce
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
def draw(grid):
n = len(grid)
z_max = np.max(grid)
# 设置长宽高
x, y, z = np.indices((n, n, z_max))
cubes = []
for i in range(n):
for j in range(n):
# 这是每个柱体
cubes.append((x == i) & (y == j) & (z < grid[i][j]))
# 创建voxels，包含所有柱体
voxels = reduce(lambda x, y: x | y, cubes)
# 设置颜色
colors = np.empty(voxels.shape, dtype=object)
for cube in cubes:
colors[cube] = 'red'
# 画图！
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.voxels(voxels, facecolors=colors, edgecolor='k')
plt.show()
# 画一下示例2的3D图
draw([[1, 2], [3, 4]])
//代码来源https://mp.weixin.qq.com/s/IZCw6GtdFLUcixp3EiP5yg
