J. Stone game(背包问题)
题:https://nanti.jisuanke.com/t/41420
定义 dp[x][y] 表示第 x 个数到最后一个数能组成和为 y 的方案数
#include<bits/stdc++.h> using namespace std; typedef long long ll; const int N=303; const int M=150000; const int mod=1e9+7; int dp[N][M]; int a[N]; int main(){ int t; scanf("%d",&t); while(t--){ int n,sum=0; scanf("%d",&n); for(int i=1;i<=n;i++){ scanf("%d",&a[i]); sum+=a[i]; } sort(a+1,a+1+n); for(int i=0;i<=sum;i++) dp[n+1][i]=0; int ans=0; dp[n+1][0]=1; for(int i=n;i>=1;i--){ for(int j=0;j<=sum;j++){ dp[i][j]=dp[i+1][j]; if(j>=a[i]){//刚好能加当前的a[i],和为j dp[i][j]+=dp[i+1][j-a[i]]; //加之前的是因为dp[i][j]可能有多个来源,得从多个来源中选取符合条件的 if(j>=(sum-j)&&abs(j-(sum-j))<=a[i]) ans+=dp[i+1][j-a[i]]; ans%=mod; } dp[i][j]%=mod; } } printf("%lld\n",ans); } return 0; }