中南林业科技大学第十一届程序设计大赛 I-背包问题

const int maxn=30;
ll v[maxn];
vector<ll>sum;
int main(){
   ll n,m;
   while(scanf("%lld%lld",&n,&m)!=EOF){
       sum.clear();
        for(int i=0;i<n;i++)
            scanf("%lld",&v[i]);
        int n1=n>>1;
        int n2=n-n1;
       for(int i=0;i<(1<<n1);i++){
            ll sum1=0;
            for(int j=0;j<n1;j++){
                if(i&(1<<j))
                    sum1+=v[j];
        }
        sum.push_back(sum1);
       }
       sort(sum.begin(),sum.end());
       ll ans=0;
       for(int i=0;i<(1<<n2);i++){
           ll sum1=0;
        for(int j=0;j<n2;j++){
            if(i&(1<<j)){
                sum1+=v[n1+j];
            }
       }
         ans+=lower_bound(sum.begin(), sum.end(), m-sum1+1) - sum.begin();
    }
    printf("%lld\n",ans);
   }
    return 0;
}

链接: https://www.nowcoder.com/acm/contest/124/I
来源:牛客网

01背包是是一个普通的动态规划入门问题:

一共有n个物品, 第i个物品的体积为v[i];

有一个背包容量为m,现在我要挑选一些物品放入这个背包

我现在知道在总体积不超过背包容量的情况下,他一共有多少种放法(总体积为0也算一种放法)。

1 <= n <= 30, 1 <= m , v[i]<= 1e9

这就是一个很简单的01背包问题,我可以告诉你核心代码怎么写:

很简单吧,但是……,你试一试吧。

输入描述:

输入有多组,每一组第一行是n和m

接下来第二行到第n+1行,第i+1行表示v[i]。

输出描述:

输出每个样例的方案数,每个答案占据一行。


const int maxn=30;
ll v[maxn];
vector<ll>sum;
int main(){
   ll n,m;
   while(scanf("%lld%lld",&n,&m)!=EOF){
       sum.clear();
        for(int i=0;i<n;i++)
            scanf("%lld",&v[i]);
        int n1=n>>1;
        int n2=n-n1;
       for(int i=0;i<(1<<n1);i++){
            ll sum1=0;
            for(int j=0;j<n1;j++){
                if(i&(1<<j))
                    sum1+=v[j];
        }
        sum.push_back(sum1);
       }
       sort(sum.begin(),sum.end());
       ll ans=0;
       for(int i=0;i<(1<<n2);i++){
           ll sum1=0;
        for(int j=0;j<n2;j++){
            if(i&(1<<j)){
                sum1+=v[n1+j];
            }
       }
         ans+=lower_bound(sum.begin(), sum.end(), m-sum1+1) - sum.begin();
    }
    printf("%lld\n",ans);
   }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/wxl7777/article/details/80512083
今日推荐