[蓝桥杯python] 无聊的逗
问题描述
逗志芃在干了很多事情后终于闲下来了,然后就陷入了深深的无聊中。不过他想到了一个游戏来使他更无聊。他拿出n个木棍,然后选出其中一些粘成一根长的,然后再选一些粘成另一个长的,他想知道在两根一样长的情况下长度最长是多少。
1、资源限制
资源限制
时间限制:1.0s 内存限制:256.0MB
2、输入格式
第一行一个数n,表示n个棍子。第二行n个数,每个数表示一根棍子的长度。
3、输出格式
一个数,最大的长度。
4、样式输入及输出
样例输入
4
1 2 3 1
样例输出
3
5、代码及解析
具体解析请大家自己看一下代码中的备注,在此不多做解释。
(为此我还想说逗志芃是真的恶心,后面还有很多他的题,我对他很无语T_T)
思路:
- 我的思路是动态规划寻找等和子集,这道题其实就可以理解为寻找等和子集,如果不能找到,那就删除一个数再寻找
- 我的代码是先查找所给出的数组能否找到一个等和子集
- 然后再每次删除数组中的一个数然后再寻找是否能找到等和子集
- 将所有情况都记录下来,最终输出最大的等和子集
注意:代码里面很冗长,有些步骤可能没有必要,但是我不知道怎么优化。
如果大家想到了解决方法请在评论区说下,大家一起努力
n = int(input())
N = list(map(int,input().split()))
#动态规划寻找等和子集的和函数
def find_all(nums):
n = len(nums)
if n < 2:
return
total = sum(nums)
maxNum = max(nums)
if total & 1:
return
target = total // 2
if maxNum > target:
return
dp = [[False] * (target + 1) for _ in range(n)]
for i in range(n):
dp[i][0] = True
dp[0][nums[0]] = True
for i in range(1, n):
num = nums[i]
for j in range(num, target + 1):
if j >= num:
dp[i][j] = dp[i - 1][j] | dp[i - 1][j - num]
else:
dp[i][j] = dp[i - 1][j]
if dp[n - 1][target] == True:
return target
#找到等和函数中最大的数
def maxnum(n,N):
result = []
#这一步先找给出的数列中可不可以直接分为等和函数
a = find_all(N)
if a != None:
result.append(a)
#依次减去每一个数后查看是否能找到等和子集
for i in range(n):
mid = N[0:i] + N[i+1:n]
a = find_all(mid)
if a != None:
result.append(a)
#输出等和子集中最大的一个
print(max(result))
maxnum(n,N)
结果:
但是可以看出,代码还是有缺陷,但是可以解决问题
希望有大神可以在评论区提提优化意见,大家一起努力 !!!
自己写的所以有点复杂,但是至少能完成嘿嘿。如果各位有优化欢迎评论区讨论!!