0/1背包问题多种解法

目录

递推代码

记忆优化

交替滚动

自我滚动

题目练习转接

总结

---

递推代码

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
const int n=1005;
int w[n],v[n];
int dp[n][n];

int solve(int N,int V){
  for(int i=1;i<=N;++i){
    for(int j=0;j<=V;++j){
      if(v[i]>j)dp[i][j]=dp[i-1][j];
      else dp[i][j]=max(dp[i-1][j],dp[i-1][j-v[i]]+w[i]);
    }
  }

  return dp[N][V];  
}

int main()
{ 
  int N,V;
  cin>>N>>V;
  for(int i=1;i<=N;++i){
    cin>>v[i]>>w[i];
  }
  cout<<solve(N,V);
  return 0;
}

记忆优化

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
const int n=1005;
int w[n],v[n];
int dp[n][n];

int solve(int i,int j){
  if(dp[i][j]!=0){
    return dp[i][j];
  }
  if(0==i)return 0;
  int res;
  if(v[i]>j) res=solve(i-1,j);
  else res=max(solve(i-1,j),solve(i-1,j-v[i])+w[i]);
  return dp[i][j]=res;  
}

int main()
{ 
  int N,V;
  cin>>N>>V;
  for(int i=1;i<=N;++i){
    cin>>v[i]>>w[i];
  }
  cout<<solve(N,V);
  return 0;
}

交替滚动

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
const int n=1005;
int w[n],v[n];
int dp[2][n];

int solve(int N,int V){
  int now=0,old=1;
  for(int i=1;i<=N;++i){
    swap(old,now);
    for(int j=0;j<=V;++j){
      if(v[i]>j)dp[now][j]=dp[old][j];
      else dp[now][j]=max(dp[old][j],dp[old][j-v[i]]+w[i]);
    }
  }
  return dp[now][V];
}

int main()
{ 
  int N,V;
  cin>>N>>V;
  for(int i=1;i<=N;++i){
    cin>>v[i]>>w[i];
  }
  cout<<solve(N,V);
  return 0;
}

自我滚动

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
const int n=1005;
int w[n],v[n];
int dp[n];

int solve(int N,int V){
  for(int i=1;i<=N;++i){
    for(int j=V;j>=v[i];--j){
      dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
    }
  }
  return dp[V];
}

int main()
{ 
  int N,V;
  cin>>N>>V;
  for(int i=1;i<=N;++i){
    cin>>v[i]>>w[i];
  }
  cout<<solve(N,V);
  return 0;
}

题目练习转接

蓝桥杯-小明的背包1

总结

1.继续努力奋斗哦~

2.有问题还请多多宽容和反馈。

3.有问题才有进步！