[PS][KAKAO BLIND RECRUITMENT][2020][가사 검색]

prog-kakao-2020-가사검색-0.cpp

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

bool check(string a, string b) {
    if (a.size() != b.size()) return false;
    int size = a.size();
    for (int i = 0; i < size; i++) {
        if (b[i] == '?') continue;
        if (a[i] != b[i]) return false;
    }
    return true;
}

vector<int> solution(vector<string> words, vector<string> queries) {
    vector<int> answer;
    int qsize = queries.size();
    int wsize = words.size();
    for (int i = 0; i < qsize; i++) {
        answer.push_back(0);
        for (int j = 0; j < wsize; j++) {
            if (check(words[j], queries[i])) {
                answer[i]++;
            }
        }
    }
    return answer;
}

int main() {
    int n = 0;
    vector<string> word, query;
    vector<int> ans;
    for (int i = 0; i < 6; i++) {
        string s;
        cin >> s;
        word.push_back(s);
    }
    for (int i = 0; i < 5; i++) {
        string s;
        cin >> s;
        query.push_back(s);
    }
    ans = solution(word, query);
    int size = ans.size();
    for (int i = 0; i < size; i++) {
        cout << ans[i] << " ";
    }
    cout << "\n";
    return 0;
}