码迷,mamicode.com
首页 > 其他好文 > 详细

uva 1508 Equipment(暴力+枚举子集)

时间:2015-02-04 16:41:19      阅读:144      评论:0      收藏:0      [点我收藏+]

标签:c语言   uva   

                             uva 1508 Equipment


题目大意给出n个5元组,要求从中选取k个,要求5个位置上的数的最大值的和尽量大。

解题思路:一开始没思路,看别人题解过的。5位可以组成32种情况,在DFS前预处理,找出32种情况在n组数据中的最大值,然后将这组数据带入DFS中枚举计算。


#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
using namespace std;
int num[10005][6], temp[40], ans, n, k;
void DFS(int pos, int sum, int d) {
	if (d == k) {
		ans = max(sum, ans);
		return;
	}
	for (int i = pos; i; i = (i - 1) & pos) {
		DFS(pos^i , sum + temp[i], d + 1);
	}
	return;
}
int main() {
	int T;
	scanf("%d\n", &T);
	while (T--) {
		memset(num, 0, sizeof(num));
		memset(temp, 0, sizeof(temp));
		scanf("%d %d\n", &n, &k);
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < 5; j++) {
				scanf("%d", &num[i][j]);
				temp[j] = max(temp[j], num[i][j]);
			}
		}
		if (k >= 5) {
			printf("%d\n", temp[0] + temp[1] + temp[2] + temp[3] + temp[4]);
			continue;
		}
		memset(temp, 0, sizeof(temp));
		for (int i = 0; i < n; i++) {
			for (int pos = 0; pos < 32; pos++) { //5位32种情况
				int sum = 0;
				for (int j = 0; j < 5; j++) {
					if (pos & (1<<j)) { 
						sum += num[i][j];
					}
				}
				temp[pos] = max(temp[pos], sum); //32种情况,每种情况的最大值
			}
		}
		ans = 0;
		DFS(31, 0, 0);
		printf("%d\n", ans);
	}
	return 0;
}



 



uva 1508 Equipment(暴力+枚举子集)

标签:c语言   uva   

原文地址:http://blog.csdn.net/llx523113241/article/details/43486319

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!