题解 | #称砝码#

称砝码

https://www.nowcoder.com/practice/f9a4c19050fc477e9e27eb75f3bfd49c

题解讲的很明白了,相当于一个滚动数组不断更新砝码数量
注意以下几点,python的集合{}初始化需要和字典区分开及集合的初始化是{0}
遍历的时候weights由于size会变因此需要重新创建一个list存之前的weights
然后data的数据集我看大部分都是用的双遍历
其实我们可以利用字符串的方式将个数(int)乘以重量+“ ”(str)最后用split便是列表形式了,这样节省了时间和内存,我的通关内存超越了90.1%可能是因为这个原因
n=int(input())
dict={}
w_str=input().split()
s=list(map(int,input().split()))
data=""
for i in range(n):
    temp=(w_str[i]+" ")*s[i]
    data+=temp
data=list(map(int,data.split()))
#s=list(map(int,input().split()))
weights={0,}
for i in data:
    #必须加List因为不加迭代的时候weights会变,需要再创建一个列表
    for j in list(weights):
        weights.add(i+j)
    #第二种写法这个不需要加是因为先执行括号里面的集合 i+j所有子集再与weights相加
    #weights=weights.union({i+j for j in weights})
print(len(weights))


全部评论

相关推荐

点赞 评论 收藏
分享
写不来代码的小黑:这么小的城市能有做it的公司也不容易
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
1
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务