[프로그래머스 Level 1] 가운데 글자 가져오기

문제

가운데 글자 가져오기  

 

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

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

 

코드

#include <string>

using namespace std;

string solution(string s) 
{
    string answer;
    int mid = s.size() / 2;
    
    if(s.size() % 2 == 1)
        answer = s[mid];
    else
        answer = s.substr(mid-1, 2);
    
    return answer;
}

 

나의 생각

string substr을 잘 쓰자!