01背包 uva12563+运算符重载

题意：见紫书。

思路：01背包。

一开始自己写了个wa+tle的版本。wa的原因主要是,刚开始dp都是0，根据代码，第一首歌就变成必选。这是以后需要注意的地方。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define inf 0x3f3f3f3f
#define debug cout<<"debug"<<endl;
using namespace std;
const int maxn=1e4+50;
int T,n,t,dp[maxn][maxn],a[maxn],cnt,d[maxn][2];///dp[i][j] 表示捡第i个物品 背包容量还剩j 此时所唱的最多歌曲 
int main(){
	//freopen("out.txt","w",stdout);
	while(scanf("%d",&T)!=EOF){
		for(int cas=1;cas<=T;cas++){
			memset(dp,0,sizeof(dp));
			memset(a,0,sizeof(a));
			memset(d,0,sizeof(d));
			cnt=0;
			d[1][1]=a[1];
			scanf("%d%d",&n,&t);
			for(int i=1;i<=n;i++){
				scanf("%d",&a[i]);
			}
			for(int i=1;i<=n;i++)
			for(int j=t;j>=a[i];j--){
				if(dp[i-1][j]>dp[i-1][j-a[i]]){
					dp[i][j]=dp[i-1][j];
					d[i][0]=max(d[i-1][1],d[i-1][0]);
				}
				else if(dp[i-1][j]<=dp[i-1][j-a[i]]){///因为必有dp[0][j]==dp[0][j-a[i]] 导致第一首必选，而实际上不一定要选第一首
==					dp[i][j]=dp[i-1][j-a[i]]+1;
					d[i][1]=max(max(d[i-1][0]+a[i],d[i-1][1]+a[i]),d[i][1]);						
				} 					
			}
			for(int i=0;i<=t;i++){
				if(cnt<dp[n][i]){
					cnt=dp[n][i];
				}
			}
			printf("Case %d: %d %d\n",cas,cnt+1,max(d[n][0],d[n][1])+678);
		}
	}
	return 0;
}

AC代码：参考别人代码写出来得，通过结构体重载运算符来达到比较效果，非常巧妙！

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define debug cout<<"debug"<<endl; 
using namespace std;
const int maxn=1e4+50;
int T,t,n,a[maxn],sum;

struct Song{
	int time,cnt;
	bool operator < (const Song &rhs) const{
		return (cnt<rhs.cnt||(cnt==rhs.cnt&&time<rhs.time)); ///重载运算符
	} 
}dp[maxn];

int main(){
	while(scanf("%d",&T)!=EOF){
		for(int cas=1;cas<=T;cas++){
			memset(a,0,sizeof(a));
			memset(dp,0,sizeof(dp));
			sum=0; 
			scanf("%d%d",&n,&t);
			for(int i=1;i<=n;i++){
				scanf("%d",&a[i]);
				sum+=a[i];
			}
			t=min(t-1,sum);///比较出真正用来唱前n首歌的时间
			for(int i=1;i<=n;i++)
			for(int j=t;j>=a[i];j--){
				Song temp;///如果唱第i首歌 
				temp.time=dp[j-a[i]].time+a[i];
				temp.cnt=dp[j-a[i]].cnt+1;
				if(dp[j]<temp) dp[j]=temp; ///选择更优
			}
			printf("Case %d: %d %d\n",cas,dp[t].cnt+1,dp[t].time+678);///记得最后把《劲歌金曲就》加上
		}
	}
	return 0;
}