动态规划法求解0-1背包问题2(改进算法)
#include <iostream>
#include <algorithm>
using namespace std;
int p[1000][1000];
int x[5]= {0},head[100];
int W=10;
template<class T>
T GKnapSack(int n,T w[],T v[])
{
head[0]=0;
p[0][0]=0;
p[0][1]=0;
int Left=0,Right=0,next=1;
head[0]=1;
for(int i=1; i<=n; i++)
{
int k=Left;
for(int j=Left; j<=Right; j++)
{
if(p[j][0]+w[i-1]>W)
break;
int y=p[j][0]+w[i-1];
int m=p[j][1]+v[i-1];
while(k<=Right&&(p[k][0]<y))
{
p[next][0]=p[k][0];
p[next++][1]=p[k++][1];
}
if(k<=Right&&p[k][0]==y)
{
if(m<p[k][1])
m=p[k][1];
k++;
}
if(m>p[next-1][1])
{
p[next][0]=y;
p[next++][1]=m;
}
while(k<=Right&&p[k][1]<=p[next-1][1])
{
k++;
}
}
while(k<=Right)
{
p[next][0]=p[k][0];
p[next++][1]=p[k++][1];
}
Left=Right+1;
Right=next-1;
head[i]=next;
}
return p[next-1][1];
}
template<class T>
void Traceback(int n,T w[],T v[])
{
T j=p[head[n]-1][0],m=p[head[n]-1][1];
for(int i=n; i>=1; i--)
{
for(int k=head[i-1]-1; k>=head[i-2]; k--)
{
if(p[k][0]+w[i-1]==j&&p[k][1]+v[i-1]==m)
{
x[i-1]=1;
j=p[k][0];
m=p[k][1];
break;
}
}
}
for(int i=0; i<n; i++)
{
cout<<x[i]<<" ";
}
}
int main()
{
int n=5,w[]= {2,2,6,5,4},v[]= {6,3,5,4,6},MaxValue=0;
MaxValue=GKnapSack(n,w,v);
cout<<MaxValue<<endl;
Traceback(n,w,v);
return 0;
}