数组-Leetcode-661

78 阅读2分钟

0️⃣python数据结构与算法学习路线
学习内容:

  • 基本算法:枚举、排序、搜索、递归、分治、优先搜索、贪心、双指针、动态规划等…
  • 数据结构:字符串(string)、列表(list)、元组(tuple)、字典(dictionary)、集合(set)、数组、队列、栈、树、图、堆等…

题目:

包含整数的二维矩阵 M 表示一个图片的灰度。你需要设计一个平滑器来让每一个单元的灰度成为平均灰度 (向下舍入) ,平均灰度的计算是周围的8个单元和它本身的值求平均,如果周围的单元格不足八个,则尽可能多的利用它们。

输入输出:

输入:
[[1,1,1],
[1,0,1],
[1,1,1]]
输出:
[[0, 0, 0],
[0, 0, 0],
[0, 0, 0]]
解释:
对于点 (0,0), (0,2), (2,0), (2,2): 平均(3/4) = 平均(0.75) = 0
对于点 (0,1), (1,0), (1,2), (2,1): 平均(5/6) = 平均(0.83333333) = 0
对于点 (1,1): 平均(8/9) = 平均(0.88888889) = 0

解题思路:

算法实现:

class Solution(object):
    def imageSmoother(self, M):
        ans = [[0] * len(M[0]) for _ in M] #初始化和M同长宽的二维数组,填入0
        for row,col in [(i,j) for i in range(len(M)) for j in range(len(M[0]))]:
            count = 0
            for dx,dy in [(i,j) for i in range(row - 1,row + 2) for j in range(col - 1,col + 2)]:
                if 0 <= dx < len(M) and 0<= dy < len(M[0]): 
                    ans[row][col] += M[dx][dy]
                    count += 1
            ans[row][col] //= count
        return ans

出现问题:

  1. 如何获取二维数组的长和宽?
  • 如一定义二维数组array[][],则获得该数组的长度(即行数)的代码为array.length
  • 获得第i行的宽度array[i].length

获取二维list的长宽
M, N = len(nums), len(nums[0])

  1. 生成初始值为0的m*n的二维数组
  • 使用list
arr = [[0 for i in range(n)] for j in range(m)]
  • 使用numpy
 arr = numpy.zeros((m, n))

误区:
arr = [[0]*n]*m
使用该方法生成的二维数组无法对数组值进行单个修改。