알고리즘/문자열
[C++] 백준 9996번 한국이 그리울 땐 서버에 접속하지
키베이루
2022. 12. 28. 17:05
1) 문제설명
https://www.acmicpc.net/problem/9996
9996번: 한국이 그리울 땐 서버에 접속하지
총 N개의 줄에 걸쳐서, 입력으로 주어진 i번째 파일 이름이 패턴과 일치하면 "DA", 일치하지 않으면 "NE"를 출력한다. 참고로, "DA"는 크로아티어어로 "YES"를, "NE"는 "NO"를 의미한다.
www.acmicpc.net
2) 아이디어
문자열을 분리하는 split을 사용하여 *을 기준으로 앞부분을 prefix, *의 뒷부분을 suffix로 나누어
prefix가 입력받는 문자열(result)의 앞 부분에 위치하고 suffix가 입력받는 문자열(result)의 뒷부분에 위치한다면 DA를 출력 그렇지 않다면 NE를 출력하였다.
★ prefix와 suffix로 나눈 부분이 입력받는 문자열(result)보다 작다면 런타임 오류가 나니 예외처리를 하여야 한다.
3) 코드
#include<iostream>
#include<vector>
#include<algorithm>
#include<cstdio>
using namespace std;
int main() {
int n;
string s;
cin >> n;
cin >> s;
int pivot;
for (int i = 0; i < s.length(); i++) {
if (s[i] == '*') {
pivot = i;
}
}
string prefix = s.substr(0, pivot);
string suffix = s.substr(pivot + 1, s.length());
// as*as
// asas
for (int i = 0; i < n; i++) {
string result;
cin >> result;
if (prefix.length() + suffix.length() > result.length()) {
cout << "NE" << endl;
}
else if (prefix == result.substr(0, pivot) && suffix == result.substr(result.length()-suffix.length(), result.length())) {
cout << "DA" << endl;
}
else {
cout << "NE" << endl;
}
}
}