DP问题之求具体方案+有依赖的背包问题+分组背包模型+方案数

1.求具体方案：

核心思想：判断每一个物品是否被选，也就是根据一个结果推断前一个是否备选

对于背包问题也就是看f[n][m]=f[n-1][m] or f[n-1][m-v]+w，相等时都可以选。

https://www.acwing.com/problem/content/12/

注意到字典序min,我们可以反转顺序（按照原来顺序的话不知道1该不该选），也就是令f[i][j]定义为从第i个元素到最后一个元素总容量为j的最优解,答案就是f[1][m]，我们再考虑具体方案，从1开始，能选就选。

下面是AC代码：

#include<bits/stdc++.h>
using namespace std;
int n,v1;
int v[2000],w[2000];
int dp[2000][2000];
int main(){
    cin>>n>>v1;
    for(int i=1;i<=n;i++) cin>>v[i]>>w[i];
    for(int i=n;i>=1;i--){
        for(int j=0;j<=v1;j++){
            if(j<v[i]) dp[i][j]=dp[i+1][j];
            else dp[i][j]=max(dp[i+1][j],dp[i+1][j-v[i]]+w[i]);
        }
    }
    int jj=v1;
    for(int i=1;i<=n;i++){
        if(jj<v[i]) continue;
        if(dp[i][jj]==dp[i+1][jj-v[i]]+w[i]){
            cout<<i<<" ";
            jj-=v[i];
        }
    }
}

落实到具体问题：https://www.acwing.com/problem/content/1015/

#include<bits/stdc++.h>
using namespace std;
int n,m;//n:公司；m:机器
int a[20][20];
int dp[20][20];
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++) cin>>a[i][j];
    }
    for(int i=n;i>=1;i--)
    {
        for(int j=m;j>=1;j--)
        {
            dp[i][j]=dp[i+1][j];
            for(int k=1;k<=j;k++)
            {
                dp[i][j]=max(dp[i][j],dp[i+1][j-k]+a[i][k]);
            }
        }
    }
    
    cout<<dp[1][m]<<endl;
    int jj=m;
    for(int i=1;i<=n;i++){
        for(int k=0;k<=jj;k++){
            if(dp[i][jj]==dp[i+1][jj-k]+a[i][k]){
                cout<<i<<" "<<k<<endl;
                jj-=k;
                break;
            }
        }
    }
}

2.有依赖的背包问题：https://www.acwing.com/problem/content/10/

#include<bits/stdc++.h>
using namespace std;
int n,v1,p;
vector<int> edge[110];
int v[110],w[110];
int root;
int dp[110][110];//dp[i][j]表示选了i又在i的子树中选<=j的max
void dfs(int root){
    for(int k=v[root];k<=v1;k++) dp[root][k]=w[root];
    for(int i=0;i<edge[root].size();i++){
        int x=edge[root][i];
        dfs(x);
        for(int j=v1;j>=v[x];j--){
            //选
            for(int k=0;k<=j;k++){
                dp[root][j]=max(dp[root][j],dp[root][j-k]+dp[x][k]);
            }
        }
    }
}
int main(){
    cin>>n>>v1;
    for(int i=1;i<=n;i++){
        cin>>v[i]>>w[i]>>p;
        if(p==-1){
            root=i;
            continue;
        }
        edge[p].push_back(i);
    }
    memset(dp,0xc0,sizeof(dp));
    dfs(root);
    cout<<max(dp[root][v1],0);
}

3.方案数问题：https://www.acwing.com/problem/content/11/

核心思想：只要用f[i][j]表示dp[i][j]取max的方案数

下面是AC代码：

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

ll n,v1;
ll v[1010],w[1010];
ll dp[1010][1010];
ll f[1010][1010];
ll mod=1e9+7;
int main(){
    cin>>n>>v1;
    for(ll i=0;i<=v1;i++) f[0][i]=1;
    for(ll i=1;i<=n;i++) cin>>v[i]>>w[i];
    for(ll i=1;i<=n;i++){
        for(ll j=0;j<=v1;j++){
            if(j<v[i]){
                dp[i][j]=dp[i-1][j];
                f[i][j]=f[i-1][j];
            }
            else{
                dp[i][j]=max(dp[i-1][j],dp[i-1][j-v[i]]+w[i]);
                if(dp[i][j]==dp[i-1][j]) f[i][j]=f[i-1][j];
                if(dp[i][j]==dp[i-1][j-v[i]]+w[i]) f[i][j]=(f[i][j]+f[i-1][j-v[i]])%mod;
            }
        }
    }
    cout<<f[n][v1];
}

4.分组背包：https://www.acwing.com/problem/content/489/

对于每一个主件以及他的附件，我们要么不选，要么选主见，要么选主见+附件，也就抽象成分组背包，下面是AC代码：

#include<bits/stdc++.h>
using namespace std;
int n,m;
int v[100],p[100],q;
int cnt[100][100];
int cc[100];
bool z[100];
int dp[33000];
struct node{
    int jian;
    int zhi;
};
vector<node> e[100];
int main(){
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        cin>>v[i]>>p[i]>>q;
        if(q==0) z[i]=1;
        else{
            cc[q]++;
            cnt[q][cc[q]]=i;
        }
    }
    //预处理
    for(int i=1;i<=m;i++){
        if(!z[i]) continue;
        e[i].push_back({0,0});
        e[i].push_back({v[i],p[i]*v[i]});
        for(int j=0;j<=(1<<cc[i])-1;j++){
            int hao=v[i];
            int va=v[i]*p[i];
            for(int k=0;k<=cc[i]-1;k++){
                if((j>>k)&1){//选第k+1
                    hao+=v[cnt[i][k+1]];
                    va+=v[cnt[i][k+1]]*p[cnt[i][k+1]];
                }
            }
            e[i].push_back({hao,va});
        }
    }
    for(int i=1;i<=m;i++){
        if(!z[i]) continue;
        for(int j=n;j>=0;j--){
            for(int k=0;k<e[i].size();k++){
                if(j>=e[i][k].jian) dp[j]=max(dp[j],dp[j-e[i][k].jian]+e[i][k].zhi);
            }
        }
    }
    cout<<dp[n];
}