[BaeKJoon/C++] 백준 5026 c++ 박사 과정

https://www.acmicpc.net/problem/5026

5026번: 박사 과정

 

문제

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

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

입력

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

출력

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

예제 입력 1 복사

4
2+2
1+2
P=NP
0+0

예제 출력 1 복사

4
3
skipped
0

코드

#include <iostream>
#include <string>
#include <stdlib.h>
 
using namespace std;
 
int main() {
 
    int test;
    cin>>test;
 
    for(int i=0; i<test; i++){
 
        string str;
        cin>>str;
 
        int a,b;
 
        if(str=="P=NP"){
            cout<<"skipped"<<"\n";
        }
 
        else{
            int sever=str.find('+');
            a=stoi(str.substr(0,sever)); 
            b=stoi(str.substr(sever+1));
            // 문자열 추출
            cout<<a+b<<"\n";
        }
        
    }
}