'commit'

TangDou/Dfs/DFS4_XuanShu_ByNumber.cpp

@@ -0,0 +1,44 @@
+#include <bits/stdc++.h>
+
+using namespace std;
+const int N = 30;
+int a[N];
+int n, k;
+int cnt;
+
+bool isPrime(int n) {
+    if (n < 2) return false;
+    for (int i = 2; i <= n / i; i++)
+        if (n % i == 0) return false;
+    return true;
+}
+
+/**
+ * 功能:遍历每个卡片（数字）,携带已经选择了几个的m参数
+ * @param step  第几个卡片
+ * @param sum   已经选择的数字总和
+ * @param m     已经选择了几个数字
+ * @return      可行方法数
+ */
+void dfs(int step, int sum, int m) {
+    //已经完成选择k个数，判断总和是不是质数.如果是质数，说明找到了一种合法解
+    if (m == k && isPrime(sum)) {
+        cnt++;
+        return;
+    }
+    //这句需在放在后面,这是因为如果是第n个摆放完的数字，满足上面的判断条件，也是可以的
+    if (step == n + 1) return;
+
+    //选择当前数字,总和增大a[step],同时个数增加了1
+    dfs(step + 1, sum + a[step], m + 1);
+    //放弃当前数字,总和不变，个数也不变
+    dfs(step + 1, sum, m);
+}
+
+int main() {
+    cin >> n >> k;
+    for (int i = 1; i <= n; i++) cin >> a[i];
+    dfs(1, 0, 0);
+    cout << cnt << endl;
+    return 0;
+}