c++/알고리즘

백준 9996

hojung 2022. 3. 19.
728x90
반응형

#include <iostream>
#include <string>
#include <algorithm>
using namespace std;
int n;
string pattern;
string filename;
string pre;
string back;
int main() {
    cin >> n;
    cin >> pattern;
    for(int i = 0; i < pattern.size(); i++) {
        if(pattern[i] == 42)
        {
            pre = pattern.substr(0, i-1);
            back = pattern.substr(i+1, pattern.size() - (i + 1));
        }
    } // 접두사 접미사 추출
    for(int i = 0; i < n; i++){
        cin >> filename;
        if(filename.substr(0, pre.size()) == pre && filename.substr(filename.size() - back.size()) == back) cout << "DA";
        else cout << "NE";

    }
    return 0;
}

나의 답

#include <iostream>
using namespace std;

int n;
string s, ori_s, pre, suf;
int main() {
    cin >> n;
    cin >> ori_s;
    int pos = ori_s.find('*');
    pre = ori_s.substr(0, pos);
    suf = ori_s.substr(pos+1);
    for(int i = 0; i < n ;i ++) {
        cin >> s;
        if(pre.size() + suf.size() > s.size()){
            cout << "NE\n";
        }
        else{
            if(pre == s.substr(0, pre.size() && suf == s.substr(s.size() - suf.size()))) cout << "DA\n";
            else cout << "NE\n";
        }
        
    }

    return 0;
}

정답 

나의 답과 정답의 차이점은 find함수의 사용유무이다. 문자열을 다루는 문제에서 find함수와 substr함수의 사용법을 익혀야겠다. 

 

728x90
반응형

'c++ > 알고리즘' 카테고리의 다른 글

백준2178번-미로BFS  (0) 2022.03.28
shell sort c++  (0) 2022.03.23
백준 1159  (0) 2022.03.18
c++ 행렬 곱셈  (0) 2022.03.16
피보나치 수열  (0) 2022.03.16

댓글