小蒋的技术栈记录

小蒋的技术栈记录

砝码承重

【问题描述】

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

思路:可以用 DFS,每个砝码有三种状态,放左边,放右边或都不放,时间复杂度是 O^3,N 为 15,O^3为 10^7 左右,可以过 50% 的数据。

#include<bits/stdc++.h>
using namespace std;
const int M = 1e5+10,N = 1e2+5;
int n,ans,w[N],vis[M];

void dfs(int i,int left, int right){
    if(i>n){
        vis[max(left,right)-min(left,right)]=1;
        return;
    }
    dfs(i+1,left+w[i],right);
    dfs(i+1,left,right+w[i]);
    dfs(i+1,left,right);
}

int main( ){
    cin>>n;
    for(int i=1;i<=n;i++)cin>>w[i];
    dfs(1,0,0);
    for(int i=1;i<M;i++)if(vis[i])ans++;
    cout<<ans<<'\n';
    return 0;
}

02-07 12:53