sgu 304 Mars Stomatology

题目链接:http://acm.sgu.ru/problem.php?contest=0&problem=304

题目大意:

有N(1<=N<=600)个牙齿,K(1<=K<=N)种gum,用gum[i]需要话费b[i],治疗牙齿[i]要花费a[i],且如果没打过gum[c[i]]还要花费b[c[i]],求有p元,最多治疗几个牙齿.

题目思路:

很容易想到背包,不过p太大,所以要变异一下,并且用到贪心的想法,由于治疗牙齿不一定都要算上gum的钱,所以我们排序把相同gum的牙齿排在一起,相同gum的需要贪心,把花费大的排前面(这和接下来dp的策略有关.)

状态方程:dp[i][j] = min{dp[pre[i][j-(i-k)]+sum[k+1~i]}+ (i==k)? 0:b[c[i]];

dp[i][j],表示前i个牙齿中治疗j个牙齿的最小花费.

根据状态方程我们可知,如果花费小的在前

假设一组数据

b[]={1,2}

a    c    pre

3    1   0

1    2   1

2    2   1

3    2   1

那么dp[3][1]=4而不是3,因为花费小的在前时dp的时候,是从大的加到小的,而显然如果要治疗得多,我们肯定要挑花费小的先治疗(不考虑gum时).

ps:dp时,dp[i][0]也是需要的,因为在dp[大于i][j]时可能会用到dp[i][0],虽然我们知道dp[i][0]肯定等于0,但是由于本题要打印路径,所以如果dp[i][0]没有dp,那么fa[i][0](回溯路径的数组)肯定也没有,那么到时候回溯路径时就会断掉了~

代码:

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;

#define ll long long
#define ls rt<<1
#define rs ls|1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle (l+r)>>1
#define eps (1e-9)
#define type int
#define clr_all(x,c) memset(x,c,sizeof(x))
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))
#define MOD 1000000007
#define inf 0x3f3f3f3f
#define pi acos(-1.0)
#define M 600 +5

template <class T> void _swap(T &x,T &y){T t=x;x=y;y=t;}
template <class T> T _max(T x,T y){return x>y? x:y;}
template <class T> T _min(T x,T y){return x<y? x:y;}
int test,cas;

int n,m,p;
int b[M];
struct node{
	int a,c,id;
	void read(){scanf("%d%d",&a,&c);}
	bool operator < (const node& t){
		return (c != t.c)? c < t.c : a > t.a;
		return c < t.c;
	}
}tee[M];
int dp[M][M],pre[M],sum[M],fa[M][M];

void dfs(int x,int y,int cnt){
	for(int i=fa[x][y]+1;i<=x;i++)
		printf("%d%c",tee[i].id,(--cnt)? ' ':'\n');
	if(pre[x]) dfs(pre[x],y-(x-fa[x][y]),cnt);
}

void run(){
	int i,j,k;
	for(i=1;i<=m;i++) scanf("%d",&b[i]);
	for(i=1;i<=n;i++) tee[i].read(),tee[i].id=i;
	sort(tee+1,tee+n+1);
	sum[0]=0,sum[1]=tee[1].a,pre[1]=0;
	for(i=2;i<=n;i++){
		pre[i]=(tee[i].c==tee[i-1].c)? pre[i-1]:(i-1);
		sum[i]=tee[i].a+sum[i-1];
	}
	for(i=1;i<=n;i++) for(j=0;j<=i;j++){
		dp[i][j]=inf;
		for(k=pre[i];k<=i;k++) if(j>=(i-k) && (j-(i-k))<=pre[i]){
			int tmp=dp[pre[i]][j-(i-k)]+sum[i]-sum[k]+b[tee[i].c]*(i!=k);
			if(tmp<dp[i][j]) dp[i][j]=tmp,fa[i][j]=k;
		}
	}
	for(i=n;i>0;i--) if(dp[n][i]<=p){
		printf("%d\n",i);
		dfs(n,i,i);return;
	}
	puts("0");
}

void preSof(){
}

int main(){
	//freopen("input.txt","r",stdin);
	//freopen("output.txt","w",stdout);
	preSof();
	//run();
	while(~scanf("%d%d%d",&n,&m,&p)) run();
	//for(scanf("%d",&test),cas=1;cas<=test;cas++) run();
	return 0;
}