牛客网字符迷阵

注意：本题允许使用C/C++/Java/python进行解答，其他编程语言提交均视作无效处理。

字符迷阵是一种经典的智力游戏。玩家需要在给定的矩形的字符迷阵中寻找特定的单词。
在这题的规则中，单词是如下规定的：
1. 在字符迷阵中选取一个字符作为单词的开头；
2. 选取右方、下方、或右下45度方向作为单词的延伸方向；
3. 以开头的字符，以选定的延伸方向，把连续得到的若干字符拼接在一起，则称为一个单词。
这里写图片描述

以图1为例，如果要在其中寻找单词”WORD”，则绿色框所标示的都是合法的方案，而红色框所标示的都是不合法的方案。
现在的问题是，给出一个字符迷阵，及一个要寻找的单词，问能在字符迷阵中找到多少个该单词的合法方案。注意合法方案是可以重叠的，如图1所示的字符迷阵，其中单词”WORD”的合法方案有4种。

输入描述:
输入的第一行为一个正整数T，表示测试数据组数。接下来有T组数据。每组数据的第一行包括两个整数m和n，表示字符迷阵的行数和列数。接下来有m行，每一行为一个长度为n的字符串，按顺序表示每一行之中的字符。再接下来还有一行包括一个字符串，表示要寻找的单词。数据范围：对于所有数据，都满足1<=T<=9，且输入的所有位于字符迷阵和单词中的字符都为大写字母。要寻找的单词最短为2个字符，最长为9个字符。字符迷阵和行列数，最小为1，最多为99。对于其中50%的数据文件，字符迷阵的行列数更限制为最多为20。

输出描述:
对于每一组数据，输出一行，包含一个整数，为在给定的字符迷阵中找到给定的单词的合法方案数。

3
10 10
AAAAAADROW
WORDBBBBBB
OCCCWCCCCC
RFFFFOFFFF
DHHHHHRHHH
ZWZVVVVDID
ZOZVXXDKIR
ZRZVXRXKIO
ZDZVOXXKIW
ZZZWXXXKIK
WORD
3 3
AAA
AAA
AAA
AA
5 8
WORDSWOR
ORDSWORD
RDSWORDS
DSWORDSW
SWORDSWO
SWORD

本题思路，两种方法：

1、直接遍历，每行搜索，每列搜索，每个斜对角搜索，这个方法写的好长，最后才20分，实在不向找Bug了，用第二种

2、深度优先搜索：先找到这个单词的第一个字母，然后向三个方向遍历，代码比第一种少，而且一次ac。。。

#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
int n, m, src_len;
int result;
string src;
vector<vector<char> >v_c;

void dp(int ii, int jj, int &result, int next_point, int dir){
    //找到了首字母，向三个方向寻找
    if(next_point==src_len) {
        result ++;
        return ;
    }
    else{
        //向右边走
        if(jj+1<m && v_c[ii][jj+1] == src[next_point] && dir==1) dp(ii, jj+1, result, next_point+1, 1);
        //向下边走
        if(ii+1<n && v_c[ii+1][jj] == src[next_point] && dir==2) dp(ii+1, jj, result, next_point+1, 2);
        //向斜下方走
        if(ii+1<n && jj+1<m && v_c[ii+1][jj+1] == src[next_point] && dir==3) dp(ii+1, jj+1, result, next_point+1, 3);
    }
}
int main()
{
    int T;
    cin>>T;
    while(T--){
        result=0;
        cin>>n>>m;
        v_c.clear();
        for (int i = 0; i < n; ++i) {
            vector<char> vv_c;
            for (int j = 0; j < m; ++j) {
                char c;
                cin>>c;
                vv_c.push_back(c);
            }
            v_c.push_back(vv_c);
        }

        cin>>src;
        src_len = src.length();
        for (int i = 0; i < n; ++i) {
            for (int j = 0; j < m; ++j) {
                //找到了首字母
                if(v_c[i][j]==src[0]) {
                    // 1、2、3代表的三个方向，因为方向是固定的
                    dp(i, j, result, 1, 1);
                    dp(i, j, result, 1, 2);
                    dp(i, j, result, 1, 3);
                }
            }
        }
        cout<<result<<endl;
    }
}
/*
3 3
ABC
ABC
ABC
ABC
 */