알고리즘/문자열

[C++] 백준 9996번 한국이 그리울 땐 서버에 접속하지

키베이루 2022. 12. 28. 17:05

 

1) 문제설명

백준 Online Judge 9996번

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;
		}
	}

}