题目描述
对一个长度为n的数组,某个数加一的同时对另一个数减一,任意次操作后数组最长的排序子序列的长度是多少?
新数组的元素不能为0
思路
数组的元素和是固定的,suma
将列表和suma 重新分配给lenth = n 的列表,求能组成的排列(1,2,3,4,5,...,k,...)的最大长度k
因为新数组的元素不能为0,预先给每个位置留下1
相当于放豆子游戏,n个位置,一共有suma颗豆子,每个位置先放一个豆子,剩下的位置,下标为i的地方放i个豆子,如果下标为i的地方不够i个豆子了,说明最大的排列就到i
如果每个位置都放完了,还有剩下豆子,那这部分剩下的就需要都放在最后一个位置,前面n-1个可以构成排列。
代码
python
# 将列表和suma 重新分配给lenth = n 的列表,求能组成的排列(1,2,3,4,5,...,k,...)的最大长度k
def ans(arr):
suma = sum(arr)
n = len(arr)
suma -= n
for i in range(1, n):
if suma < i :
return i
suma -= i
if suma == 0:
return n
return n-1
print(ans([2]))