[ boj : 22232 ] 가희와 파일 탐색기

https://www.acmicpc.net/problem/22232

22232번: 가희와 파일 탐색기

해설 : 문제는 그냥 되게 간단한데 ㅋㅋㅋㅋㅋ 한가지 조건이라도 놓쳐버리면 고생할 문제네요.

연산자 오버로딩만 잘 어떻게 처리해주면 쉬운 문제입니다.

전 확장자가 둘다 붙거나, 둘다 붙지 않으면 3번째 조건으로 넘어가야 되는데 둘다 붙은 경우에만 생각해서 좀 해맸습니다.

연산자 오버로딩은 자유자재로 구현할 줄 알아야겠죵?

#include<bits/stdc++.h>

using namespace std;

const int SIZE = (int)2e5 + 1;

struct info {
	string str, front,back;
	bool flag=false;
}A[SIZE];

string str[SIZE];
set<string> st;

void solve();

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int t = 1;
	while (t--)solve();
}
void solve()
{
	int n, m;
	string input;
	cin >> n >> m;
	for (int i = 0; i < n; i++) {
		cin >> str[i];
		string x = str[i].substr(0, str[i].find("."));	
		string y = str[i].substr(str[i].find(".")+1, str[i].length());
		A[i].str = str[i];
		A[i].front = x;
		A[i].back = y;
	}
	for (int i = 0; i < m; i++) {
		cin >> input;
		st.insert(input);
	}
	for (int i = 0; i < n; i++) {
		if (st.find(A[i].back) != st.end())
			A[i].flag = true;
	}
	sort(A, A + n, [](const info x, const info y) {
		if (x.front == y.front) {
			if ((x.flag && y.flag) || (!x.flag && !y.flag)) {
				return x.back < y.back;
			}
			if (!x.flag)return false;
			else return true;
		}
		return x.front < y.front;
		});
	for (int i = 0; i < n; i++)
		cout << A[i].str << "\n";	
}