튜플 C++ 2019 KAKAO Winter Internship

문제 설명

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

• (a1, a2, a3, ..., an)

튜플은 다음과 같은 성질을 가지고 있습니다.

1. 중복된 원소가 있을 수 있습니다. ex : (2, 3, 1, 2)
2. 원소에 정해진 순서가 있으며, 원소의 순서가 다르면 서로 다른 튜플입니다. ex : (1, 2, 3) ≠ (1, 3, 2)
3. 튜플의 원소 개수는 유한합니다.

원소의 개수가 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 함수를 완성해주세요.

[제한사항]

• s의 길이는 5 이상 1,000,000 이하입니다.
• s는 숫자와 '{', '}', ',' 로만 이루어져 있습니다.
• 숫자가 0으로 시작하는 경우는 없습니다.
• s는 항상 중복되는 원소가 없는 튜플을 올바르게 표현하고 있습니다.
• s가 표현하는 튜플의 원소는 1 이상 100,000 이하인 자연수입니다.
• return 하는 배열의 길이가 1 이상 500 이하인 경우만 입력으로 주어집니다.

---

[입출력 예]

s	result
"{{2},{2,1},{2,1,3},{2,1,3,4}}"	[2, 1, 3, 4]
"{{1,2,3},{2,1},{1,2,4,3},{2}}"	[2, 1, 3, 4]
"{{20,111},{111}}"	[111, 20]
"{{123}}"	[123]
"{{4,2,3},{3},{2,3,4,1},{2,3}}"	[3, 2, 4, 1]

입출력 예에 대한 설명

입출력 예 #1

문제 예시와 같습니다.

입출력 예 #2

문제 예시와 같습니다.

입출력 예 #3

(111, 20)을 집합 기호를 이용해 표현하면 {{111}, {111,20}}이 되며, 이는 {{20,111},{111}}과 같습니다.

입출력 예 #4

(123)을 집합 기호를 이용해 표현하면 {{123}} 입니다.

입출력 예 #5

(3, 2, 4, 1)을 집합 기호를 이용해 표현하면 {{3},{3,2},{3,2,4},{3,2,4,1}}이 되며, 이는 {{4,2,3},{3},{2,3,4,1},{2,3}}과 같습니다.

구현

level2 치고는 아주 쉬운문제에 속한다.

우선 string으로 들어오는 s를 순서대로 잘라서 vector에 따로 저장해준다.
저장해준 vector를 순서대로 size의 순서가 적은대로 sorting한 뒤 하나씩 넣어주면 끝이다.

vector<pair<int, vector<int>>> current;

for (int i = 2; i < s.size() - 1; i++) {
	if (s[i] - '0' >= 0 && s[i] - '0' < 10) {
		num *= 10;
		num += (s[i] - '0');
	}
	else if (s[i] == ',') {
		if (s[i - 1] == '}') {
			continue;
		}
		temp.push_back(num);
		num = 0;
	}
	else if (s[i] == '}') {
		temp.push_back(num);
		current.push_back(make_pair(temp.size(), temp));
		num = 0;
		temp.clear();
	}
}

우선 string s를 새로운 vector<pair<int, vector<int>>> current에 저장해줄 것이다.

위와 같은 코드를 통하여 저장을 해준다.

1. 숫자가 올 경우 num값에 해당 숫자를 int로 변환하여 저장해준다.
2. ,가 왔을 경우 앞에 }가 온다면 다음 집합으로 넘어가는 것이기에 무시해주고
   }가 오지않는다면 다음 숫자로 넘어가는 것이기에 숫자를 temp vector에 저장해주고 num에 넣어놓은 숫자는 초기화해준다.
3. }가 왔을 경우에는 temp vector에 저장해놓은 숫자를 current에 pair를 만들어 순서대로 저장해준다.
   이후 temp vector를 초기화 시켜준다.

이후에 current vector를 sort 시켜주면 pair를 만들어 주었기 때문에 size가 오름차순인 순서로 sorting이 된다.

---

위 과정까지 완료 되면

s	current
"{{2},{2,1},{2,1,3},{2,1,3,4}}"	"{{[1],[2]},{[2],[2,1]},{[3],[2,1,3]},{[4],[2,1,3,4]}}"

와 같은 형식으로 저장될 것이다.

set<int> res;
for (pair<int, vector<int>> p : current) {		// current와 같은 형식의 pair를 만들어 돌아준다.
	for (int num : p.second) {			// num은 current를 도는 p의 두번째를 돌아주면서
		if (res.find(num) == res.end()) {	// num이 set에 없다면
			res.insert(num);		// set에 num을 저장해준다.
			answer.push_back(num);		// 이후 num을 answer vector에 저장해준다.
		}
	}
}

이후 저장된 current를 돌면서 위와 같은 방식으로 숫자를 순서대로 저장해준다.

---

위에 나오는 다음 코드는

if (res.find(num) == res.end())

set의 find 함수를 이용하는 방법인데 인자로 들어오는 num값이 set에 있으면 해당 인자가 있는 위치의 iterator를 반환하고 없을 경우 end()를 반환한다.

---

전체 코드는 아래에서 볼 수 있다.

#include <string>
#include <vector>
#include <set>
#include <algorithm>

using namespace std;

vector<int> solution(string s) {
	vector<int> answer;
	int num = 0;
	vector<int> temp;
	vector<pair<int, vector<int>>> current;
	for (int i = 2; i < s.size() - 1; i++) {
		if (s[i] - '0' >= 0 && s[i] - '0' < 10) {
			num *= 10;
			num += (s[i] - '0');
		}
		else if (s[i] == ',') {
			if (s[i - 1] == '}') {
				continue;
			}
			temp.push_back(num);
			num = 0;
		}
		else if (s[i] == '}') {
			temp.push_back(num);
			current.push_back(make_pair(temp.size(), temp));
			num = 0;
			temp.clear();
		}
	}
	sort(current.begin(), current.end());

	set<int> res;
	for (pair<int, vector<int>> p : current) {
		for (int num : p.second) {
			if (res.find(num) == res.end()) {
				res.insert(num);
				answer.push_back(num);
			}
		}
	}
	return answer;
}