蓝桥杯第十二届省赛真题-砝码称重(C++)

砝码称重

-->练习此题<--

题目描述:

你有一架天平和 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。

本人AC代码:

#include<bits/stdc++.h>
using namespace std;
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int m=0;
    while(cin>>m){
    vector<int> ssb(m);
    int n=0;
    for(int i=0;i<m;i++){
        cin>>ssb[i];
        n+=ssb[i];
    }
    vector<vector<int>> sb(m+1,vector<int>(n+1,0));
    for(int i=0;i<=m;i++){
        sb[i][0]=true;
    }
    for(int i=1;i<=m;i++){
        for(int j=1;j<=n;j++){
            sb[i][j]=sb[i-1][j]|sb[i-1][abs(j-ssb[i-1])]|sb[i-1][j+ssb[i-1]>n?n:(j+ssb[i-1])];
        }
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        if(sb[m][i]) ans++;
    }
    cout<<ans<<endl;
    }
    return 0;
}