[C++][프로그래머스] 튜플

프로그래머스

 

📄 문제

셀수있는 수량의 순서있는 열거 또는 어떤 순서를 따르는 요소들의 모음을 튜플(tuple)이라고 합니다. n개의 요소를 가진 튜플을 n-튜플(n-tuple)이라고 하며, 다음과 같이 표현할 수 있습니다.

(a1, a2, a3, ..., an)
튜플은 다음과 같은 성질을 가지고 있습니다.

중복된 원소가 있을 수 있습니다. ex : (2, 3, 1, 2)
원소에 정해진 순서가 있으며, 원소의 순서가 다르면 서로 다른 튜플입니다. ex : (1, 2, 3) ≠ (1, 3, 2)
튜플의 원소 개수는 유한합니다.
원소의 개수가 n개이고, 중복되는 원소가 없는 튜플 (a1, a2, a3, ..., an)이 주어질 때(단, a1, a2, ..., an은 자연수), 이는 다음과 같이 집합 기호 '{', '}'를 이용해 표현할 수 있습니다.

{{a1}, {a1, a2}, {a1, a2, a3}, {a1, a2, a3, a4}, ... {a1, a2, a3, a4, ..., an}}
예를 들어 튜플이 (2, 1, 3, 4)인 경우 이는

{{2}, {2, 1}, {2, 1, 3}, {2, 1, 3, 4}}
와 같이 표현할 수 있습니다. 이때, 집합은 원소의 순서가 바뀌어도 상관없으므로

{{2}, {2, 1}, {2, 1, 3}, {2, 1, 3, 4}}
{{2, 1, 3, 4}, {2}, {2, 1, 3}, {2, 1}}
{{1, 2, 3}, {2, 1}, {1, 2, 4, 3}, {2}}
는 모두 같은 튜플 (2, 1, 3, 4)를 나타냅니다.

특정 튜플을 표현하는 집합이 담긴 문자열 s가 매개변수로 주어질 때, s가 표현하는 튜플을 배열에 담아 return 하도록 solution 함수를 완성해주세요.

 

📝 풀이

#include <string>
#include <vector>
#include <queue>
#include <set>

using namespace std;

vector<int> solution(string s) {
	s.erase(0, 1);
	s.erase(s.size() - 1, 1);

	priority_queue<pair<int, vector<string>>> q;

	for (int i = 0; i < s.size(); ++i)
		if (s[i] == '{')
		{
			++i;
			vector<string> v;
			string builder;

			for (; i < s.size(); ++i)
				if (s[i] == '}')
				{
					v.push_back(builder);
					q.push({ -(int)v.size(), v });
					break;
				}
				else if (s[i] == ',')
				{
					v.push_back(builder);
					builder.erase(0);
				}
				else
				{
					builder.push_back(s[i]);
				}
		}

	set<string> set;
	vector<int> answer;
	while (!q.empty())
	{
		vector<string> v = q.top().second;
		q.pop();

		for (int i = 0; i < v.size(); ++i)
			if (set.find(v[i]) == set.end())
			{
				answer.push_back(stoi(v[i]));
				set.insert(v[i]);
			}
			else
				continue;
	}

	return answer;
}

• set과 queue를 이용하여 문제를 해결합니다.
• 문자열을 추출하여 원소를 vector<string> v에 삽입한 후 이 값을 원소의 개수를 내림차순한 큐에 삽입합니다.
• 큐를 하나씩 뽑으면, 원소의 개수가 작은것부터 나오게 되며, 중복되지 않는 값을 set을 이용하여 찾습니다.
• 중복되는 값이 없으면, 해당 원소가 다음 answer에 삽입되고, 중복되면 건너뜁니다.