알고리즘 연습 심화 - 3

조재형·2023년 5월 25일
0

스터디

목록 보기
18/19

1번 문제 - (https://www.acmicpc.net/problem/10828)

import java.util.Scanner;
public class Prac_Java_3_4 {

    public static int[] stack;
    public static int size = 0;


    public static void main(String[] args) {

        Scanner in = new Scanner(System.in);
        StringBuilder sb = new StringBuilder();

        int N = in.nextInt();

        stack = new int[N];

        for(int i = 0; i < N; i++) {

            String str = in.next();

            switch (str) {

                case "push":
                    push(in.nextInt());
                    break;

                case "pop":
                    sb.append(pop()).append('\n');
                    break;

                case "size":
                    sb.append(size()).append('\n');
                    break;

                case "empty":
                    sb.append(empty()).append('\n');
                    break;

                case "top":
                    sb.append(top()).append('\n');
                    break;
            }

        }
        System.out.println(sb);
    }

    public static void push(int item) {
        stack[size] = item;
        size++;
    }

    public static int pop() {
        if(size == 0) {
            return -1;
        }
        else {
            int res = stack[size - 1];
            stack[size - 1] = 0;
            size--;
            return res;
        }
    }

    public static int size() {
        return size;
    }

    public static int empty() {
        if(size == 0) {
            return 1;
        }
        else {
            return 0;
        }
    }

    public static int top() {
        if(size == 0) {
            return -1;
        }
        else {
            return stack[size - 1];
        }
    }

}

2번 문제 - (https://www.acmicpc.net/problem/10773)

import java.util.*;
public class Prac_Java_3_5 {
    public static void main(String[] args){

        Stack<Integer> stack = new Stack<Integer>();
        Scanner scan = new Scanner(System.in);



        int K = scan.nextInt();	// K 정수 입력(입력 개수)

        for(int i = 0; i < K; i++) {
            int number = scan.nextInt();	// 정수 입력

            if(number == 0) {	// 0이라면 스택에 저장된 top 원소를 지운다.
                stack.pop();
            }
            else {	// 아닐경우는 스택에 원소 추가
                stack.push(number);
            }
        }

        int sum = 0;
        for(int o : stack) {	// 스택에 담긴 원소 합 구하기
            sum += o;
        }
        System.out.println(sum);
    }

}

3번 문제 - (https://www.acmicpc.net/problem/18258)

/*
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.*;

public class Prac_java_3_6 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        Integer cycle = Integer.parseInt(sc.nextLine());
        LinkedList<Integer> queue = new LinkedList<>();
        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < cycle ; i++) {
            // push 1
            String[] commands = sc.nextLine().split(" ");
            // push
            String command = commands[0];

            if (command.equals("push")) {
                // 1
                queue.add(Integer.parseInt(commands[1]));
            } else if (command.equals("pop")) {
                if (queue.isEmpty()) {
                    // queue가 비어있으면 -1 출력
                    sb.append("-1\n");
                } else {
                    // queue가 차있으면 queue에 값을 넣고 그 수를 출력
                    sb.append(queue.poll() + "\n");
                }
            } else if (command.equals("size")) {
                sb.append(queue.size() + "\n");
            } else if (command.equals("empty")) {
                if (queue.isEmpty()) {
                    sb.append("1\n");
                } else {
                    sb.append("0\n");
                }
            } else if (command.equals("front")) {
                if (queue.isEmpty()) {
                    sb.append("-1\n");
                } else {
                    sb.append(queue.peekFirst() + "\n");
                }
            } else if (command.equals("back")) {
                if (queue.isEmpty()) {
                    sb.append("-1\n");
                } else {
                    sb.append(queue.peekLast() + "\n");
                }
            }
        }
        System.out.println(sb.toString());
    }
}
*/ // 1.스캐너 쓰기 하지만 제한시간 초과로 버퍼를 씀

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Prac_java_3_6 {
    public static void main(String[] args) throws IOException {
//        Scanner sc = new Scanner(System.in);
//        Integer cycle = Integer.parseInt(sc.nextLine());

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Integer cycle = Integer.parseInt(br.readLine());


        LinkedList<Integer> queue = new LinkedList<>();
        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < cycle ; i++) {
            // push 1
            String[] commands = br.readLine().split(" ");
            // push
            String command = commands[0];

            if (command.equals("push")) {
                // 1
                queue.add(Integer.parseInt(commands[1]));
            } else if (command.equals("pop")) {
                if (queue.isEmpty()) {
                    // queue가 비어있으면 -1 출력
                    sb.append("-1\n");
                } else {
                    // queue가 차있으면 queue에 값을 넣고 그 수를 출력
                    sb.append(queue.poll() + "\n");
                }
            } else if (command.equals("size")) {
                sb.append(queue.size() + "\n");
            } else if (command.equals("empty")) {
                if (queue.isEmpty()) {
                    sb.append("1\n");
                } else {
                    sb.append("0\n");
                }
            } else if (command.equals("front")) {
                if (queue.isEmpty()) {
                    sb.append("-1\n");
                } else {
                    sb.append(queue.peekFirst() + "\n");
                }
            } else if (command.equals("back")) {
                if (queue.isEmpty()) {
                    sb.append("-1\n");
                } else {
                    sb.append(queue.peekLast() + "\n");
                }
            }
        }
        System.out.println(sb.toString());
    }
}

profile
안녕하세요.

0개의 댓글