题目

现有A,B两种物品(各有n1和n2个) 组成一个排列(全部物品都用到),

且满足条件 : 不会出现连续k个物品 (对A物品 k=k1, B物品k=k2)

解答

线性dp,设f[i][j][k][0/1] 前i个物品,j个A,k个B,最后一个取A或B

枚举最后一个连续相同物品的区间即可

f[i][j][k][0]+=f[i-1][j-x][k]     x<=k1,x<=j 
f[i][j][k][1]+=f[i-1][j][k-x]     x<=k2,x<=k

状态i可以不用(滚动数组


// f[i][j][k][0]+=f[i-1][j-x][k] x<=k1,x<=j 
// f[i][j][k][1]+=f[i-1][j][k-x] x<=k2,x<=k

#include <iostream>
#include <cstring>
using namespace std;
 const int N=102;
 typedef long long ll;
  ll n1,n2,k1,k2,f[N][N][2];
  const ll mod=1e8;
  
 int main(){
 	ll j,k;
 	cin>>n1>>n2>>k1>>k2;
 	f[0][0][0]=f[0][0][1]=1;
 	
 	for(j=0;j<=n1;j++)
 	  for(k=0;k<=n2;k++) {
 	  	if(j==0&&k==0) continue;
 	  		for(int x=1;x<=min(k1,j);x++)
  		 	f[j][k][0]+=f[j-x][k][1],f[j][k][0]%=mod;
  		 	
 	  		for(int x=1;x<=min(k2,k);x++)
 	  	 	f[j][k][1]+=f[j][k-x][0],f[j][k][1]%=mod;
		}
	cout<<(f[n1][n2][0]+f[n1][n2][1])%mod;
 } 
 
 

posted on 2022-10-08 17:52  towboat  阅读(5)  评论(0编辑  收藏  举报