<프로그래머스> Level1 가운데 글자 가져오기

가운데 글자 가져오기

문제

문제 설명

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

재한사항

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

문제 풀이

간단한 문제이다. s의 길이가 짝수일떄와 홀수일떄로 나누어서 짝수일때는 string 클래스의 substr()를 사용하여 문자 두개를 가져온다. 홀수일때는 인덱싱을 통하여 가져온다.

c++ 코드

#include <string>
#include <vector>
 
using namespace std;
 
string solution(string s) {
    string answer = "";
    int len = s.length();
    if(len%2==0){
        answer = s.substr(len/2-1,2);
    }else{
        answer = s[len/2];
    }
    return answer;
}