프로그래머스/C++ flag에 따라 다른 값 반환하기

#include <string>
#include <vector>

using namespace std;

int solution(int a, int b, bool flag) {
    int answer = 0;
    if(flag == true){
        answer = a + b;
    }
    else{
        answer = a-b;
    }
    
    return answer;
}