[프로그래머스/lv1]가운데 글자/c++

가운데 글자

 

문제 설명

 

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

 

문제 풀이

 

기초 문제라서 substr을 이용해 풀었다.

 

#include <string>
#include <vector>

using namespace std;

string solution(string s) {
    string answer = "";
    int num = s.size();
    int di = num/2;
    if(num % 2 ==1){
        answer = s.substr(di,1);
    }
    else{
        answer = s.substr(di-1,2);
    }
    return answer;
}