AcWing3417.砝码称重——学习笔记

目录

题目

代码

AC结果

思路 

一、获取数据&声明变量

二、状态转移

三、遍历计数

---

题目

3417. 砝码称重 - AcWing题库https://www.acwing.com/problem/content/description/3420/

---

代码

import java.util.Scanner;

public class Main {
    public static int M = 200010;
    public static void main(String[] args){
        //获取数据&声明变量
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        int m = 0;
        int[] data = new int[n + 1];
        for(int i = 1; i <= n; i++){
            data[i] = input.nextInt();
            m = m + data[i];
        }
        boolean[][] f = new boolean[n + 1][M];

        //状态转移
        f[0][0] = true;//初始状态
        for(int i = 1; i <= n; i++){//砝码个数
            for(int j = m; j >= 0; j--){//0-m重量
                f[i][j] = f[i - 1][j] || f[i - 1][Math.abs(j - data[i])] || f[i - 1][j + data[i]];
            }
        }

        //遍历数答案
        int ans = 0;
        for(int j = 1; j <= m; j++){
            if (f[n][j]){
                ans++;
            }
        }
        System.out.println(ans);


    }
}

---

AC结果

---

思路 

这题是典型的01背包问题。下一个状态就是从上一个状态转换而来，其中下一个状态与下一个状态的区别是，对于新的一个砝码的使用。一个砝码有三种使用方法：不放，放左边，放右边。题目给各个砝码的重量，则他们的总和（m）就是可以计算的最大值，我们判断0-m中哪几个重量是可以组成的，然后记录在f数组中。

一、获取数据&声明变量

        //获取数据&声明变量
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        int m = 0;
        int[] data = new int[n + 1];
        for(int i = 1; i <= n; i++){
            data[i] = input.nextInt();
            m = m + data[i];
        }
        boolean[][] f = new boolean[n + 1][M];

n表示数据的个数，m表示数据的总和，data数组用于记录数据，f数组表示状态。

二、状态转移

        //状态转移
        f[0][0] = true;//初始状态
        for(int i = 1; i <= n; i++){//砝码个数
            for(int j = m; j >= 0; j--){//0-m重量
                //状态转移方程
                f[i][j] = f[i - 1][j] || f[i - 1][Math.abs(j - data[i])] || f[i - 1][j + data[i]];
            }
        }

f[0][0]为初始状态，初始值设置为true，表示使用前0个砝码可称重量为0。然后从前1个砝码开始，每使用一个砝码，就有三种使用方法可以选择（不放，放左边，放右边），然后判断0-m这些重量中哪些是它可以组成的。判断某一个重量时，三种使用方法中其中有一种可行（为true）即表示这个重量可以被组合出来。（根据初中学的左物右码的习惯，我们设定在左边放砝码则表示负值）因为j-data[i]可能为负值，但是重量不可能为负，则需要取他的绝对值。

三、遍历计数

        //遍历数答案
        int ans = 0;
        for(int j = 1; j <= m; j++){
            if (f[n][j]){
                ans++;
            }
        }
        System.out.println(ans);

完成第二步之后，f[n][j]已经存好了使用所有砝码可以组合的重量，然后遍历f[n]数组，若为true则表示这个重量可以被组合，则ans++记录下来，最后输出ans。