题目:3417. 砝码称重
思路1:dp,一个砝码有三种选择,一个是不选,一个是放左边,一个是放右边。我们遍历n个砝码,
注意:数组容量建议开的比原来的大一倍
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=2e5+10;
int a[N],f[110][N];
int main(){
int n;
cin>>n;
int m=0;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
m+=a[i];
}
f[0][0]=1;
for(int i=1;i<=n;i++){
for(int j=0;j<=m;j++){
f[i][j]=f[i-1][abs(j-a[i])]||f[i-1][j+a[i]]||f[i-1][j];
}
}
int ans=0;
for(int i=1;i<=m;i++){
if(f[n][i]) ans++;
}
cout<<ans;
return 0;
}
思路2:(1+x-x)循环n次
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
int a[110][3];
LL c1[200010],c2[200010];
int main(){
int n;
cin>>n;
for(int i=0;i<n;i++){
scanf("%d",&a[i][0]);
a[i][1]=-a[i][0];
a[i][2]=0;
}
//c1[a[0][0]]=1;
for(int i=0;i<n;i++){
for(int j=0;j<3;j++){
c1[0]=1;
for(int k=0;k<=100000;k++){
if(abs(k+a[i][j])!=0)c2[abs(k+a[i][j])]+=c1[k];
}
for(int z=0;z<200010;z++){
if(c2[z])c1[z]=1;
else c1[z]=0;
}
}
}
int ans=0;
for(int i=1;i<=100000;i++){
if(c2[i]) ans++;
}
cout<<ans;
return 0;
}