SWEA 8931 제로
문제 정보
- 문제 출처 : SW Expert Academy
- 문제 번호 : 8931
- 문제 제목 : 제로
- 문제 난이도 : D3
- 풀이 언어 : Java
문제
재현이는 재민이를 도와서 동아리 장부를 관리하고 있다. 재현이는 영수증을 모아서 동아리의 지출 금액을 세고 있고, 재민이는 재현이가 부르는 액수를 순서대로 적고 있다. 재현이는 가끔 잘못된 수를 부르는 실수를 하는데, 이 때마다 0을 외쳐서, 가장 최근에 재민이가 쓰고 지우지 않았던 수를 지우게 시킨다.
재현이가 모든 수를 부른 후 재민이가 받아 적은 수의 합은 무엇인지 구하는 문제
접근
스택(Stack)을 사용하여 문제를 해결하였다.
풀이
- 첫 줄에는 테스트 케이스 개수 T 가 주어진다.
- 각 테스트 케이스 첫 줄에는 정수 K 가 주어진다.
- 이후 K개의 줄에 정수가 하나씩 주어진다.
- 각 테스트 케이스에 대해, Stack 타입의 stack을 생성한다.
- 변수 sum을 0으로 초기화한다.
- for 루프를 사용하여 K번 연산을 수행한다. 각 반복에서 숫자 num을 입력 받는다.
- num이 0이 아니면, 스택에 num을 추가하고 sum에 num을 더한다.
- num이 0이면, 스택이 비어있지 않은 경우 스택의 최상단 요소를 제거하고 sum에서 그 값을 뺀다.
- 각 테스트 케이스의 결과를 #”테스트 케이스 번호” 형식으로 출력한 후, 최종 sum 값을 출력한다.
후기
자주 쓰이는 자료구조 스택(Stack)을 사용한 문제이다. 다른 유형의 문제도 더 풀어볼 계획이다.
코드
Java
import java.util.Scanner;
import java.io.FileInputStream;
import java.util.Stack;
class Solution
{
public static void main(String args[]) throws Exception
{
Scanner sc = new Scanner(System.in);
int T;
T=sc.nextInt();
for(int test_case = 1; test_case <= T; test_case++)
{
Stack<Integer> stack = new Stack<>();
int sum = 0;
int K = sc.nextInt();
for (int i = 0; i < K; i++) {
int num = sc.nextInt();
if (num != 0) {
sum += stack.push(num);
} else if (!stack.isEmpty()) {
sum -= stack.pop();
}
}
System.out.print("#" + test_case + " ");
System.out.println(sum);
}
}
}