[Java/Algorithm] Programmers: 가운데 글자 가져오기

문제 설명

단어 s의 가운데 글자를 반환하는 함수, solution을 만들어 보세요. 단어의 길이가 짝수라면 가운데 두글자를 반환하면 됩니다.

재한사항

• s는 길이가 1 이상, 100이하인 스트링입니다.

입출력 예

s	return
"abcde"	"c"
"qwer"	"we"

 

해법: 가운데 인덱스를 찾아서 (길이를 2로 나눠줍니다), 길이가 홀수일때, 그리고 짝수일때에 따라 substring 을 이용해서 새로운 스트링을 리턴합니다. substring function은 O(N)으로 알려져 있으므로, 이 솔루션의 Big O 는 O(N) 입니다.

class Solution {
    public String solution(String s) {
        if(s.length() == 0){
            return s;
        }
        int mid = s.length()/2;
        if(s.length() % 2 == 0){
            return s.substring(mid - 1, mid + 1);
        } else {
            return s.substring(mid, mid + 1);
        }
    }
}