【华为机试题】光伏场地建设规划 [Python]

2024/11/8 3:34:10 来源：https://blog.csdn.net/qq_29654325/article/details/143607680 浏览: 次关键词：【华为机试题】光伏场地建设规划 [Python]

题目

在这里插入图片描述

代码

class Solution:def func(self, input_args, area_list):count = 0for i in range(input_args[0] - input_args[2] + 1):for j in range(input_args[1] - input_args[2] + 1):count += 1 if self.area_compute(area_list,i,j,input_args[2],input_args[3]) else 0print(count)return countdef area_compute(self, area_list, x, y, length, value):compute_value = 0if length == 1:return True if area_list[x][y] >= value else Falseelse:for i in range(length):compute_value += area_list[x+i][y]if compute_value >= value:return Truefor i in range(length):compute_value += area_list[x][y+i]if compute_value >= value:return Truereturn Falses = Solution()
# input_args = [2, 5, 2, 6]
# area_list = [[1, 3, 4, 5, 8], [2, 3, 6, 7, 1]]input_args = [2, 5, 1, 6]
area_list = [[1, 3, 4, 5, 8], [2, 3, 6, 7, 1]]
s.func(input_args, area_list)