蓝桥杯 砝码称重

#include<bits/stdc++.h>
using namespace std;
int dp[105][100005];
int w[105];
int main(){
	int n;
	int sum = 0;
	cin>>n;
	//dp[i][j]表示前i个物体能否乘出j重量，假设物体都在左边
	for(int i = 1; i <= n; i++) {
		cin>>w[i];
		sum += w[i];
	}
	memset(dp, 0, sizeof(dp));
	for(int i = 0; i <= n; i++) dp[i][0] = 1;
	for(int i = 1; i <= n; i++){
		for(int j = 1; j <= sum; j++){
            //当前物品不用放和当前物品放右边
			if(dp[i - 1][j] || dp[i - 1][j + w[i]]){
				dp[i][j] = 1;
			}
            //当前物品放左边
			else if ((j - w[i] >= 0) && dp[i - 1][j - w[i]]){
				dp[i][j] = 1;
			}
            //当前物品放左边，但把原先物品挪到右边，
            //如求dp[3][10],w[3]=13。如果dp[2][3]=1，则可通过13放左边，3放右边从而转移到10.
			else if((w[i] - j >= 0) && dp[i - 1][w[i] - j]){
				dp[i][j] = 1;
			}
		}
	}
	int cnt = 0;
	for(int i = 1; i <= sum; i++){
		if(dp[n][i]) {
			cnt++;
			//cout<<i<<endl;
		}
	}
	cout<<cnt<<endl;
	return 0;
}