DFS模板

浅浅写一下dfs。

目录

 基本模板：

题目一 路径计数：

 题目二  玩具蛇

题目三  N皇后问题

---

 基本模板：

一般情况输出的都是结果种数，所以有输出其他的自己再修改就可了。

int v[N]={0);    //用作标记

void dfs(int w){
    //判断越界与否
    if(){
        return;
    }
    //递归出口，合题的话计数加一
    if(){
        cnt++;
        return;
    }
    //普通情况，继续递归
    else{
        v[w]=1;    //标记本次遍历
        dfs(w++);
        v[w]=0;    //回溯
    }
}

int main(){
    //递归入口，注意判断入口有多少种可能  
    dfs(0);
    //打印结果
    printf("%d",cnt);
    return 0;
}

题目一 路径计数：

从一个 5×5 的方格矩阵的左上角出发，沿着方格的边走，满足以下条件的路线有多少种？

1. 总长度不超过 12；
2. 最后回到左上角；
3. 路线不自交；
4. 不走出 5×5 的方格矩阵范围之外。 如下图所示，ABC 是三种合法的路线。注意 B 和 C 由于方向不同，所以 视为不同的路线。

注：我们每次都是从左上角出发，所以入口只有一个，就不用再枚举了。 

#include<bits/stdc++.h>
using namespace std;

//递归方向数组，也可以写成两个
int d[4][2]={
    {1,0},
    {-1,0},
    {0,1},
    {0,-1}
};
int sum=0;
int v[5][5]={0};    //初始化标记数组

void dfs(int w,int x,int y){
    int i,tx,ty;

    //判断越界
    if( x<0 || y<0 || x>4 || y>4)
        return;
    if( w>13)
        return;

    //递归出口，计数加一
    if( w>3 && x==0 && y==0 && v[x][y]){
        sum++;
        return;
    }

    //普通情况，递归
    for(i=0;i<4;i++){
        tx=x+d[i][0];
        ty=y+d[i][1];
        if(!v[tx][ty]){
            v[tx][ty]=1;        //标记
            dfs(w+1,tx,ty);
            v[tx][ty]=0;        //回溯
        }
    }
}

int main(){
    int i;    

    //从（0，0）点开始进入递归
    dfs(1,0,0);

    //打印结果
    printf("%d",sum);

    return 0;
}

 题目二  玩具蛇

小蓝有一条玩具蛇，一共有 16 节，上面标着数字 1 至 16。每一节都是一个正方形的形状。相邻的两节可以成直线或者成 90 度角。

小蓝还有一个 4×4 的方格盒子，用于存放玩具蛇，盒子的方格上依次标着字母 A 到 P 共 16 个字母。

小蓝可以折叠自己的玩具蛇放到盒子里面。他发现，有很多种方案可以将玩具蛇放进去。

下图给出了两种方案：

请帮小蓝计算一下，总共有多少种不同的方案。如果两个方案中，存在玩具蛇的某一节放在了盒子的不同格子里，则认为是不同的方案。

 注：这道题的话，玩具蛇头可以放的位置有十六个，所以要注意先枚举一波递归入口。

#include<bits/stdc++.h>
using namespace std;

int d[4][2]={
    {1,0},
    {-1,0},
    {0,1},
    {0,-1}
};
int sum=0;
int v[4][4]={0};

void dfs(int w,int x,int y){
    int i,tx,ty;

    //越界条件
    if(x<0||y<0||x>3||y>3)
        return;
    
    //递归出口
    if(w==16){
        sum++;
        return;
    }

    //普通情况
    for(i=0;i<4;i++){
        tx=x+d[i][0];
        ty=y+d[i][1];
        if(!v[tx][ty]){
            v[tx][ty]=1;
            dfs(w+1,tx,ty);
            v[tx][ty]=0;
        }
    }
}

int main(){
    int i;  

    //递归入口  
    for(i=0;i<16;i++){
        v[i%4][i/4]=1;
        dfs(1,i%4,i/4);
        v[i%4][i/4]=0;
    }

    //打印结果
    printf("%d",sum);
    
    return 0;
}

题目三  N皇后问题

在 N×N 的方格棋盘放置了 N 个皇后，使得它们不相互攻击（即任意 2 个皇后不允许处在同一排，同一列，也不允许处在与棋盘边框成 45 角的斜线上。你的任务是，对于给定的 N，求出有多少种合法的放置方法。

输入描述

输入中有一个正整数 N≤10，表示棋盘和皇后的数量

输出描述

为一个正整数，表示对应输入行的皇后的不同放置数量。

输入输出样例

示例 1

输入

5

输出

10

以四乘四为例，每次从第一行开始进入递归

（0，0）位置被标记，dfs进行下一行 

 显然不合理，同列

 同对角线，也不合理

找到合题位置，继续dfs下一列，同上。

代码：

#include<bits/stdc++.h>
using namespace std;

int n,sum=0;
int a[11][11]={0};
int cc[250]={0};    //标记列
int ll[250]={0};    //标记行
int uu[250]={0};    //标记对角线

void dfs(int w){
    int i,j;

    //递归出口
    if(w==n){
        sum++;
        return;
    }

    //普通情况，进行递归
    for(i=0;i<n;i++){

        //判断是否合题
        if(!cc[i] && !ll[w+i] && !uu[w-i+n]){

            //标记
            a[w][i]=1;
            cc[i]=1;
            ll[w+i]=1;
            uu[n+w-i]=1;

            dfs(w+1);
            
            //回溯
            a[w][i]=0;
            cc[i]=0;
            ll[w+i]=0;
            uu[n+w-i]=0;
        }
    }
}

int main(){
    scanf("%d",&n);

    //递归入口
    dfs(0);

    //打印结果
    printf("%d",sum);

    return 0;
}

。