백준 15705번 : 단어 찾기

• Solved.ac 기준 : 실버 2
• 사용언어 C++

문제 해석 및 풀이

• 전 배열을 돌면서 8방향으로 찾아야 하는 글자가 있는지 여부 판별
• 배열의 원소와 첫 글자의 동일 여부를 판별하여 계산 줄이기

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

int dx[8] = { 0, 0, -1, 1, 1, 1, -1, -1 };
int dy[8] = { -1, 1, 0, 0, -1, 1, -1, 1 };

bool isWordFound(vector<string>& arr, int n, int m, string& str, int x, int y, int dir) {
    int len = str.length();
    for (int i = 0; i < len; ++i) {
        int nx = x + dx[dir] * i;
        int ny = y + dy[dir] * i;
        if (nx < 0 || nx >= n || ny < 0 || ny >= m || arr[nx][ny] != str[i]) {
            return false;
        }
    }
    return true;
}

void checkArr(vector<string>& arr, int n, int m, string& str) {
    int len = str.length();
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < m; ++j) {
            if (arr[i][j] == str[0]) {
                for (int dir = 0; dir < 8; ++dir) {
                    if (isWordFound(arr, n, m, str, i, j, dir)) {
                        cout << 1 << endl;
                        return;
                    }
                }
            }
        }
    }
    cout << 0 << endl;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    string s;
    cin >> s;
    int n, m;
    cin >> n >> m;

    vector<string> arr(n);
    for (int i = 0; i < n; ++i) {
        cin >> arr[i];
    }

    checkArr(arr, n, m, s);

    return 0;
}