P1706 全排列问题（dfs）、P1036 [NOIP2002 普及组] 选数（dfs）

题目描述

按照字典序输出自然数 1 到 n 所有不重复的排列，即 n 的全排列，要求所产生的任一数字序列中不允许出现重复的数字。

输入格式

一个整数 n。

输出格式

由 1∼n 组成的所有不重复的数字序列，每行一个序列。

每个数字保留 55 个场宽。

输入输出样例

输入 #1复制

3

输出 #1复制

1 2 3

1 3 2

2 1 3

2 3 1

3 1 2

3 2 1

说明/提示

1<n≤9。

题解：

用dfs，如图从第一层的根节点依次遍历到每一列的最后一层，遍历完后从下一个根节点依次遍历，直到根节点到n。然后输出该组排列数

代码如下:

#include<iostream>
#include<math.h>
#include<algorithm>
#include<queue>
using namespace std;
int ans[20],book[20],n;
void dfs(int a){
    if(a>n){
        for(int i=1;i<=n;i++)
            printf("%5d",ans[i]);
            cout<<endl;
        return;
    }
    for(int i=1;i<=n;i++){
        if(!book[i]){
            ans[a] = i; book[i] = 1;
            dfs(a+1);
            book[i] = 0;
        }
        
    }
}
int main(){
scanf("%d",&n);
    dfs(1);
    return 0;
}

题目描述

已知 n 个整数x1,x2,⋯,xn，以及 1 个整数 k（k<n）。从 �n 个整数中任选 �k 个整数相加，可分别得到一系列的和。例如当 n=4，k=3，4个整数分别为 3,7,12,19时，可得全部的组合与它们的和为：

3+7+12=223+7+12=22

3+7+19=293+7+19=29

7+12+19=387+12+19=38

3+12+19=343+12+19=34

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3+7+19=293+7+19=29。

输入格式

第一行两个空格隔开的整数 n,k（1≤n≤20，k<n）。

第二行 n 个整数，分别为 x1,x2,⋯,xn（1≤xi≤5×10^6）。

输出格式

输出一个整数，表示种类数。

输入输出样例

输入 #1复制

4 3

3 7 12 19

输出 #1复制

1

题解：

1. 先写出求质数的函数

int prime(int p) {         
    if (p < 2)
        return 0;
    for (int i = 2; i * i <= p; i++)
        if (p % i == 0)
            return 0;
    return 1;
}

2. 深度优先求解组合数目

void dfs(int step, int sum, int left) {                
    int i;
    if (step == k) {           
        if (prime(sum))
            cnt++;
        return;
    }
    
    for (i = left; i < n; i++)                
        dfs(step + 1, sum + a[i], i + 1);
    return;
}

AC代码：

#include<stdio.h>
 
int n, k, a[25], cnt;               
int prime(int p) {         
    if (p < 2)
        return 0;
    for (int i = 2; i * i <= p; i++)
        if (p % i == 0)
            return 0;
    return 1;
}
           
void dfs(int step, int sum, int left) {                
    int i;
    if (step == k) {           
        if (prime(sum))
            cnt++;
        return;
    }
    
    for (i = left; i < n; i++)                
        dfs(step + 1, sum + a[i], i + 1);
    return;
}
 
int main() {
    int i;
    scanf("%d%d", &n, &k);
    for (i = 0; i < n; i++)
        scanf("%d", &a[i]);           
    dfs(0, 0, 0);                
    printf("%d", cnt);
    return 0;
}