洛谷 P4141 消失之物(dp方案數)
阿新 • • 發佈:2021-07-26
傳送門
解題思路
用dp[i][j]表示沒有消失時前i個物品組成裝滿容積為j的揹包的方案數。
很顯然,
仿照揹包,i這一維可以省去,j要逆序。
然後再設f[i][j]表示去掉第i個物品時裝滿容積為j的方案數。
那麼在dp中減去用到i這個物品的方案數即可。
注意這時去掉第一維後,j還是要正序列舉。
AC程式碼
#include<iostream> #include<cstring> #include<algorithm> #include<cstdlib> #include<vector> #include<map> #include<queue> #include<cmath> #include<stack> #include<set> #include<bitset> #include<cstdio> #include<ctime> using namespace std; inline int read() { int f=0; char cc=getchar(); while(cc<'0'||cc>'9')cc=getchar(); while(cc>='0'&&cc<='9')f=f*10+cc-'0',cc=getchar(); return f; } int n,m,w[2005],dp[2005],f[2005]; int main(){ n=read(); m=read(); for(int i=1;i<=n;i++) w[i]=read(); dp[0]=1; for(int i=1;i<=n;i++){ for(int j=m;j>=w[i];j--){ dp[j]+=dp[j-w[i]]; dp[j]%=10; } } for(int i=1;i<=n;i++){ f[0]=1; for(int j=1;j<=m;j++){ if(j>=w[i]) f[j]=dp[j]-f[j-w[i]]; else f[j]=dp[j]; f[j]=(f[j]%10+10)%10; printf("%d",f[j]); } printf("\n"); } return 0; }