67. 找单词

题目描述

给一个字符串和一个二维字符数组，如果该字符串存在于该数组中，则按字符串的字符顺序输出字符串每个字符所在单元格的位置下标字符串，如果找不到返回字符串“N”。

1.需要按照字符串的字符组成顺序搜索，且搜索到的位置必须是相邻单元格，其中“相邻单元格”是指那些水平相邻或垂直相邻的单元格。

2.同一个单元格内的字母不允许被重复使用。

3.假定在数组中最多只存在一个可能的匹配。

输入描述

第1行为一个数字N指示二维数组在后续输入所占的行数。

第2行到第N+1行输入为一个二维大写字符数组，每行字符用半角,分割。

第N+2行为待查找的字符串，由大写字符组成。

二维数组的大小为N*N，0<N<=100。

单词长度K，0<K<1000。

输出描述

输出一个位置下标字符串，拼接格式为：第1个字符行下标+”,”+第1个字符列下标+”,”+第2个字符行下标+”,”+第2个字符列下标… +”,”+第N个字符行下标+”,”+第N个字符列下标。

用例

输入	4 A,C,C,F C,D,E,D B,E,S,S F,E,C,A ACCESS
输出	0,0,0,1,0,2,1,2,2,2,2,3
说明	ACCESS分别对应二维数组的[0,0] [0,1] [0,2] [1,2] [2,2] [2,3]下标位置。

思路

dfs

#include <iostream>
#include <vector>
#include <string>
#include <sstream>
using namespace std;

int N;
vector<vector<char>> matrix;
vector<vector<bool>> used;
string word;

bool dfs(int i, int j, int k, vector<string>& path)
{
    if (i < 0 || i >= N || j < 0 || j >= N || matrix[i][j] != word[k] || used[i][j])
    {
        return false;
    }

    path.push_back(to_string(i) + "," + to_string(j));

    if (path.size() == word.size())
    {
        return true;
    }
    used[i][j] = true;

    bool res = dfs(i - 1, j, k + 1, path) || dfs(i + 1, j, k + 1, path) || dfs(i, j - 1, k + 1, path) || dfs(i, j + 1, k + 1, path);

    if (!res)
    {
        path.pop_back();
        used[i][j] = false;
    }
    return res;
}

int main()
{
    cin >> N;
    cin.ignore();
    vector<vector<char>> temp(N);
    for (int i = 0; i < N; i++)
    {
        string str;
        getline(cin, str);
        istringstream ss(str);
        while (getline(ss, str, ','))
        {
            temp[i].push_back(str[0]);
        }
    }
    matrix.swap(temp);

    getline(cin, word);
    used = vector<vector<bool>>(N, vector<bool>(matrix[0].size(), false));

    for (int i = 0; i < N; i++)
    {
        for (int j = 0; j < N; j++)
        {
            vector<string> path;
            if (dfs(i, j, 0, path))
            {
                for (int k = 0; k < path.size(); k++)
                {
                    if (k == 0)
                        cout << path[k];
                    else
                        cout << "," << path[k];
                }
                return 0;
            }
        }
    }
    cout << "N" << endl;
    return 0;
}