2021蓝桥AB组，砝码称重

试题G: 砝码称重

【问题描述】
你有一架天平和N 个砝码，这N 个砝码重量依次是W1, W2…WN。
请你计算一共可以称出多少种不同的重量？
注意砝码可以放在天平两边。
样例：

input：
3
1 4 6
output：
10

import java.util.Scanner;

public class Main_lanqiao {
    public static void main(String[] args) {
        Scanner cin = new Scanner(System.in);
        boolean[][] dp = new boolean[2][100005];
        // 因为有序的从后往前更新很难做到，所以用滚动数组
        int weight; // 新秤砣的重量

        int N = cin.nextInt();
        int sum = 0; // 求出当前秤砣重量的和，最大不超过sum，减少循环次数

        while (N != 0) {
            N--;
            weight = cin.nextInt();
            sum += weight;
            // 从前往后也行
            for (int i = sum; i >= 0; i--) { // 如果是一维一定要从后往前，我做不到
                if (dp[0][i]) { // 将上一个状态求出的可能加减这次的重量
                    int x = Math.abs(weight - i);
                    dp[1][x] = true;
                    dp[1][weight + i] = true;
                }
            }// 思路：新的值加减之前状态求出新的状态

            for (int i = 1; i <= sum; i++) {
                if (dp[1][i]) {
                    dp[0][i] = true; // 更新，不滚了
                }
            }
            dp[0][weight] = true;
            // 记录后更新，细节可以更好，就这样吧
        }

        int and = 0; // 遍历求个数
        for (int i = 1; i <= sum ; i++) {
            if (dp[0][i]) {
                and++;
            }
        }

        System.out.println(and);
    }
}

~有错误请指点