Baekjoon 13420 사칙연산 JAVA

13420번: 사칙연산

문제

사칙연산은 덧셈, 뺄셈, 곱셈, 나눗셈으로 이루어져 있으며, 컴퓨터 프로그램에서 이를 표현하는 기호는 +, -, *, / 와 같다. 아래는 컴퓨터 프로그램에서 표현한 사칙 연산의 예제이다.

3 * 2 = 6

문제와 답이 주어졌을 때, 이를 계산하여 올바른 수식인지 계산하는 프로그램을 만들려고 한다. 만약 주어진 데이터가 3 * 2 = 6 이라면 정답으로, 3 * 2 = 7 이면 오답으로 채점을 하면 된다. 문제와 답이 주어졌을 때, 이를 채점하는 프로그램을 작성하시오.

입력

입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 구성된다. 입력의 첫 번째 줄에 테스트 케이스의 개수를 나타내는 자연수 T가 주어진다. 각 테스트 케이스의 첫 번째 줄에는 수식이 주어진다. 수식은 문자와 기호가 공백으로 구분되어 주어지고, 사칙연산 기호는 1개만 사용된다. 나눗셈의 경우 항상 나누어떨어지는 경우로만 주어진다.

출력

출력은 표준 출력을 사용한다. 입력받은 데이터에 대해, 각 테스트 케이스의 답을 순서대로 1줄에 1개씩 출력한다. 주어진 수식이 정답일 경우 “correct”를, 오답일 경우 “wrong answer”를 출력한다. 문제의 정답이 32비트 정수가 넘어갈 수 있다. 모든 범위가 부호가 있는 64비트 정수 이내이다.

풀이

입력 받는 값에 대한 연산을 직접적으로 해주고, 입력받은 답과 비교하여 정답일시 correct, 오답일 경우 wrong answer을 출력한다.

A operator B = ans 형태로 입력되기에 A operator B ans를 각각 입력받고

operator를 이용하여 사칙연산을 하고 연산의 결과와 입력받은 ans를 비교하여 동일한 경우 correct, 다를 경우 wrong answer를 출력하면 된다.

코드

import java.util.*;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int n=sc.nextInt();
		for(int i=0;i<n;i++) {
			long a=sc.nextLong();
			String op=sc.next();
			long b=sc.nextLong();
			String t=sc.next();
			long c=sc.nextLong();
			long ans=0;
			if(op.equals("+"))
				ans=a+b;
			else if(op.equals("-"))
				ans=a-b;
			else if(op.equals("*"))
				ans=a*b;
			else if(op.equals("/"))
				ans=a/b;
			t=ans==c?"correct":"wrong answer";
			System.out.println(t);
		}
	}
}