CF1032E The Unbearable Lightness of Weights(dp,排列組合)
阿新 • • 發佈:2021-10-09
傳送門
解題思路
先不考慮砝碼質量相等的情況,設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; }