uva1508(状态压缩 +枚举)

题意:
有n个5元组,要从中选k个,问在所选的k个中每一元的最大值相加最大是多少;
思路:
首先如果要选的组大于5个,那好办,每元都选一个最大的加起来就行了;
但是如果小于5个,那就要用状态压缩来枚举情况了;
我们知道,每一组数中,都有几个数字,可能是最后要相加的5个数;
那么我们就用状态压缩来表示这组数中哪几个被选进去,11111代表5个数字都进去了;

那么我们一开始的状态应该是31,也就是11111,现在这些位置都还空着;那假如现在你选择了一组数字,它的状态是10000,也就是它的第一个数字选进最后里面了,那么剩下继续递归的状态就是01111;但是已经去除10000这种状态了,那么所有n个数组中,那个数组在10000这种状态最大?所以我们应该实现算好,在所有状态下最大的是哪一组数:

AC代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int N = 100005;
int g[N][5];
int res[5];
int m[35];
int n,k;

void init() {
	memset(m , 0 ,sizeof(m));
	for(int i = 0; i < n; i++) {
		for(int s = 0 ; s < 32 ; s++) {
			int temp = 0;
			for(int k = 0 ; k < 5 ;k++) {
				if(s & (1 << k))
					temp += g[i][k];
				m[s] = max(m[s],temp);
			}
		}
	}
}
int search(int s ,int num) {
	if(num == 0)
		return 0;
	int temp = 0;
	for(int i = s ; i > 0 ; i = (i - 1)&s) {
		temp = max(temp , m[i] + search((i^s),num - 1));
	}
	return temp;
}
int main() {
	int t;
	scanf("%d",&t);
	while(t--) {
		int ans = 0;
		memset(res , 0 ,sizeof(res));
		scanf("%d%d",&n,&k);
		for(int i = 0; i < n;i++) {
			for(int j = 0; j < 5; j++) {
				scanf("%d",&g[i][j]);
				if(g[i][j] > res[j])
					res[j] = g[i][j];
			}
		}
		if(k >= 5) {
			int ans = 0;
			for(int i = 0; i < 5; i++) {
				ans += res[i];
			}
			printf("%d\n",ans);
			continue;
		}
		init();
		printf("%d\n",search(31 , k));
	}
}