J. Stone game（背包问题）

Posted 2022-11-02 starve

题：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;

View Code