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;
}
View Code

 

posted @ 2019-09-15 23:06  starve_to_death  阅读(271)  评论(0编辑  收藏  举报