蓝桥杯 砝码称重（纪念dp）

第一次遇到dp问题，值得写篇博文纪念一下！！！

题目解析

        我们规定 f [ i ] [ j ] 为一个集合，为bool型，记录的是所有可能称出的重量，即从 1~M (M为总砝码重量），而 f [ i ] [ j ] 表示只抽取前 i 个砝码，其可以称的重量为 j 。这样我们就枚举所出的重量1~M ，看是否可以凑出重量 j 。

        对于每一个砝码 i ，会面临三种选择：将砝码 i 放到天平左边（+ i），将砝码 i 放到天平右边（- i），不放砝码 i 。这三种选择又会对应三种集合的状态：将砝码 i 放到天平左边（+ i）对应着集合 f [ i-1 ] [ j-wi ] 的状态；将砝码 i 放到天平右边（- i）对应着集合 f [ i-1 ] [ j+wi ] 的状态；不放砝码 i 对应着集合 f [ i-1] [ j ] 的状态。

        f [ i ] [ j ] 这个集合的状态便取决于上面列举的三种选择的集合的状态，只要有其中一个为 true，那么 f [ i ] [ j ] 便为 true。

       对于 j 的取值是从 -M 到 M，且有 j - wi 的情况，解决办法就是给 j 加上一个偏移量 b=M。

声明：以上方法学习自yxc大佬。详情可以去acwing查看。

import java.util.*;
class Main{
	public static void main(String args[]) {
		int n,m=0;
		Scanner in=new Scanner (System.in);
		n=in.nextInt();
		int []w=new int[n+1];
		for(int i=1;i<=n;i++) {
			w[i]=in.nextInt();
			m+=w[i];
		}
		int b=m;
		boolean [][]f=new boolean[n+1][2*m+1];
		f[0][b]=true;
		for(int i=1;i<=n;i++) {
			for(int j=-m;j<=m;j++) {
				f[i][j+b]=f[i-1][j+b];
				if(j-w[i]>=-m) f[i][j+b]|=f[i-1][j+b-w[i]];
				if(j+w[i]<=m) f[i][j+b]|=f[i-1][j+b+w[i]];
				
			}
		}
		int count=0;
		for(int i=1;i<=m;i++) {
			if(f[n][i+b]==true) 
				
				++count;
		}
		System.out.print(count);
	}
}