蓝桥杯每日一练——蓝桥杯2021年第十二届省赛真题-砝码称重 python

题目描述
你有一架天平和 N 个砝码,这 N 个砝码重量依次是 W1, W2, · · · , WN。
请你计算一共可以称出多少种不同的重量?
注意砝码可以放在天平两边。
输入
输入的第一行包含一个整数 N。
第二行包含 N 个整数:W1, W2, W3, · · · , WN。
输出
输出一个整数代表答案。
样例输入
3
1 4 6
样例输出
10
提示
【样例说明】
能称出的 10 种重量是:1、2、3、4、5、6、7、9、10、11。
1 = 1;
2 = 6 4 (天平一边放 6,另一边放 4);
3 = 4 1;
4 = 4;
5 = 6 1;
6 = 6;
7 = 1 + 6;
9 = 4 + 6 1;
10 = 4 + 6;
11 = 1 + 4 + 6。
【评测用例规模与约定】
对于 50% 的评测用例,1 ≤ N ≤ 15。
对于所有评测用例,1 ≤ N ≤ 100,N 个砝码总重不超过 100000。

n = int(input())
lists = list(map(int,input().strip().split()))
sum = sum(lists)
lists1 = [[j for j in lists] for i in range(sum+1)]
for i in range(1,sum+1):
    for j in range(n):
        lists1[i][j] = 0
# print(lists1)
for i in range(1,sum+1):
    for j in range(n):
        if i == lists[j]:
            lists1[i][j] = 1
            continue
        try:
            if lists1[i-lists[j]][j-1] == 1 and i-lists[j] > 0 and j-1 >= 0:
                lists1[i][j] = 1
                continue
        except:
            pass
        if lists1[i][j-1] == 1 :
            lists1[i][j] = 1
            continue
for i in range(1,sum+1):
    for j in range(n):
        try:
            if lists1[i+lists[j]][n-1] == 1 and i+lists[j] <= sum:
                lists1[i][j] = 1
                continue
        except:
            pass
        if lists1[i][j-1] == 1 :
            lists1[i][j] = 1
            continue

sum1 = 0
for i in range(1,sum+1):
    if lists1[i][n-1] == 1:
        sum1 += 1
print(sum1)

你可能感兴趣的:(蓝桥杯,算法,图论)