C++ 算法篇 穷举（枚举）

 穷举

计算机的特点之一就是运算速度快，善于重复做一件事。“穷举”正是基于这一特点的最古老的算法。它一般是在一时找不出解决问题的更好途径，即从数学上找不到求解的公式或规则时，根据问题中的“约束条件”，将解的所有可能情况一一列举出来，然后再逐个验证是否符合整个问题的求解要求，从而求得问题的可行解或者最优解。

 

一、 穷举算法的应用举例

 

例1、楼层编号

【问题描述】

小林在 NOIP 比赛期间住在“新世界”酒店。和其他酒店不一样的是，这个酒店每天都有一个高能的数字 t，这个数字在楼层中是不会出现的，以 t=3 为例，则 3、13、31、33 等楼层是不存在的，楼层编号为 1，2，4，5，…所以实际上的 4 楼才是 3 楼。

已知小林预订了编号为 m 层的房间，并且当天高能数字是 t，现在他想知道房间所在的真实楼层是多少。

【输入格式】

一行两个整数 m 和 t，1≤m≤100000，0≤t≤9，保证 m 对 t 合法。

【输出格式】

一行一个整数，表示真实楼层。

【输入样例】

14 3

【输出样例】

12

 

【问题分析】

        根据题意，只要从 1~m 穷举楼层编号，将所有含高能数字 t 的楼层计数存储在 ans 中，最后的答案就是 m-ans。

 

 例2、火柴棒等式 

【问题描述】

给出 n 根火柴棒，可以拼出多少个形如“A+B=C”的等式？

等式中的 A、B、C 是用火柴棒拼出的整数（若该数非零，则最高位不能是 0）。用火柴棒拼数字 0~9 的拼法如图 9.7-1 所示。

 

需要注意以下几点：

（1） 加号与等号各自需要两根火柴棒。

（2） 如果 A ≠ B，则 A+B=C 与 B+A=C 视为不同的等式（A、B、C 均大于或等于 0）。

（3） n 根火柴棒必须全部用上（n≤24）。

【输入样例】

14

【输出样例】

2

【样例说明】

两个等式分别为：0+1=1 和 1+0=1。

 

【问题分析】

首先，预处理每个数字（0~9）需要用几根火柴棒，存储在数组 f 中。然后，穷举 a 和 b，算出它们的和 c，再判断下列约束条件是否成立：f （a）+ f （b）+ f （c）= n-4。现在的问题是：a 和 b 的范围有多大？可以发现尽量用数字 1 拼成的数比较大，分析可知最多不会超过 1111。程序实现时，分别用三个循环语句预处理好所有两位数、三位数、四位数构成所需要的火柴棒数量。

 

 

 例3、比例简化 

【问题描述】

在社交媒体上，经常会看到针对某一个观点同意与否的民意调查以及结果。例如，对某观点表示支持的有 1498 人，反对的有 902 人，那么其比例可以简单地记为1498∶902。

因该比例的数值太大，难以一眼看出它们的关系。若把比例记为 5∶3，虽然与真实结果有一定的误差，但依然能够较为准确地反映调查结果，同时也显得比较直观。

现给出支持人数 A 和反对人数 B，以及一个上限 L，请将 A 比 B 化简为 A′ 比 B′，要求在 A′和 B′ 均不大于 L，且 A′ 和 B′ 互质（两个整数的最大公约数为 1）的前提下，A′/B′≥ A/B 且 A′/B′-A/B 的值尽可能小。

 

【输入格式】

一行三个整数 A，B，L，每两个整数之间用一个空格隔开，分别表示支持人数、反对人数以及上限。

【输出格式】

一行两个整数 A′ 和 B′，中间用一个空格隔开，表示化简后的比例。

【输入样例】

1498 902 10

【输出样例】

5 3

【数据规模】

对于 100% 的数据满足：1≤A≤10 6 ，1≤B≤10 6 ，1≤L≤100，A/B≤L。

 

【问题分析】

首先，答案为一个分数，而且分子，分母都小于或等于 L。所以，可以直接穷举分子 i （对应题目中的 A′）和分母 j （对应题目中的 B′）。结合题目的具体要求分析：

（1） i，j≤L，这个可以作为穷举的范围。

（2） i/j≥A/B，且 i/j-A/B 的值尽量小。前者只要转换成 i*B≥j*A，后者可以“打擂台”实现。

假设用K1/K2表示最后的答案，初值设置为1000000/1，min=abs （k1*B-k2*A）。如果abs （i/j-A/B）< abs（K1/K2-A/B），即如果 abs（i*B-j*A）< abs（k1*B-k2*A），则更新 K1、K2 和 min。

（3） 答案的分子、分母要互质，这个只要从小到大穷举 i、从大到小穷举 j，第一个符合条件的答案肯定就是最简分数。假设 L=10，如果先穷举到 1/5 得到一个答案，后面的 2/10 是不会更新答案的。

 

 例4、奶牛碑文 

【问题描述】

小伟暑假期间到大草原旅游，在一块石头上发现了一些有趣的碑文。碑文似乎是一个神秘古老的语言，只包括三个大写字母 C、O 和 W。尽管小伟看不懂，但是令他高兴的是，C、O、W的顺序形式构成了一句他最喜欢的奶牛单词“COW”。现在，他想知道有多少次 COW 出现在文本中。

如果 COW 内穿插了其他字符，只要 COW 字符出现在正确的顺序，小伟也不介意。甚至，他也不介意出现不同的 COW 共享一些字母。例如，CWOW 出现了 1 次 COW，CCOW 算出现了2 次 COW，CCOOWW 算出现了 8 次 COW。

 

【输入格式】

第 1 行为 1 个整数 N。

第 2 行为 N 个字符的字符串，每个字符是一个 C、O 或 W。

【输出格式】

输出 COW 作为输入字符串的字串出现的次数（不一定是连续的）。

提示：答案会很大，建议用 64 位整数（long long）。

【输入样例】

6

COOWWW

【输出样例】

6

【数据规模】

对于 50% 的数据满足：N≤60。

对于 100% 的数据满足：N≤105 。

 

【问题分析】

因为只有 3 个字母，所以可以穷举字符串中的每一个“O”，假设位置 i，然后分别计算其左边“C” 的个数 l[i] 和右边“W” 的个数 r[i]，再利用乘法原理进行计数 l[i]*r[i]，每次把答案累加到 ans 中。

 

 

二、 穷举算法的优化

穷举算法的特点是算法设计、实现都相对简单，但时间复杂度和空间复杂度往往较大。因此，用穷举算法解决问题时，往往需要尽量优化算法，从而减少穷举的次数，提高穷举的效率。

穷举算法优化的思路主要有结合约束条件，通过数学推导，减少穷举的范围和数量；通过预处理（如部分和、是否素数等），以空间换时间，避免在穷举过程中重复计算或判断等。

 

 

 例5、三角形个数 

 

【问题描述】

输入一根木棒的长度 n，1≤n≤10000，将该木棒分成三段，每段的长度为正整数，输出由该三段小木棒组成的不一样的三角形个数。

【输入样例】

10

【输出样例】

2

【样例说明】

两个能组成的三角形边长分别为 2、4、4 和 3、3、4。

 

【问题分析】

穷举三角形三条边长（假设为 a、b、c）的可能值，判断能否构成一个三角形，若能则计数，最后输出计数器的值。为了保证组成的三角形不重复，只要在穷举时设定 1≤a≤b≤c≤n-2。优化思想很简单但很重要，“能算不举”，穷举两条边，根据木棒长度直接计算出第三条边长。

 

  

例6、阿姆斯特朗数 

【问题描述】

编程找出所有的三位数到七位数中的阿姆斯特朗数。阿姆斯特朗数也叫水仙花数，它的定义如下：若一个 n 位自然数的各位数字的 n 次方之和等于它本身，则称这个自然数为阿姆斯特朗数。例如，153（153=1×1×1+3×3×3 +5×5×5）是一个三位的阿姆斯特朗数，8208 则是一个四位的阿姆斯特朗数。

【问题分析】

由于阿姆斯特朗数是没有规律的，只能采用穷举法一一验证 100~9999999 内的所有数是否是阿姆斯特朗数，若是，则打印之。但是，如果对任意一个数 K，都去求它的各位的若干次方，再求和判断是否等于K，效率比较差。注意到，每个位只可能是0~9，而且只会算到3~7次方。所以，为了使得程序尽快运行出正确结果，采用“以空间换时间”的策略，使用一个数组 p 预处理出所有数字的各次幂之值，p[i,j]表示 i 的 j 次方。另外，为了避免每次都对一个数进行逐位分解操作，直接用数组 a[8]存储一个数的每一位，穷举 100~9999999。

 

 例7、金币 

【问题描述】

国王将金币作为工资，发放给忠诚的骑士。第一天，骑士收到一枚金币；之后两天（第二天和第三天）里，每天收到两枚金币；之后三天（第四、五、六天）里，每天收到三枚金币；之后四天（第七、八、九、十天）里，每天收到四枚金币……这种工资发放模式会一直这样延续下去。当连续 n天每天收到 n 枚金币后，骑士会在之后的连续 n+1 天里，每天收到 n+1 枚金币（n 为正整数）。

请编程确定从第一天开始的给定天数内，骑士一共获得了多少金币。

【输入格式】

输入包含至少一行，但不多于 1000 行。

 

除最后一行外，输入的每行是一组输入数据，包含一个正整数 n，表示天数。

输入的最后一行为 0，表示输入结束。

【输出格式】

对每个数据输出一行一个整数，表示该数据对应总金币数。

【输入样例】

10

6

7

11

15

16

100

10000

1000

21

22

0

【数据规模】

对于 60% 的数据满足：n≤103 ；

对于 80% 的数据满足：n≤106 ；

对于 100% 的数据满足：n≤1012 。

 

【问题分析】

每次穷举每天获得的金币数，或者将答案保存在数组中直接输出，可以获得部分分。下面利用数学推导进行优化。

因为：1 2 +2 2 +3 2 +…+k 2 = k ×（k+1）×（2k+1）/ 6，假设前 n 个数为 1，2，2，3，3，3，…，k，k，k，k，如何求 k 呢？根据 1+2+3+…+k = n，即 k 2 +k-2n = 0，把 k 看成未知数，解方程可求出 k =（-1 + sqrt （1+8n））/ 2 = sqrt （2n+0.25）- 0.5，另一个负数解不合法舍去。

 

 

三、 二进制穷举思想

对于二进制数 00000，00001，00010，…，11111，它们是严格递增有序的，如何生成类似的二进制数字序列，就是“二进制穷举”思想，其应用非常广泛。“二进制穷举”的实现代码如下:

 

#include<cstdio>
using namespace std;
int main(){
	int n;
	int b[100];
	scanf( “ %d ” ,&n);
	for(int i = 0; i <= n; i++) b[i] = 0;
	while(b[0] == 0){
		for(int i = 1; i <= n; i++) printf( “ %d ” ,b[i]);
		printf( “ \n ” );
		int j = n;
		while(b[j] == 1) j--;
		b[j]++;
		for(int i = j + 1; i <= n; i++) b[i] = 0;
	}
	return 0;
}

 

例8、0-1 背包问题

【问题描述】

有 n 件物品，每件物品有一个重量和一个价值，分别记为 W1 ，W2 ，…，Wn 和 C1 ，C2 ，…，Cn 。现在有一个背包，其容量为 WK，要从 n 件物品种任取若干件，要求：

（1） 重量之和小于或等于 WK。

（2） 价值之和最大。

【输入格式】

第 1 行 2 个整数，表示 n和WK，1≤n≤20，1≤WK≤106 。

第 2 行 n 个整数，表示每一个物品的重量，1≤Wi ≤104 。

第 3 行 n 个整数，表示每一个物品的价值，1≤Ci ≤108 。

【输出格式】

一行一个数，表示符合背包容量的最大价值。

【输入样例】

8 200

79 58 86 11 28 62 15 68

83 14 54 79 72 52 48 62

【输出样例】

334

 

【问题分析】

背包问题有很多类型，本题是最简单的一种模型“0-1 背包”，即每件物品只有不取和取两种情况，对应着二进制中的 0 和 1。0-1 背包有多种解决算法，具体在第 13 课详细讲解。这里采用二进制穷举思想，从 000…00 穷举到 111…11，即定义一个数组 b，元素 b[i] = 0 或者 1，表示第 i 件物品不取或者取，每次判断背包能否装的下，同时对于价值打擂台取最大值。

 

  

例9、组合数的生成 

【问题描述】

从 1、2、3、4、5、6 这 6 个数字中任取 4 个数的组合有1 2 3 4、1 2 3 5、1 2 3 6、1 2 4 5、1 2 4 6、1 2 5 6、1 3 4 5、1 3 4 6、1 3 5 6、1 4 5 6、2 3 4 5、2 3 4 6、2 3 5 6、2 4 5 6、3 4 5 6，共 15 种。若把它们看成 4 位数，发现是递增的。

编程，输入 n 和 r，1≤r≤n≤20，按照以上顺序，输出从 n 个数字（1~n）中任取 r 个数的所有组合。

【输入样例】

3 2

【输出样例】

1 2

1 3

2 3

 

【问题分析】

观察第一个数到最后一个数的变化规律，可以看出最后一位先变化（加 1），变到 n 就要“进位”到上一位；对于倒数第二位，变化到 n-1 就要产生“进位”……每次“进位”后，本位及以后的所有位都要重新置成一个递增的序列。这个算法的思路就是二进制穷举思想，不过不是固定的“n 进制”。