【问题描述】
在 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 Axes3Ddef 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