2020-7-4动态规划

问题 D: 天平问题

时间限制: 1 Sec  内存限制: 256 MB
提交 状态

题目描述

 小C为了试验小X，便为物竞的小X出了一道物理相关的题：现在给出n个质量的砝码，问小X能称出多少种质量的物品，可是总有好事者想要破坏，于是乎，n达到了500，远远超出了小X能够承受的范围，锲而不舍的他决定寻求你们的帮助。
 

输入

第一行输入一个的正整数N
以下N行每行一个不超过200的正整数，依次表示每个砝码的质量。
 

输出

输出总共能称出多少种不同质量的物品。

样例输入 Copy

3
1
3
9

样例输出 Copy

13

提示

注意：天平有两边，两边均可放。100%:  n<=500  数据保证砝码的质量之和不超过20000

提交状态

GPLv2 licensed by HUSTOJ 2020

刚刚搜了一下这个题，这个讲解更到位

第一个循环可以得到1 3 9 4 10 12 13

第二个循环可以得到 2 8 11 6 7 5

有一个很重要的想法是

因为是天平，所以可以左边一个1 +3 右边一个9，此时能称出一个5。所以才会有一个循环

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[505];
int dp[2000005];
int main()
{
    //std::ios::sync_with_stdio(false);
    //std::cin.tie(0);
   int n;
   cin>>n;
   for(int i=1;i<=n;i++)
   {
       scanf("%d",a+i);
   }
   ll ans=0;
   dp[0]=1;
  for(int i=1;i<=n;i++)
  {
      for(int j=20010;j>=a[i];j--)
        dp[j]=max(dp[j],dp[j-a[i]]);
  }
  for(int i=1;i<=n;i++)
  {
      for(int j=0;j<=20010;j++)
      {
           dp[j]=max(dp[j],dp[j+a[i]]);
      }
  }
  for(int i=1;i<=20010;i++)
    if(dp[i]) ans++;
 
cout<<ans;
    return 0;
}

 

这个用01背包写的思路是错误的，虽然包含了一种选与不选的思想，但是1+9和3+9都是属于3个里面选2个，所以此方法错误。

#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[505];
int dp[505][505];
map<int,int>mp;
int main()
{
    //std::ios::sync_with_stdio(false);
    //std::cin.tie(0);
   int n;
   cin>>n;
   for(int i=1;i<=n;i++)
   {
       scanf("%d",a+i);
   }
   ll ans=0;
   for(int i=1;i<=n;i++)//个背包
     {

      for(int j=1;j<=n;j++)//i个物品选j个
        {
           
            dp[i][j]=dp[i-1][j];//不选
            if(mp[dp[i][j]]==0&&dp[i][j]!=0) {
                    mp[dp[i][j]]=1;ans++;
                    printf("%d\n",dp[i][j]);
            }
            dp[i][j]=dp[i-1][j-1]+a[j];
            if(mp[dp[i][j]]==0&&dp[i][j]!=0) {mp[dp[i][j]]=1;ans++; printf("%d\n",dp[i][j]);}
        }
     }

cout<<ans;
    return 0;
}