【JAVA】PAT 乙级 1005 继续(3n+1)猜想

当我们验证卡拉兹猜想的时候，为了避免重复计算，可以记录下递推过程中遇到的每一个数。例如对 n=3 进行验证的时候，我们需要计算 3、5、8、4、2、1，则当我们对 n=5、8、4、2 进行验证的时候，就可以直接判定卡拉兹猜想的真伪，而不需要重复计算，因为这 4 个数已经在验证3的时候遇到过了，我们称 5、8、4、2 是被 3“覆盖”的数。我们称一个数列中的某个数 n 为“关键数”，如果 n 不能被数列中的其他数字所覆盖。

现在给定一系列待验证的数字，我们只需要验证其中的几个关键数，就可以不必再重复验证余下的数字。你的任务就是找出这些关键数字，并按从大到小的顺序输出它们。

输入格式：
每个测试输入包含 1 个测试用例，第 1 行给出一个正整数 K (<100)，第 2 行给出 K 个互不相同的待验证的正整数 n (1<n≤100)的值，数字间用空格隔开。

输出格式：
每个测试用例的输出占一行，按从大到小的顺序输出关键数字。数字间用 1 个空格隔开，但一行中最后一个数字后没有空格。

输入样例：
6
3 5 6 7 8 11

输出样例：
7 6



这题和1001 害死人不偿命的(3n+1)猜想类似，可以使用之前写的计算步数的方法。
我用了一个set来放计算步数时出现过的数字，set里不会出现相同的数字
输出的时候，如果该数字没有出现在set里，就输出，否则不输出。

import java.util.Arrays;
import java.util.Scanner;
import java.util.Set;
import java.util.TreeSet;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int[] k = new int[n];
		for (int i = 0; i < n; i++) {
			k[i] = sc.nextInt();
			c(k[i], 0);//对k[i]进行验证猜想
		}
		Arrays.sort(k);
		int count = 0;
		for (int i = k.length - 1; i >= 0; i--) {
		//题目要求降序输出，所以排序后往前
			if (!s.contains(k[i])) {
			//k[i]没有在进行猜想过程中出现过
				if (count == 0) {
					System.out.print(k[i]);
				} else {
					System.out.print(" " + k[i]);
				}
				count++;
			}
		}
		

	}
	
	//用于存放出现过的数字
	static Set<Integer> s = new TreeSet<Integer>();
	
	//因为是直接复制1001的代码，所以是返回int，影响不大。
	static int c(int n, int count) {
		//相比1001的代码增加一个将出现过的数字放入s里的步骤
		if (count > 0) {
			s.add(n);
		}
		if (n == 1) {
			return count;
		}
		return n % 2 == 0 ? c(n / 2, ++count) : c((3 * n + 1) / 2, ++count);
	}
}