题解 | #分割等和子集#

分割等和子集

https://www.nowcoder.com/practice/65ade309fa4d4067a9add749721bfdc0

def inputArr():
    n = int(input())
    arr = list(map(int, input().split()))
    return n, arr


if __name__ == "__main__":
    n, arr = inputArr()
    # 全部元素相加总和
    total = sum(arr)
    # 总和不是偶数或者数组中的最大值大于总和的一半(剩下的全部加起来也不可能等于总和的一半)
    if total % 2 != 0 or max(arr) > total // 2:
        print("false")
    else:
        values = total // 2
        # dp[i]:取值有1或0,1表示在某个数组中的能找到相加等于i的组合
        # 全部初始为0
        dp = [0 for i in range(values + 1)]
        # 数组中只有1个元素,那么当arr[0] = i时,dp[i] = 1
        if arr[0] <= values:
            dp[arr[0]] = 1
        # print(total, values, arr)
        # print(dp)
        for i in range(n):
            # 当j小于arr[i]时,直接就等于上一轮的dp[j],因为此时不可能取arr[i]
            # 所以第二层循环只用到arr[i]。这样循环中的if语句就不需要了,节约了多余判断的,并减少了循环次数
            for j in range(values, arr[i]-1, -1):
                # if j >= arr[i]:
                    # dp[j] = max(dp[j], dp[j-arr[i]])
                    # dp[j]的值有两种选择:用或者不用arr[i],如果两者都不能得到j,dp[j]就为0,否则为1
                    dp[j] = dp[j] | dp[j - arr[i]]
            # print(dp)
            # 前面几个元素中,已经有相加等于目标值的组合,后面的元素就没必要再算了,直接打印结果,结束循环
            if dp[-1]:
                print("true")
                break
        else:
            print("false")

全部评论

相关推荐

自从我室友在计算机导论课上听说了“刷&nbsp;LeetCode&nbsp;是进入大厂的敲门砖”,整个人就跟走火入魔了一样。他在宿舍门口贴了一张A4纸,上面写着:“正在&nbsp;DP,请勿打扰,否则&nbsp;Time&nbsp;Limit&nbsp;Exceeded。”日记本的扉页被他用黑色水笔加粗描了三遍:“Talk&nbsp;is&nbsp;cheap.&nbsp;Show&nbsp;me&nbsp;the&nbsp;code。”连宿舍聚餐,他都要给我们讲解:“今天的座位安排可以用回溯算法解决,但为了避免栈溢出,我建议用动态规划。来,这是状态转移方程:dp[i][j]&nbsp;代表第&nbsp;i&nbsp;个人坐在第&nbsp;j&nbsp;个位置的最优解。”我让他去楼下取个快递,他不直接去,非要在门口踱步,嘴里念念有词:“这是一个图的遍历问题。从宿舍楼(root)到驿站(target&nbsp;node),我应该用&nbsp;BFS&nbsp;还是&nbsp;DFS?嗯,求最短路径,还是广度优先好。”和同学约好出去开黑,他会提前发消息:“集合点&nbsp;(x,&nbsp;y),我们俩的路径有&nbsp;k&nbsp;个交点,为了最小化时间复杂度,应该在&nbsp;(x/2,&nbsp;y/2)&nbsp;处汇合。”有一次另一个室友低血糖犯了,让他帮忙找颗糖,他居然冷静地分析道:“别急,这是一个查找问题。零食箱是无序数组,暴力查找是&nbsp;O(n)。如果按甜度排序,我就可以用二分查找,时间复杂度降到&nbsp;O(log&nbsp;n)。”他做卫生也要讲究算法效率:“拖地是典型的岛屿问题,要先把连通的污渍区块都清理掉。倒垃圾可以用双指针法,一个指针从左往右,一个从右往左,能最快匹配垃圾分类。”现在我们宿舍的画风已经完全变了,大家不聊游戏和妹子,对话都是这样的:“你&nbsp;Two&nbsp;Sum&nbsp;刷了几遍了?”“别提了,昨天遇到一道&nbsp;Hard&nbsp;题,我连暴力解都想不出来,最后只能看题解。你呢?”“我动态规划还不行,总是找不到最优子结构。今天那道接雨水给我整麻了。”……LeetCode&nbsp;真的害了我室友!!!
老六f:编程嘉豪来了
AI时代还有必要刷lee...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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