【DFS + 剪枝】POJ 1010 STAMPS

这段时间要沉迷刷题一段时间了，就让CSDN陪我一起吧！

一、题目大意

题目的大致意思就是有若干测试用例，每个测试用例包含两行输入，第一行是代表邮票不同种类的面值，第二行是代表顾客所需邮票的总面值，要求对于每个顾客所需邮票的总面值，给出不同种类邮票的组合来满足要求。当然组合的选择是有要求的，第一个选择条件是邮票的种类数应该尽可能的多，第二个选择条件是邮票的总张数应该尽可能的少，第三个选择条件是所选组合中的最大面值应该尽可能的大。

二、题目思路以及AC代码

一开始看到题目还以为很简单，毕竟因为是1010题嘛，这么靠前，我以为是道水题。结果… 呵呵，看了题解才明白怎么做。

主要的思路还是dfs，不过需要进行剪枝，不然会超时。基本思路就是dfs，然后记录每次的结果，再记录一个最优结果，每次对其进行比较，就可以了，这里提醒几个要剪枝的地方，其余的思路在代码中都已经有所体现。

1. 张数不能超过4
2. 定序剪枝，也就是dfs选择到第i个元素的时候，之前的元素不需要考虑，新的dfs只需从i开始就行了
3. 输入的时候，相同面值的邮票如果超过5种，只需存储5种就可以了，因为多了也没用。

如果想看思路的话，可以参见

https://blog.csdn.net/lyy289065406/article/details/6647948

下面给出AC代码：

#include <iostream>
#include <algorithm>
#define MAXN 30
using namespace std;

bool flag = false;				// 标记之前是否已经找到过解
bool flag_tie = false;			// 根据对solve和bestSolve的比较结果，标记是否tie

int value[MAXN];				// 记录各种邮票的面值
int sizev;						// sizev是对value的索引
int times[MAXN];				// times[i]记录第i种邮票在组合中出现的次数
int type[MAXN];					// type[i]记录输入同一面值的邮票出现几次

int solve[6];		// solve[0]存储邮票张数，solve[6]存储邮票种类，solve[1..4]存储邮票类别
int bestSolve[6];	// 记录solve的最优值

// 一些初始化操作
void init() {
	for (int i = 0; i < MAXN; i++) {
		times[i] = 0;
	}
	for (int i = 0; i < 6; i++) {
		solve[i] = 0;
		bestSolve[i] = 0;
	}
	flag = false;
	flag_tie = false;
}


// 将solve的值，更新到bestSolve 如果满足条件的话
void update(int num, int type) {
	bestSolve[0] = num;
	bestSolve[5] = type;

	for (int i = 1; i <= 4; i++) {
		bestSolve[i] = solve[i];
	}
}

// 求取数组s[1] - s[4]的最大值
int Max(int* s) {
	int m = -1;
	for (int i = 1; i <= 4; i++) {
		if (s[i] > m) m = s[i];
	}
	return m;
}

// dfs  need: 还需要凑的面值		num: 已经选择的邮票张数		
//		type: 已经选择的邮票种数	pre: 用于剪枝，当已经选择到pre的时候，前面如果要选择就早已选择了，所以可以直接从pre开始选择
void dfs(int need, int num, int type, int pre) {

	if (num == 5)			// 选择张数到5的时候，截断
		return;

	if (need == 0) {

		if (flag) {										// 如果已经有一个最优值
			if (bestSolve[5] == type) {					// 如果最优值种类数等于当前组合种类数
				if (bestSolve[0] == num) {				// 如果最优值张数等于当前组合张数
					int maxS = Max(solve);
					int maxBS = Max(bestSolve);
					if (maxBS == maxS) {				// 如果最优值面值最大值还等于当前组合面值最大值
						flag_tie = true;				// 就tie
						update(num, type);
					}
					else if (maxS > maxBS) {			// 否则
						flag_tie = false;
						update(num, type);
					}
				}
				else if (bestSolve[0] > num) {			// 否则
					flag_tie = false;
					update(num, type);
				}
			}
			else if (bestSolve[5] < type) {				// 否则
				flag_tie = false;
				update(num, type);
			}
		}
		else {											// 如果之前没有最优值，也就是这是找到的第一个组合
			flag = true;								// 直接更新其为最优值
			update(num, type);
		}

		return;
	}

	for (int i = pre; i < sizev; i++) {					// 遍历dfs
		if (need >= value[i]) {

			solve[num + 1] = value[i];

			if (times[i] != 0) {
				times[i]++;
				dfs(need - value[i], num + 1, type, i);
			}
			else {
				times[i]++;
				dfs(need - value[i], num + 1, type + 1, i);
			}

			times[i]--;
			solve[num + 1] = 0;
		}
		else
			return;
	}
	
	return;
}

int main() {

	while (true) {

		sizev = 0;
		for (int i = 0; i < MAXN; i++) {
			times[i] = 0;
			type[i] = 0;
		}

		while (true) {
			int tmp;

			if (scanf("%d", &tmp) == EOF) {
				exit(0);
			}
			if (!tmp) break;

			if (type[tmp] < 5) {					// 限制每个面值的邮票种数不能超过5，因为超过5没用
				value[sizev++] = tmp;
				type[tmp]++;
			}
		}

		// sort(value, value + sizev);					// 升序排列，有的题解中使用，但其实没必要，这采用的是定序剪枝
		int need;
		while (cin >> need && need) {
			init();
				
			dfs(need, 0, 0, 0);

			cout << need << " ";
			if (!flag) {
				cout << "---- none" << endl;
				continue;
			}
			cout << "(" << bestSolve[5] << "):";
			if (flag_tie) {
				cout << " tie" << endl; continue;
			}

			int size = bestSolve[0];
			sort(bestSolve + 1, bestSolve + size + 1);

			for (int i = 1; i < size + 1; i++) {
				cout << " " << bestSolve[i];
			}
			cout << endl;
		}
	}

	return 0;
}

如果有问题，欢迎大家指正！！！