[프로그래머스] Lv. 0 접미사 배열 JAVA

접미사 배열

 

문제 설명

요구사항

• my_string의 앞글자를 하나씩 제거하여 모든 접미사 구하기
• 모든 접미사 중복을 제거 하여 사전순으로 정렬하여 리턴

 

테스트

package lv0;

import java.util.Iterator;
import java.util.TreeSet;

public class 접미사_배열 {
	public String[] solution(String my_string) {
        String[] answer = {};
        
        TreeSet<String> set = new TreeSet<String>(); // 중복이 제거 되며 오름차순으로 정렬되는 컬렉션
        
        for(int i = 0; i < my_string.length(); i++) {
        	set.add(my_string.substring(i));
        }
        
        answer = new String[set.size()];
        int index = 0;
        
        Iterator<String> it = set.iterator();
        
        while(it.hasNext()) {
            answer[index++] = it.next();
        }
        
        return answer;
    }
}

 

프로그래머스

import java.util.Iterator;
import java.util.TreeSet;

class Solution {
    public String[] solution(String my_string) {
        String[] answer = {};
        
        TreeSet<String> set = new TreeSet<String>();
        
        for(int i = 0; i < my_string.length(); i++) {
        	set.add(my_string.substring(i));
        }
        
        answer = new String[set.size()];
        int index = 0;
        
        Iterator<String> it = set.iterator();
        
        while(it.hasNext()) {
            answer[index++] = it.next();
        }
        
        return answer;
    }
}

 

결과