[codeforces] 코드포스 Asterisk-Minor Template 문제풀이(C++)

문자열 나와서 쫄았는데 예외처리 잘 생각하니까 어렵지 않게 풀 수 있었음

 

#include<iostream>
#include<vector>
#include<string>
#include<cstring>
 
using namespace std;
/*
<문제> - 2초, 128메가
a, b 두 문자열 - 소문자 Latin letters로 이루어짐
t는 소문자와 *로 이루어짐
별 수가 글자 수보다 작거나 같으면 asterisk-minor라 함
 
 
t의 각 별표를 소문자의 문자열로 바꿔서 s가 되면 s는 t에 매칭한다고 함
(별 수는 글자 수보다 작거나 같아야 함)
 
a와 b를 모두 매치시키는 asterisk-minor템플릿을 찾기
없으면 없다고 해야 함
답이 여러개면, 그중 아무거나 출력
 
<입력>
t    testcase(~10000)
a
b
(a, b는 문자열 / 각 50자 이하 소문자로 구성)
 
<출력>
YES or NO
YES인 경우, 가능한 템플릿 아래줄에 출력
 
<풀이>
1. 아이디어
 - 처음이나 끝이 같거나, 중간에 두글자만 같으면 됨
2. 시간
 - O(N^2) 인데, 문자열이랑 testcase가 작아서 2억5천번 정도 계산이면 2초안에 가능할 듯
3. 변수
 - 문자열 저장할 변수 두개
*/
int main() {
    int t; cin >> t;
    while (t--) {    //testcase
        string a, b; cin >> a >> b;
 
        if (a[0] == b[0]) {    //첫 글자가 같은경우
            cout << "YES" << endl << a[0] << "*" << endl;
            continue;
        }
        if (a[a.size() - 1] == b[b.size() - 1]) {    //끝 글자가 같은경우
            cout << "YES" << endl << "*" << a[a.size() - 1] << endl;
            continue;
        }
 
        bool flag = false;
        for (int i = 0; i < a.size() - 1; i++) {    //중간의 두 글자가 같은경우
            for (int j = 0; j < b.size() - 1; j++) {
                if (a[i] == b[j] && a[i + 1] == b[j + 1]) {
                    flag = true;
                    cout << "YES" << endl << "*" << a[i] << b[j + 1] << "*" << endl;
                    break;
                }
            }
            if (flag) break;
        }
        if (flag == false) cout << "NO" << endl;
 
    }
    return 0;
}