USACO 1.6.3 优质牛肋骨

1.6.3 优质牛肋骨

题目考查

枚举 dfs 剪枝

考虑到当$n==8$的时候, 会有$10^9$个数字, 这显然是无法接受的, 于是考虑优化.

我们考虑超级质数的特征: 开头的数字一定是质数. 因此开头的数字只有 { 2 , 3 , 5 , 7 } \{2, 3, 5, 7\} {2,3,5,7}. 接下来考虑剩余数位: 2的倍数一定不行, 5的倍数也一定不行, 因此只剩下 { 1 , 3 , 7 , 9 } \{1, 3, 7, 9\} {1,3,7,9}.

此时的最坏情况, 会产生$4^8$个数字, 再加上素数的判断, 大概是我们所能接受的复杂度.

题目细节

无

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
vector<int> res;
bool fact(int x) {
	for (int i = 2; i <= x / i; ++i) {
		if (x % i == 0) return 0;
	}
	return 1;
}
void dfs(int x, int n) {
	if (!n) {
		res.push_back(x);
		return;
	}

	int a[] = { 1, 3, 7, 9 };
	for (auto& op : a) {
		int qaq = x * 10 + op;
		if (fact(qaq)) dfs(qaq, n - 1);
	}
}
int main()
{
    int n; cin >> n;
	dfs(2, n - 1), dfs(3, n - 1);
	dfs(5, n - 1), dfs(7, n - 1);

	sort(res.begin(), res.end());
	for (auto& op : res) printf("%d\n", op);

    return 0;
}

END