洛谷 P4141 消失之物(dp方案數)

傳送門


解題思路

用dp[i][j]表示沒有消失時前i個物品組成裝滿容積爲j的揹包的方案數。
很顯然,

\[dp[i][j]+=dp[i-1][j-w[i]] \]

仿照揹包,i這一維可以省去,j要逆序。

然後再設f[i][j]表示去掉第i個物品時裝滿容積爲j的方案數。
那麼在dp中減去用到i這個物品的方案數即可。

\[f[i][j]=dp[n][j]-f[i][j-w[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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章