AcWing 3417. 砝码称重

原题链接
考察:背包dp
yxc:没做出来本题的反思自己有没有认真听基础课
我:菜狗不配()
思路:
  dfs写的,当时觉得铁TLE,但是没有想到别的做法(.)
  f[i][j]表示前i个物品重量为j是否能实现.
dp方程:
f[i][j] = f[i-1][j]
f[i][j] = f[i-1][j+w[i]]
f[i][j] = f[i-1][j-w[i]]
  下标为负直接偏移就行了

Code

#include <iostream>
#include <cstring>
using namespace std;
const int N = 110,M = 100010;
int w[N],n,cnt;
bool f[N][M<<1];
int main()
{
    scanf("%d",&n);
    int sum = 0;
    for(int i=1;i<=n;i++) scanf("%d",&w[i]),sum+=w[i];
    f[0][sum] = 1;
    for(int i=1;i<=n;i++)
      for(int j=0;j<=2*sum;j++)
      {
          f[i][j] = f[i-1][j];
          if(j+w[i]<=2*sum) f[i][j] = max(f[i][j],f[i-1][j+w[i]]);
          if(j-w[i]>=0) f[i][j] = max(f[i][j],f[i-1][j-w[i]]);
      }
    for(int i=sum+1;i<=2*sum;i++)
        cnt += f[n][i];
    printf("%d\n",cnt);
    return 0;
}
posted @ 2021-06-04 00:00  acmloser  阅读(49)  评论(0编辑  收藏  举报