uva10817(01背包 +状态压缩)

题目的意思就是有个学校要招老师.要让没门课至少有两个老师可以上.

每个样样例先输入三个数字课程数量s,已经在任的老师数量,和应聘的老师数量.已经在任的一定要聘请.

首先先说怎么状态压缩,课程最多8门,但是每门课要有2个两个老师,所以状态要有16位,第一位和第二位代表第一门课.

一开始所有状态都是1.如果这门有一个老师可以上了.那这一位就变为了0,所以最终状态是0;


处理已经在任的老师的时候,直接把他们可以上的课全部标记完(这个就是初始状态).并记录下他们的工资(在最终结果加上就好).

然后开始动态规划,这其实是一个01背包,每个老师可以选择聘或者不聘;要比较两者.

不聘也就是dp(n - 1 , t)    (t是当前状态,也就表示当前状态不变,然后n- 1继续看下一个老师)

还设有聘dp(n - 1 , changesta(t , sta[n])) + cost[n]

这里的n - 1和前面一个意思,后面有一个函数changesta,也就是改变状态,t是当前状态,而sta[n]是之前算好的第n个老师的状态,里面也是用状态压缩存了他上了哪些课,然后用changesta将t里面sta[n]可以上的课也标记,并且加上这个老师的工资就行了.


AC代码:

#include
#include
#include 
using namespace std;

const int INF = 0x3f3f3f3f;
const int N = 105;

int s, n, m;
int f[N][1 << 17];
int cost[N], sta[N];

int changesta(int st, int x) {
	for(int i = 0; i < 2 * s; i += 2)
		if(x & (1 << i)) {
			int st1 = 1 << i;
			int st2 = 1 << (i + 1);

			if(st & st2) {
				st -= st2;
				continue;
			}
			if(st & st1) {
				st -= st1;
				continue;
			}
		}
	return st;
}

int dp(int n, int t) {
	if(t == 0)
		return 0;
	if(n == 0)
		return INF - 1;
	if(f[n][t] < INF)
		return f[n][t];
	return f[n][t] = min(f[n][t], min(dp(n - 1, t), dp(n - 1, changesta(t, sta[n])) + cost[n]));
}

int main() {
	while(scanf("%d%d%d",&s,&m,&n) && s) {
		int temp, sum = 0;
		int	t = (1 << (2 * s)) - 1;
		char ch;
		for(int i = 0; i < m; i++) {
			scanf("%d%c", &temp, &ch);
			sum += temp;
			while(ch != '\n') {
				scanf("%d%c", &temp, &ch);
				int t1 = 1 << (2 * temp - 1);
				int t2 = 1 << (2 * temp - 2);
				if(t & t1) {
					t -= t1;
					continue;
				}
				if(t & t2) {
					t -= t2;
					continue;
				}
			}
		}
		for(int i = 1; i <= n; i++) {
			scanf("%d%c", &temp, &ch);
			cost[i] = temp;
			sta[i] = 0;
			while(ch != '\n') {
				scanf("%d%c", &temp, &ch);
				int temp2 = 1 << (2 * temp - 2);
				sta[i] |= temp2;
			}
		}
		memset(f,INF,sizeof(f));
		printf("%d\n", dp(n, t) + sum);
	}
	return 0;
}



你可能感兴趣的:(DP)