[백준 1181: 단어 정렬] - C++

www.acmicpc.net/problem/1181

1181번: 단어 정렬

#include<iostream>
#include<string>
#include<algorithm>
#include<vector>

using namespace std;

bool compare(string a, string b)
{
	if (a.length() == b.length())
	{
		return a < b;
	}
	return a.size() < b.size();
}
int main()
{
	int n;
	cin >> n;

	string input;
	vector<string> arr;

	for (int i = 0; i < n; i++)
	{
		cin >> input;

		if (find(arr.begin(), arr.end(), input) == arr.end())
		{
			arr.push_back(input);
		}
	}


	sort(arr.begin(), arr.end(), compare);

	for (int i = 0; i < arr.size(); i++)
	{
		cout << arr[i] << '\n';
	}

	return 0;
}

 

 

* 정렬테스트는 '\n'를 신경쓰자