[C++] 문자열 배열에서 세번째로 긴 문자열 출력

문제) 주어진 문자열 배열에서 세번째로 긴 문자열 출력하라.

지난 번 폰 인터뷰에 나왔던 문제이고, 이전 포스팅인 '배열에서 두번째로 작은 수를 구하기'와 거의 유사하다. 다만, string class를 좀 제대로(?) 사용해보고자 다시 해봤다.

#include <iostream>
#include <string>

using namespace std;

void swap(string *a, string *b) {
	string t;
	t = *a;
	*a = *b;
	*b = t;
}

string thirdLongestString(string *str, int len) {
	int last = 0;
	for (int i = 1; i < len; i++) {
		if (str[last].size() < str[i].size()) {
			swap(str[last], str[i]);
		}
	
		for (int j = last; j > 0; j--) {
			if (str[j].size() > str[j-1].size()) {
				swap(str[j], str[j-1]);
			} else {
				break;
			}
		}

		if (last <= 2) {
			last++;
		}
	}
	
	return str[2];	
}

int main() {
	int num = 0; 

	while (num < 3) {
		cin >> num;
	}
	
	string str[num];
	
	cin.ignore();	// to ignore '\n'
	
	for (int i = 0; i < num; i++) {
		getline(cin, str[i]);
	}

	cout << "3rd Longest String: " << thirdLongestString(str, num) << endl;

	for (int i = 0; i < num; i++) {
		cout << i << ": " << str[i] << endl;
	}

	return 0;
}