[백준/BOJ/C++] 5026번 박사 과정

[문제]

동혁이는 박사 학위 논문을 쓰던 중 두 수를 더하는 방법을 까먹었다. 동혁이는 덧셈 문제와 컴퓨터 과학 문제로 이루어진 문제지를 풀어야 군면제를 받을 수 있다.

문제지의 덧셈 문제는 "a+b"와 같은 형식이고, 컴퓨터 과학 문제는 "P=NP" 하나이다. 동혁이의 문제지가 주어졌을 때, 답을 모두 구하는 프로그램을 작성하시오.

[입력]

첫째 줄에 문제의 개수 N이 주어진다. (1 ≤ N ≤ 1000) 다음 N개 줄에는 "a+b"형식의 덧셈 문제나 "P=NP"가 주어진다. a,b ∈ [0,1000]이며 a와 b는 정수이다.

[출력]

P=NP가 문제인 경우에는 skipped를, 덧셈 문제인 경우에는 덧셈한 결과를 출력한다.

[Source Code]

#include <stdio.h>
#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
using namespace std;

int main() {
	int test;
	cin >> test;

	int *result = new int[test];

	for(int i = 0; i < test; i++){
		string input;
		cin >> input;
		if(input[1] == '=')
			result[i] = -1;

		else{
			int index = 0;
			string left, right;
			while(input[index] != '+'){
				left += input[index];
				index++;
			}
			for(int i = index+1; i < input.length(); i++){
				right += input[i];
			}

			result[i] = stoi(left) + stoi(right);
		}
	}

	for(int i = 0; i < test; i++){
		if(result[i] == -1)
			cout << "skipped\n";
		else
			cout << result[i] << "\n";
	}
}

[결과 화면]