2017网易游戏雷火盘古实习生招聘笔试真题：最大和 [python]

'''
[编程题] 最大和
时间限制：1秒
空间限制：32768K
在一个N*N的数组中寻找所有横，竖，左上到右下，右上到左下，四种方向的直线连续D个数字的和里面最大的值 
输入描述:
每个测试输入包含1个测试用例，第一行包括两个整数 N 和 D :
3 <= N <= 100
1 <= D <= N
接下来有N行，每行N个数字d:
0 <= d <= 100




输出描述:
输出一个整数，表示找到的和的最大值


输入例子1:
4 2
87 98 79 61
10 27 95 70
20 64 73 29
71 65 15 0


输出例子1:
193
'''


'''
解题思路：遍历所有D*D大小的区域 并找出其中的符合条件的最大值
  1、找到D*D大小的区域
  2、计算出横，竖，左上到右下，右上到左下，四种方向的直线连续D个数字的和里面最大的值返回
  3、遍历D*D大小的所有区域，在这些最大值中找出最大的值
  
  这一次又超时了，模仿神经网络中的池化，可以尝试下窗函数遍历的思路，看能不能省点内存和时间
'''


'''
代码运行结果：
运行超时:您的程序未能在规定时间内运行结束，请检查是否循环有错或算法复杂度过大。
case通过率为98.00%

'''

N, D = [each for each in map(int, input().split())]nums_array = []for i in range(N):    nums_array.append([each for each in map(int, input().split())])def get_max(slice_array):    max_horizontal = max([sum(temp) for temp in slice_array])    max_vertical = max([sum([temp[j_] for temp in slice_array]) for j_ in range(D)])    max_lt2rb = 0    max_rt2lb = 0    for x_ in range(D):        for y_ in range(D):            if x_ == y_:                max_lt2rb += slice_array[x_][y_]            if x_ + y_ == D - 1:                max_rt2lb += slice_array[x_][y_]    return max(max_horizontal, max_vertical, max_lt2rb, max_rt2lb)maximum = 0for i in range(N-D+1):    for j in range(N-D+1):        temp_slice = [each_row[j:j+D] for each_row in nums_array[i:i+D]]        temp_slice_maximum = get_max(temp_slice)        if maximum < temp_slice_maximum:            maximum = temp_slice_maximumprint(maximum)