一、定义
枚举是基于逐个尝试答案的一种问题求解策略
二、算法
依次遍历每个值,直到找到问题求解答案
三、例题
1、完美立方
确定各个变量的范围:
a:[2,N]
b:[2,a-1]
c:[b,a-1]
d:[c,a-1]
确定了各个变量的范围,避免了不必要的运算,只需要使用四个for循环,就能求出问题解了,
题解代码如下:
public class Ex_1 {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
for (int a = 2;a <= n;++a)
for (int b = 2;b <= a-1;++b)
for (int c = b;c <= a-1;++c)
for (int d = c;d <= a-1;++d)
if(a*a*a == b*b*b + c*c*c + d*d*d)
System.out.println("Cube = "+a+", Triple = ("+b+","+c+","+d+")");
}
}
运行测试结果如下:
2、生理周期
**解题思路:**定义一个k,k++,当k同时满足 (k-p)%23 == 0、(k-e)%28 == 0、(k-i)%33 == 0
但是这样会去算很多不必要的数,比如,当找到第一个(k-p)%23 == 0时,我们就要去看k是否满足(k-e)%28 == 0和(k-i)%33 == 0,当后面两者不满足时,说明这个k不是我们要找的,我们就需要进行k++,但是在[k,k+23-1]之间的k一定不满足(k-p)%23 == 0,[k,k+23-1]之间的k相当于做了无意义的运算。所以,当找到第一个k满足第一个条件(k-p)%23 == 0时,在寻找满足第二个和第三个条件时,k++应该变成 k+=23;同理,当找到k满足前两个条件时,[k,k+23* 28-1]之间的数一定不满足前两个条件,所以k的步长应该从 k+=23变成 k+=23*28。经过这样的处理,可以避免大量不必要的计算。
题解代码:
public class Ex_2 {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int p = in.nextInt();
int e = in.nextInt();
int i = in.nextInt();
int d = in.nextInt();
int k;
for (k = d + 1;(k-p)%23 != 0;k++);
for ( ;(k-e)%28 != 0;k += 23);
for ( ;(k-i)%33 != 0;k += 23*28);
System.out.println(k-d);
}
}
测试结果:
3、称硬币
注:up表示右边轻,称量结果都用右边和左边比。
**解题思路:**依次对每一枚硬币,首先假设它是重的,然后代入三个称量结果,如果都符合,那么这枚硬币是重的;如果不符合,那么假设这枚硬币的轻的,然后代入三个称量结果,如果都符合,那么这枚硬币是轻的,如果还是不符合,则判断下一枚硬币。
题解代码:
public class Ex_3 {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
String[] left = new String[3]; //每一次称重左边的硬币
String[] right = new String[3]; //每一次称重右边的硬币
String[] result = new String[3]; //每一次称重的结果
while (n > 0){
//输入
for (int i = 0;i < 3;i++){
left[i] = in.next();
right[i] = in.next();
result[i] = in.next();
}
n--;
//遍历每一个硬币
for (char c = 'A';c <= 'L';c++){
if(isFake(c,false,left,right,result)){
//假设这枚硬币是重的
System.out.println(c+" is the counterfeit coin and it is weight");
break;
}else if(isFake(c,true,left,right,result)){
//假设这枚硬币是轻的
System.out.println(c+" is the counterfeit coin and it is light");
break;
}
}
}
}
public static boolean isFake(char c,boolean isLight,String[] left,String[] right,String[] result){
//isLight表示硬币是轻的还是重的
//返回true代表结果符合,返回false代表结果不符合
for (int i = 0;i < 3;i++){
if(left[i].indexOf(String.valueOf(c)) != -1){
//c在左边
//轻的
if(isLight){
if(!result[i].equals("down"))
return false;
}
//重的
else if(!isLight)
if(!result[i].equals("up"))
return false;
}else if(right[i].indexOf(String.valueOf(c)) != -1){
//c在右边
//轻的
if(isLight){
if(!result[i].equals("up"))
return false;
}
//重的
else if(!isLight)
if(!result[i].equals("down"))
return false;
}
}
return true;
}
}
注:isFake方法来判断硬币如果是真或假在称量结果中是否合理。
测试:
4、熄灯问题
**大体思路:**如果第一行的灯确定类怎么按,那么第二行的灯也就确定了怎么按。
要熄灭第一行第i列某个亮着的灯,唯一办法是按下第二行第i列的灯,如果按完最后一行,所有灯未熄灭,则枚举下一种第一行灯的按下方法,以此类推。
首先,第一行按灯情况有2^6 = 64种,将0-63化成6位二进制,就是这64种按灯情况,1代表按下。
枚举这64种第一行按灯情况,后面每一行根据前一行的熄灯情况去按灯,如果前一行第i列灯亮,则按下该行第i列的灯。最后,检查最后一行的熄灯情况,如果最后一行灯全灭,说明所有灯全灭,输出该按灯方案。
题解代码:
public static String toBinary(int num, int digits) {
int value = 1 << digits | num;
String bs = Integer.toBinaryString(value); //0x20 | 这个是为了保证这个string长度是6位数
return bs.substring(1);
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int[][] ori_light = new int[5][6]; //原始灯情况
//输入数据
for(int i = 0;i < 5;i++){
for (int j = 0;j < 6;j++){
ori_light[i][j] = sc.nextInt();
}
}
for (int i = 0;i < 64;i++){
int[][] result = new int[5][6]; //按灯方法
//第一行灯的按下情况
int[][] light = new int[5][6];
//拷贝数组
for (int x = 0;x < 5;x++){
light[x] = ori_light[x].clone();
}
String first = toBinary(i,6);
// System.out.println(i + " --- " + first);
for (int j = 0;j < 6;j++){
//按第一行
if(String.valueOf(first.charAt(j)).equals("1")){
result[0][j] = 1;
light[0][j] = light[0][j] == 0 ? 1 : 0;
//影响周围灯
if(j > 0){
light[0][j-1] = light[0][j-1] == 0 ? 1 : 0;
}
if(j < 5){
light[0][j+1] = light[0][j+1] == 0 ? 1 : 0;
}
light[1][j] = light[1][j] == 0 ? 1 : 0;
}
}
//按剩下二到五行
for (int k = 1;k < 5;k++){
//依据上一行情况按下一行
for (int l = 0;l < 6;l++){
if(light[k-1][l] == 0){
//上一行该列未熄灭,按本行该列
result[k][l] = 1;
light[k][l] = light[k][l] == 0 ? 1 : 0;
//影响周围灯
if(l > 0){
light[k][l-1] = light[k][l-1] == 0 ? 1 : 0;
}
if(l < 5){
light[k][l+1] = light[k][l+1] == 0 ? 1 : 0;
}
if(k < 4){
light[k+1][l] = light[k+1][l] == 0 ? 1 : 0;
}
light[k-1][l] = light[k-1][l] == 0 ? 1 : 0;
}
}
}
//最后一行全熄灭,按灯结束
if(light[4][0] + light[4][1] + light[4][2] + light[4][3] + light[4][4] + light[4][5] == 6){
for (int m = 0;m < 5;m++){
for (int n = 0;n < 6;n++){
System.out.print(result[m][n]+" ");
}
System.out.println();
}
// break;
System.out.println("==========================");
}
}
}
测试数据:
]+" “);
}
System.out.println();
}
// break;
System.out.println(”==========================");
}
}
}
**测试数据:**
[外链图片转存中...(img-8HNkoHN9-1611118056359)]
手工验证该熄灯方案,确实可以将所有灯熄灭。
原文链接:# 算法学习日记(一)-----枚举
## 一、定义
枚举是基于逐个尝试答案的一种问题求解策略
## 二、算法
依次遍历每个值,直到找到问题求解答案
## 三、例题
### 1、完美立方
![image-20210119105705034](https://img-picgo.oss-cn-hangzhou.aliyuncs.com/img/image-20210119105705034.png)
确定各个变量的范围:
a:[2,N]
b:[2,a-1]
c:[b,a-1]
d:[c,a-1]
确定了各个变量的范围,避免了不必要的运算,只需要使用四个for循环,就能求出问题解了,
题解代码如下:
```java
public class Ex_1 {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
for (int a = 2;a <= n;++a)
for (int b = 2;b <= a-1;++b)
for (int c = b;c <= a-1;++c)
for (int d = c;d <= a-1;++d)
if(a*a*a == b*b*b + c*c*c + d*d*d)
System.out.println("Cube = "+a+", Triple = ("+b+","+c+","+d+")");
}
}
运行测试结果如下:
2、生理周期
**解题思路:**定义一个k,k++,当k同时满足 (k-p)%23 == 0、(k-e)%28 == 0、(k-i)%33 == 0
但是这样会去算很多不必要的数,比如,当找到第一个(k-p)%23 == 0时,我们就要去看k是否满足(k-e)%28 == 0和(k-i)%33 == 0,当后面两者不满足时,说明这个k不是我们要找的,我们就需要进行k++,但是在[k,k+23-1]之间的k一定不满足(k-p)%23 == 0,[k,k+23-1]之间的k相当于做了无意义的运算。所以,当找到第一个k满足第一个条件(k-p)%23 == 0时,在寻找满足第二个和第三个条件时,k++应该变成 k+=23;同理,当找到k满足前两个条件时,[k,k+23* 28-1]之间的数一定不满足前两个条件,所以k的步长应该从 k+=23变成 k+=23*28。经过这样的处理,可以避免大量不必要的计算。
题解代码:
public class Ex_2 {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int p = in.nextInt();
int e = in.nextInt();
int i = in.nextInt();
int d = in.nextInt();
int k;
for (k = d + 1;(k-p)%23 != 0;k++);
for ( ;(k-e)%28 != 0;k += 23);
for ( ;(k-i)%33 != 0;k += 23*28);
System.out.println(k-d);
}
}
测试结果:
3、称硬币
注:up表示右边轻,称量结果都用右边和左边比。
**解题思路:**依次对每一枚硬币,首先假设它是重的,然后代入三个称量结果,如果都符合,那么这枚硬币是重的;如果不符合,那么假设这枚硬币的轻的,然后代入三个称量结果,如果都符合,那么这枚硬币是轻的,如果还是不符合,则判断下一枚硬币。
题解代码:
public class Ex_3 {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
String[] left = new String[3]; //每一次称重左边的硬币
String[] right = new String[3]; //每一次称重右边的硬币
String[] result = new String[3]; //每一次称重的结果
while (n > 0){
//输入
for (int i = 0;i < 3;i++){
left[i] = in.next();
right[i] = in.next();
result[i] = in.next();
}
n--;
//遍历每一个硬币
for (char c = 'A';c <= 'L';c++){
if(isFake(c,false,left,right,result)){
//假设这枚硬币是重的
System.out.println(c+" is the counterfeit coin and it is weight");
break;
}else if(isFake(c,true,left,right,result)){
//假设这枚硬币是轻的
System.out.println(c+" is the counterfeit coin and it is light");
break;
}
}
}
}
public static boolean isFake(char c,boolean isLight,String[] left,String[] right,String[] result){
//isLight表示硬币是轻的还是重的
//返回true代表结果符合,返回false代表结果不符合
for (int i = 0;i < 3;i++){
if(left[i].indexOf(String.valueOf(c)) != -1){
//c在左边
//轻的
if(isLight){
if(!result[i].equals("down"))
return false;
}
//重的
else if(!isLight)
if(!result[i].equals("up"))
return false;
}else if(right[i].indexOf(String.valueOf(c)) != -1){
//c在右边
//轻的
if(isLight){
if(!result[i].equals("up"))
return false;
}
//重的
else if(!isLight)
if(!result[i].equals("down"))
return false;
}
}
return true;
}
}
注:isFake方法来判断硬币如果是真或假在称量结果中是否合理。
测试:
4、熄灯问题
**大体思路:**如果第一行的灯确定类怎么按,那么第二行的灯也就确定了怎么按。
要熄灭第一行第i列某个亮着的灯,唯一办法是按下第二行第i列的灯,如果按完最后一行,所有灯未熄灭,则枚举下一种第一行灯的按下方法,以此类推。
首先,第一行按灯情况有2^6 = 64种,将0-63化成6位二进制,就是这64种按灯情况,1代表按下。
枚举这64种第一行按灯情况,后面每一行根据前一行的熄灯情况去按灯,如果前一行第i列灯亮,则按下该行第i列的灯。最后,检查最后一行的熄灯情况,如果最后一行灯全灭,说明所有灯全灭,输出该按灯方案。
题解代码:
public static String toBinary(int num, int digits) {
int value = 1 << digits | num;
String bs = Integer.toBinaryString(value); //0x20 | 这个是为了保证这个string长度是6位数
return bs.substring(1);
}
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int[][] ori_light = new int[5][6]; //原始灯情况
//输入数据
for(int i = 0;i < 5;i++){
for (int j = 0;j < 6;j++){
ori_light[i][j] = sc.nextInt();
}
}
for (int i = 0;i < 64;i++){
int[][] result = new int[5][6]; //按灯方法
//第一行灯的按下情况
int[][] light = new int[5][6];
//拷贝数组
for (int x = 0;x < 5;x++){
light[x] = ori_light[x].clone();
}
String first = toBinary(i,6);
// System.out.println(i + " --- " + first);
for (int j = 0;j < 6;j++){
//按第一行
if(String.valueOf(first.charAt(j)).equals("1")){
result[0][j] = 1;
light[0][j] = light[0][j] == 0 ? 1 : 0;
//影响周围灯
if(j > 0){
light[0][j-1] = light[0][j-1] == 0 ? 1 : 0;
}
if(j < 5){
light[0][j+1] = light[0][j+1] == 0 ? 1 : 0;
}
light[1][j] = light[1][j] == 0 ? 1 : 0;
}
}
//按剩下二到五行
for (int k = 1;k < 5;k++){
//依据上一行情况按下一行
for (int l = 0;l < 6;l++){
if(light[k-1][l] == 0){
//上一行该列未熄灭,按本行该列
result[k][l] = 1;
light[k][l] = light[k][l] == 0 ? 1 : 0;
//影响周围灯
if(l > 0){
light[k][l-1] = light[k][l-1] == 0 ? 1 : 0;
}
if(l < 5){
light[k][l+1] = light[k][l+1] == 0 ? 1 : 0;
}
if(k < 4){
light[k+1][l] = light[k+1][l] == 0 ? 1 : 0;
}
light[k-1][l] = light[k-1][l] == 0 ? 1 : 0;
}
}
}
//最后一行全熄灭,按灯结束
if(light[4][0] + light[4][1] + light[4][2] + light[4][3] + light[4][4] + light[4][5] == 6){
for (int m = 0;m < 5;m++){
for (int n = 0;n < 6;n++){
System.out.print(result[m][n]+" ");
}
System.out.println();
}
// break;
System.out.println("==========================");
}
}
}
测试数据:
手工验证该熄灯方案,确实可以将所有灯熄灭。