7701. 염라대왕의 이름 정렬

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWqU0zh6rssDFARG&categoryId=AWqU0zh6rssDFARG&categoryType=CODE

SW Expert Academy

 

이 문제는 Comparator를 사용하면 쉽게 풀 수 있는 문제이다.

Java의 경우 Set으로 중복되는 문자열을 제거했고,

C++의 경우 Unique 함수를 사용해 중복되는 문자열을 제거했다.

<JAVA>

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

class MyStringComparator implements Comparator<String> {
	@Override
	public int compare(String o1, String o2) {
		if (o1.length() == o2.length()) {
			return o1.compareTo(o2);
		} else {
			if (o1.length() < o2.length()) {
				return -1;
			} else
				return 1;
		}
	}

}

class Solution {

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int T = Integer.parseInt(br.readLine());
		Set<String> s = new HashSet<>();
		for (int t = 1; t <= T; ++t) {
			s.clear();
			int N = Integer.parseInt(br.readLine());
			for (int i = 0; i < N; ++i) {
				s.add(br.readLine());
			}
			List<String> list = new ArrayList<>();

			for (String a : s) {
				list.add(a);
			}
			MyStringComparator myStringComparator = new MyStringComparator();
			Collections.sort(list, myStringComparator);
			System.out.println("#" + t);
			for (int i = 0; i < list.size(); ++i) {
				System.out.println(list.get(i));
			}
		}
	}
}

<C++>

#include<iostream>
#include<vector>
#include<algorithm>
#include<string>
using namespace std;
int T, N,i,len;
vector<string> s;
string temp;

bool a(string s1,string s2)
{
	if (s1.length() == s2.length())
	{
		return s1 < s2;
	}
	else
	{
		return s1.length() < s2.length();
	}
}
int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);
	
	cin >> T;
	for (int testcase = 1; testcase <= T; ++testcase)
	{
		s.clear();
		cin >> N;
		for (i = 0; i < N; ++i)
		{
			cin >> temp;
			s.push_back(temp);
		}
		sort(s.begin(), s.end(),a);
		s.erase(unique(s.begin(), s.end()), s.end());
		cout << "#" << testcase<<"\n";
		len = s.size();
		for (i = 0; i < len; ++i)
		{
			cout << s[i] << "\n";
		}
	}
}