1. 程式人生 > 其它 >CF1032E The Unbearable Lightness of Weights(dp,排列組合)

CF1032E The Unbearable Lightness of Weights(dp,排列組合)

傳送門


解題思路

先不考慮砝碼質量相等的情況,設dp[i][j][k]表示前i個砝碼選j個組成重量為k的方案數。

最後判斷是否合法就從判斷方案數是否為1,變成了是否等於C(cnt,i)。(cnt為某個數字的數量,i表示選取的數量)。

可以滾動掉一維陣列。

最後一定要記得特判數字型別只有一種或兩種的情況,這時候答案為n。

AC程式碼

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=105;
const long long mod=1e9+7;
int n,ans,cnt,id,qaq;
long long a[maxn],dp[maxn][maxn*maxn],sum,d[maxn][maxn];
int main(){
	ios::sync_with_stdio(false);
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i],sum+=a[i];
	for(int i=0;i<=n;i++){
		d[i][0]=d[i][i]=1;
		for(int j=1;j<i;j++){
			d[i][j]=(d[i-1][j]+d[i-1][j-1])%mod;
		}
	}
	sort(a+1,a+n+1);
	dp[0][0]=1;
	for(int i=1;i<=n;i++){//前i個砝碼,選j個砝碼,總重量為k的方案數
		for(int j=n;j>=1;j--){
			for(int k=sum;k>=0;k--){
				if(k>=a[i]) dp[j][k]=(dp[j][k]+dp[j-1][k-a[i]])%mod;
			}
		}
	}
	for(int i=1;i<=n+1;i++){
		if(a[i]==a[i-1]) cnt++;
		else{
			qaq++;
			for(int j=1;j<=cnt;j++){
				if(dp[j][a[i-1]*j]==d[cnt][j]) ans=max(ans,j);
			}
			cnt=1;
		}
	}
	if(qaq==2||qaq==3){
		cout<<n<<endl;
		return 0;
	}
	cout<<ans<<endl;
	return 0;
}