인프런 커뮤니티 질문&답변

안녕하세요..

제가 이문제를 다시 푸는 과정에서

reverse함수를 사용해서

풀어보았는데요

예제는 다 맞는데 왜틀리는지 모르겠습니다...

답변 부탁드립니다.

http://boj.kr/ee37f6b6daa14f158949673b90e65815

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

vector<string> solution(const int N, const string& str, const vector<string>& v)
{
    vector<string> answer;

    size_t pos = str.find('*');
    string pre = str.substr(0, pos);
    string suf = str.substr(pos + 1);
    reverse(suf.begin(), suf.end());

    // cout << "pre = " << pre << ", suf = " << suf << '\n'; // test
    for (const string& s : v)
    {
        string rs(s);
        reverse(rs.begin(), rs.end());
        if (s.find(pre) == 0 && rs.find(suf) == 0)
            answer.push_back("DA");
        else
            answer.push_back("NE");
        // cout << "str = " << s << "\nrstr = " << rs << '\n';
        // cout << "answer = " << answer.back() << "\n\n";
    }

    return answer;
}

int main()
{
    int N;
    string str;
    cin >> N >> str;

    vector<string> arr(N);
    for (string &s : arr)
        cin >> s;
    
    vector<string> answer = solution(N, str, arr);

    for (string& s : answer)
        cout << s << '\n';

    return 0;
}