취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 10105번 문제인 Assigning Partners이다.
문제는 아래 링크를 확인하자.

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

각 학생 s에 대하여 그 학생의 짝이 누구인지를 나타내는 map<string,string>형 맵 mp을 만들면 문제를 편하게 풀 수 있다.

위와 같은 맵 mp가 있을 때, 학생 s의 짝이 자기자신이 들어왔는지 여부는 s == mp[s] 값을 확인하는 것으로, s의 짝의 짝이 자기 자신이 맞는지 확인하는 것은 s == mp[mp[s]]의 값을 확인하는 것으로 알아낼 수 있다.

이를 이용해 문제를 해결하자.

아래는 제출한 소스코드이다.

#include <iostream>
#include <vector>
#include <string>
#include <map>
using namespace std;

int N;
string arr[30];
map<string, string> mp;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	cin >> N;
	for (int i = 0; i < N; i++) cin >> arr[i];
	for (int i = 0; i < N; i++) {
		string s; cin >> s;
		mp.insert(make_pair(arr[i],s));
	}

	bool chk = 1;
	for (int i = 0; i < N; i++) {
		if (arr[i] != mp[arr[i]] && arr[i] == mp[mp[arr[i]]]) continue;
		chk = 0;
	}

	if (chk) cout << "good";
	else cout << "bad";
}