[백준] 단계별로 풀어보기(단계:18,스택,JAVA)10773번, 제로

문제 링크

10773번: 제로

 

www.acmicpc.net

 

---

주의사항

• JAVA를 사용하여 프로그램을 사용하였습니다.
• 백준에서 코드를 작성하였을 때 아래 형태에서 Main에서 결과가 출력되어야 합니다.

public class Main{ 	
	public static void main(String[] args){
    }
}

---

문제 설명

---

접근 방법

기본적으로 스택이란 LIFO(후입 선출)의 자료구조입니다.

LIFO는 먼저 들어간 데이터들이 출력할 때 나중에 나온다는 이야기입니다.

예를 들어 3, 2, 1을 순서대로 스택에 저장한 뒤 하나씩 꺼내보겠습니다.

1. 스택에 3을 넣었을 때

3

2. 스택에 2을 넣었을 때

2

3

3. 스택에 1을 넣었을 때

1

2

3

4. 스택에 하나의 자료를 출력하라는 명령이 떨어졌을 때

1

2

3

5. 다시 자료를 출력하라는 명령이 떨어졌을 때

2

3

위에 표를 보면 간단히 돌탑을 생각하시고 돌을 뺄 때 나중에 올린 돌부터 빼는 것처럼 스택은 LIFO의 형태를 가진 자료구조입니다.

 

이 문제에서는 0이 입력되었을 때 스택을 pop을 진행하는 것입니다.

저는 ArrayList를 사용하여 스택에 형태를 구현하였습니다.

ArrayList는 자료를 저장하였을 때에 인덱스가 1씩 늘어나기 때문에 가장 최근에 들어간 자료에 인덱스는 ArrayList에 크기 - 1를 가지는 인덱스입니다.

그래서 pop 함수를 구현할 때 ArrayList에 값을 제거하는 인덱스를 ArrayList.size() - 1로 하였습니다.

다른 숫자일 경우 기본적인 ArrayList .add를 사용하여 스택에 추가하는 것처럼 구현하였습니다.

 

예제 입력1로 살펴보면

3 -> push 3

0 -> pop

4 -> push 4

0 -> pop

스택에 남은 값은 존재하지 않기 때문에 0이 출력됩니다.

• BufferedReader를 사용하여 입력 값을 받았습니다.
• 첫 줄에 받은 횟수만큼 for문을 통해 stack에 값을 조절하였습니다.
• 스택에 모든 값을 더하는 함수 stackSum를 구현하였습니다.
• 함수 결과를 BufferedWriter에 저장하였습니다.
• BufferedWriter을 사용하여 저장된 결과를 출력하였습니다.

• stackSum은 ArrayList.size()만큼 반복하여 모든 값을 더한 결과를 반환하도록 설계하였습니다.

결과 코드

import java.io.*;
import java.util.*;
public class Main{
	static List<Integer> list = new ArrayList<Integer>();
	public static void main(String[] args) throws IOException{
    	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        //BufferedReader를 통해 입력 값 받기
    	BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        //BufferedWriter를 통해 결과 값 출력
        //----------입력값 저장-----
    	int index = Integer.parseInt(br.readLine());
    	int result = 0;
    	for(int i=0;i<index;i++) {
        	int num = Integer.parseInt(br.readLine());
        	if(num==0)
        		list.remove(list.size()-1);	//0일 때 pop 진행
        	else
        		list.add(num);		//0이 아닐 때 push 진행
    	}
    	result = stackSum();	//함수 실행
    	bw.write(result + "\n");	//결과 BufferedWriter에 저장
    	bw.flush();		//결과 출력
    	bw.close();
    	br.close();
	}
    //---------스택에 값 모두 더하는 함수--------------
	public static int stackSum() {
		int result = 0;
		for(int i=0;i<list.size();i++)
			result += list.get(i);
		return result;
	}
}