三角形

三角形

https://ac.nowcoder.com/acm/contest/4911/B

题意是:给你n个背包,每个背包装了mi个物品,每个背包选一个,要求出选出物品价值前k小 的不同方案种数,很容易联想到背包dp。
那么怎么做呢?我们可以考虑dp[i][j]表示前i个背包选出来价值和为j的方案个数,则有

dp[i][k]+=dp[i-1][k-a[i][j]];

这个是状态转移方程,具体看代码
CODE

#include<bits/stdc++.h>
using namespace std;

int n,k;
int dp[105][10010],a[105][110];  //dp[i][j]代表前i个宝盒我选了价值为j的方案数量
void solve()
{
    cin>>n>>k;
    dp[0][0]=1;
    for(int i=1;i<=n;i++)
    {
        int t;
        cin>>t;
        for(int j=1;j<=t;j++){
            cin>>a[i][j];
            for(int k=10000;k>=a[i][j];k--)
            {
                dp[i][k]+=dp[i-1][k-a[i][j]];
            }
        }
    }

    int ans=0;
    for(int i=0;i<=10000;i++){
        if(k>dp[n][i]) ans+=(dp[n][i])*i,k-=dp[n][i];
        else{
            ans+=k*i;
            break;
        }
    }
    cout<<ans<<endl;
}
int  main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    solve();
    return 0;
}
全部评论

相关推荐

Noel_:中石油是这样的 哥们侥幸混进免笔试名单 一看给我吓尿了
点赞 评论 收藏
分享
4 收藏 评论
分享
牛客网
牛客企业服务