合唱团-python


题目描述

有 n 个学生站成一排,每个学生有一个能力值,牛牛想从这 n 个学生中按照顺序选取 k 名学生,要求相邻两个学生的位置编号的差不超过 d,使得这 k 个学生的能力值的乘积最大,你能返回最大的乘积吗?


输入描述

每个输入包含 1 个测试用例。每个测试数据的第一行包含一个整数 n (1 <= n <= 50),表示学生的个数,接下来的一行,包含 n 个整数,按顺序表示每个学生的能力值 ai(-50 <= ai <= 50)。接下来的一行包含两个整数,k 和 d (1 <= k <= 10, 1 <= d <= 50)。


输出描述

输出一行表示最大的乘积。


示例1

输入

    3
    7 4 7
    2 50

输出

    49

代码

number=int(raw_input(""))
values=[int(x) for x in raw_input("").split()]
k,d=raw_input("").split()
k=int(k)
d=int(d)

max_sing_list=[[0 for x in xrange(k)] for y in xrange(number)]
min_sing_list=[[0 for x in xrange(k)] for y in xrange(number)]

for index in xrange(number):
    max_sing_list[index][0]=values[index]
    min_sing_list[index][0]=values[index]
final_max_data=-10000
for index_i in xrange(number):
    for index_j in xrange(1,k):
        before_position=index_i-1
        stop_position=max(index_i-d,0)
        while before_position>=stop_position:

            max_sing_list[index_i][index_j]=max(max_sing_list[index_i][index_j],max_sing_list[before_position][index_j-1]*values[index_i])
            max_sing_list[index_i][index_j]=max(max_sing_list[index_i][index_j],min_sing_list[before_position][index_j-1]*values[index_i])

            min_sing_list[index_i][index_j]=min(min_sing_list[index_i][index_j],min_sing_list[before_position][index_j-1]*values[index_i])
            min_sing_list[index_i][index_j]=min(min_sing_list[index_i][index_j],max_sing_list[before_position][index_j-1]*values[index_i])
            before_position-=1
    final_max_data=max(final_max_data,max_sing_list[index_i][k-1])
print final_max_data

你可能感兴趣的:(python算法题)