hi.砝码称重（dp）

最新推荐文章于 2024-06-15 23:36:03 发布

Pretty Boy Fox

最新推荐文章于 2024-06-15 23:36:03 发布

阅读量819

点赞数 25

文章标签：算法 c++

本文链接：https://blog.csdn.net/2301_80065123/article/details/139049037

版权

[蓝桥杯 2021 省 AB] 砝码称重（dp）

题目描述

你有一架天平和 $N$ 个砝码, 这 $N$ 个砝码重量依次是 $W_{1}, W_{2}, \cdots, W_{N}$ 。请你计算一共可以称出多少种不同的重量?

注意砝码可以放在天平两边。

输入格式

输入的第一行包含一个整数 $N$ 。

第二行包含 $N$ 个整数: $W_{1}, W_{2}, W_{3}, \cdots, W_{N}$ 。

输出格式

输出一个整数代表答案。

样例 #1

样例输入 #1

3
1 4 6

样例输出 #1

提示说明

在这里插入图片描述

【样例说明】

能称出的 10 种重量是: $1 、 2 、 3 、 4 、 5 、 6 、 7 、 9 、 10 、 11$ 。

$\begin{aligned} &1=1 \\ &2=6-4(\text { 天平一边放 } 6, \text { 另一边放 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 \end{aligned}$

【评测用例规模与约定】

对于 $\%$ 的评测用例, $\leq N \leq 15$ 。

对于所有评测用例, $\leq N \leq 100, N$ 个砝码总重不超过 $10^5$ 。

蓝桥杯 2021 第一轮省赛 A 组 F 题（B 组 G 题）。

代码内容

// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <stack>//栈
// #include <deque>//队列
// #include <queue>//堆/优先队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器，存数组的数，表数组的长度
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const ll N=1e5+10;
ll w[110],dp[110][N];

int main()
{
    ll n;
    cin>>n;

    ll sum=0;
    for(ll i=1;i<=n;i++)
    {
        cin>>w[i];
        sum+=w[i];
    }
    
    for(ll i=1;i<=n;i++)
        for(ll j=sum;j;j--)
        {
            if(j==w[i])
                dp[i][j]=1;
            else if(dp[i-1][j])
                dp[i][j]=1;
            else if(dp[i-1][j+w[i]])
                dp[i][j]=1;
            else if(dp[i-1][abs(j-w[i])])
                dp[i][j]=1;
        }

    ll ans=0;
    for(ll i=1;i<=sum;i++)
    {
        if(dp[n][i])
            ans++;
    }

    cout<<ans<<endl;
    return 0;
}