문제

철수의 토마토 농장에서는 토마토를 보관하는 큰 창고를 가지고 있다. 토마토는 아래의 그림과 같이 격자 모양 상자의 칸에 하나씩 넣어서 창고에 보관한다.

창고에 보관되는 토마토들 중에는 잘 익은 것도 있지만, 아직 익지 않은 토마토들도 있을 수 있다. 보관 후 하루가 지나면, 익은 토마토들의 인접한 곳에 있는 익지 않은 토마토들은 익은 토마토의 영향을 받아 익게 된다. 하나의 토마토의 인접한 곳은 왼쪽, 오른쪽, 앞, 뒤 네 방향에 있는 토마토를 의미한다. 대각선 방향에 있는 토마토들에게는 영향을 주지 못하며, 토마토가 혼자 저절로 익는 경우는 없다고 가정한다. 철수는 창고에 보관된 토마토들이 며칠이 지나면 다 익게 되는지, 그 최소 일수를 알고 싶어 한다.

토마토를 창고에 보관하는 격자모양의 상자들의 크기와 익은 토마토들과 익지 않은 토마토들의 정보가 주어졌을 때, 며칠이 지나면 토마토들이 모두 익는지, 그 최소 일수를 구하는 프로그램을 작성하라. 단, 상자의 일부 칸에는 토마토가 들어있지 않을 수도 있다.

 

 

입력

첫 줄에는 상자의 크기를 나타내는 두 정수 M,N이 주어진다. M은 상자의 가로 칸의 수, N은 상자의 세로 칸의 수를 나타낸다. 단, 2 ≤ M,N ≤ 1,000 이다. 둘째 줄부터는 하나의 상자에 저장된 토마토들의 정보가 주어진다. 즉, 둘째 줄부터 N개의 줄에는 상자에 담긴 토마토의 정보가 주어진다. 하나의 줄에는 상자 가로줄에 들어있는 토마토의 상태가 M개의 정수로 주어진다. 정수 1은 익은 토마토, 정수 0은 익지 않은 토마토, 정수 -1은 토마토가 들어있지 않은 칸을 나타낸다.

토마토가 하나 이상 있는 경우만 입력으로 주어진다.

 

 

출력

여러분은 토마토가 모두 익을 때까지의 최소 날짜를 출력해야 한다. 만약, 저장될 때부터 모든 토마토가 익어있는 상태이면 0을 출력해야 하고, 토마토가 모두 익지는 못하는 상황이면 -1을 출력해야 한다.

 

 

예제

 

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;
 
public class Main_BJ_7576_토마토 {
 
    public static class position {
        int x, y;
 
        public position(int x, int y) {
            this.x = x;
            this.y = y;
        }
    }
 
    static int M, N, result, x, y;
    static int[] dx = { -1100 };
    static int[] dy = { 00-11 };
    static int[][] map;
    static boolean[][] visited;
    static Queue<position> queue = new LinkedList<>();
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
 
        st = new StringTokenizer(br.readLine(), " ");
 
        M = Integer.parseInt(st.nextToken());
        N = Integer.parseInt(st.nextToken());
 
        map = new int[N][M];
        visited = new boolean[N][M];
 
        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine(), " ");
            for (int j = 0; j < M; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
                if (map[i][j] == 1) {
                    queue.add(new position(i, j));
                }
            }
        }
        
       bfs();
 
        Loop: for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                if (map[i][j] == 0) {
                    result = -1;
                    break Loop;
                } 
                else result = Math.max(map[i][j], result);
            }
        }
 
        if (result != -1) result--;
 
        System.out.println(result);
 
    }
 
    public static void bfs() {
        int x, y;
 
        while (!queue.isEmpty()) {
            x = queue.peek().x;
            y = queue.peek().y;
            queue.poll();
 
            visited[x][y] = true;
 
            for (int i = 0; i < 4; i++) {
                int nx = x + dx[i];
                int ny = y + dy[i];
                if (check(nx, ny) && !visited[nx][ny] && map[nx][ny] == 0) {
                    queue.add(new position(nx, ny));
                    map[nx][ny] = map[x][y] + 1;
 
                    visited[nx][ny] = true;
                }
            }
        }
 
    }
 
    public static boolean check(int x, int y) {
        return x >= 0 && x < N && y >= 0 && y < M;
    }
 
}
cs

 

 

구현 방법

배열에 익은 토마토인 숫자 1이 들어갈 경우에 큐에 i와 j 값을 넣어줬습니다. 배열에 입력이 끝난 후, bfs() 함수로 들어가 큐에 있는 값을 꺼내 상, 하, 좌, 우로 이동할 수 있는 nx, ny를 통해 이동할 곳이 가능한 곳인지 if문을 통해 확인해 줬습니다. 만약 갈 곳이 익지 않은 토마토일 경우, 큐에 해당 nx, ny 값을 넣어주고 map 배열에는 전 배열 값 + 1을 저장해 해당 위치 토마토가 며칠에 익었는지 확인할 수 있도록 해주었습니다. 큐 안에 더이상 값이 없다면 반복문을 통해 해당 map 배열에 익지 않은 토마토가 있는지 검사하여 만약 익지 않은 토마토가 있다면 토마토가 모두 익지 못하는 상황이기 때문에 -1를 출력하고, 그렇지 않은 경우에는 배열에 저장된 값 중 가장 큰 값을 가져와 -1 을 해준 값을 출력해 주었습니다.

 

문제

상근이는 문자열에 폭발 문자열을 심어 놓았다. 폭발 문자열이 폭발하면 그 문자는 문자열에서 사라지며, 남은 문자열은 합쳐지게 된다.

폭발은 다음과 같은 과정으로 진행된다.

  • 문자열이 폭발 문자열을 포함하고 있는 경우에, 모든 폭발 문자열이 폭발하게 된다. 남은 문자열을 순서대로 이어 붙여 새로운 문자열을 만든다.
  • 새로 생긴 문자열에 폭발 문자열이 포함되어 있을 수도 있다.
  • 폭발은 폭발 문자열이 문자열에 없을 때까지 계속된다.

상근이는 모든 폭발이 끝난 후에 어떤 문자열이 남는지 구해보려고 한다. 남아있는 문자가 없는 경우가 있다. 이때는 "FRULA"를 출력한다.

폭발 문자열은 같은 문자를 두 개 이상 포함하지 않는다.

 

입력

첫째 줄에 문자열이 주어진다. 문자열의 길이는 1보다 크거나 같고, 1,000,000보다 작거나 같다.

둘째 줄에 폭발 문자열이 주어진다. 길이는 1보다 크거나 같고, 36보다 작거나 같다.

두 문자열은 모두 알파벳 소문자와 대문자, 숫자 0, 1, ..., 9로만 이루어져 있다.

 

출력

첫째 줄에 모든 폭발이 끝난 후 남은 문자열을 출력한다.

 

예제

 

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;
 
public class Main_BJ_9935_문자열폭발 {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        
        Stack<Character> stack = new Stack<>();
    
        String str = br.readLine();
        String bomb = br.readLine();
        
        for(int i=str.length()-1;i>=0;i--) {
            stack.push(str.charAt(i));
            
            if(stack.size()>=bomb.length() && stack.peek()==bomb.charAt(0)) {
                
                boolean check = true;
                
                for(int j=1; j<bomb.length(); j++) {
                    if(stack.get(stack.size()-j-1!= bomb.charAt(j)) {
                        check = false;
                        break;
                    }
                }
                
                if(check) {
                    for(int j=0; j<bomb.length(); j++)
                        stack.pop();
                }
            }
        }
        
        if(stack.isEmpty()) System.out.println("FRULA");
        else {
            while(!stack.isEmpty()) {
                sb.append(stack.pop());
            } System.out.println(sb.toString());
        }
    }
}
cs

 

구현 방법

처음에는 replace()를 사용하여 구현해봤더니 계속 메모리 초과가 발생했습니다. 그래서 출력 방식도 StringBuilder로 변경해서 제출해봤지만 똑같이 메모리 초과가 발생해서 스택을 이용해서 구현했습니다.

입력받은 문자열 끝에서부터 스택에 넣으면서 bomb과 문자가 동일하면 스택에 있는 값을 pop() 해주었습니다. 입력받은 문자열을 다 확인한 후, 마지막에 스택이 비어있으면 FRULA를 출력 해주고 아닐 경우에는 스택에 있는 값을 pop()해 출력해 주었습니다.

 

문제

지민이는 N개의 원소를 포함하고 있는 양방향 순환 큐를 가지고 있다. 지민이는 이 큐에서 몇 개의 원소를 뽑아내려고 한다.

지민이는 이 큐에서 다음과 같은 3가지 연산을 수행할 수 있다.

  1. 첫 번째 원소를 뽑아낸다. 이 연산을 수행하면, 원래 큐의 원소가 a1, ..., ak이었던 것이 a2, ..., ak와 같이 된다.
  2. 왼쪽으로 한 칸 이동시킨다. 이 연산을 수행하면, a1, ..., ak가 a2, ..., ak, a1이 된다.
  3. 오른쪽으로 한 칸 이동시킨다. 이 연산을 수행하면, a1, ..., ak가 ak, a1, ..., ak-1이 된다.

큐에 처음에 포함되어 있던 수 N이 주어진다. 그리고 지민이가 뽑아내려고 하는 원소의 위치가 주어진다. (이 위치는 가장 처음 큐에서의 위치이다.) 이때, 그 원소를 주어진 순서대로 뽑아내는데 드는 2번, 3번 연산의 최솟값을 출력하는 프로그램을 작성하시오.

 

입력

첫째 줄에 큐의 크기 N과 뽑아내려고 하는 수의 개수 M이 주어진다. N은 50보다 작거나 같은 자연수이고, M은 N보다 작거나 같은 자연수이다. 둘째 줄에는 지민이가 뽑아내려고 하는 수의 위치가 순서대로 주어진다. 위치는 1보다 크거나 같고, N보다 작거나 같은 자연수이다.

 

출력

첫째 줄에 문제의 정답을 출력한다.

 

예제

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.StringTokenizer;
 
public class Main_BJ_1021_회전하는큐 {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        
        LinkedList<Integer> list = new LinkedList<Integer>();
        
        st = new StringTokenizer(br.readLine(), " ");
        
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
        
        for(int i=1;i<=N;i++) list.add(i);
        
        int count=0;
        st = new StringTokenizer(br.readLine()," ");
        for(int i=0;i<M;i++) {
            int num = Integer.parseInt(st.nextToken());
            
            if(list.indexOf(num)<=list.size()/2) {
                while(num!=list.getFirst()) {
                    list.addLast(list.removeFirst());
                    count++;
                }
            }
            else {
                while(num!=list.getFirst()) {
                    list.addFirst(list.removeLast());
                    count++;
                }
            }
            list.removeFirst(); 
        }
        System.out.println(count);
    }
}
cs

 

구현 방법

회전하는 큐 문제에서 수행할 수 있는 연산은 총 3가지입니다. 첫 번째 원소를 뽑아내거나, 왼쪽/ 오른쪽으로 한 칸 이동시키는 연산을 할 수 있습니다. 문제에서 왼쪽, 오른쪽으로 한 칸 이동시키는 연산의 최솟값을 구하는 문제이기 때문에 여기서 중요하게 생각해야 하는 부분은 뽑아낼 원소가 왼쪽으로 한 칸 이동시키는 연산을 수행해야 할지 아님 오른쪽으로 한 칸 이동시키는 연산을 수행해야 할지 고르는 것입니다.

 

뽑아낼 원소의 수를 받기 전, list에 1부터 N까지의 값을 list에 넣어뒀습니다. 그 다음에 뽑아낼 원소의 수를 하나씩 입력받아 원소가 어느쪽에 더 가깝게 있는지 indexOf()를 통해 num값의 인덱스를 찾아냈습니다. 인덱스를 통해 list 사이즈/2와 비교하여 작거나 같을 경우 맨 앞에 있는 리스트 값을 맨 뒤로 이동시키는 연산을 찾고자하는 원소 값이 list 맨 앞에 올 때까지 while문을 돌렸습니다. 찾고자 하는 num 값이 list 인덱스 0번에 왔을 때, while문을 종료하고 마지막에 list.remove()를 사용하여 원소를 뽑아냈습니다. 마찬가지로 오른쪽으로 한 칸 이동시키는 연산도 구현해 주었습니다.

 

문제

N×N의 표에 수 N2개 채워져 있다. 채워진 수에는 한 가지 특징이 있는데, 모든 수는 자신의 한 칸 위에 있는 수보다 크다는 것이다. N=5일 때의 예를 보자.

이러한 표가 주어졌을 때, N번째 큰 수를 찾는 프로그램을 작성하시오. 표에 채워진 수는 모두 다르다.

 

입력

첫째 줄에 N(1 ≤ N ≤ 1,500)이 주어진다. 다음 N개의 줄에는 각 줄마다 N개의 수가 주어진다. 표에 적힌 수는 -10억보다 크거나 같고, 10억보다 작거나 같은 정수이다.

 

출력

첫째 줄에 N번째 큰 수를 출력한다.

 

예제

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.StringTokenizer;
 
public class Main_BJ_2075_N번째큰수 {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        
        PriorityQueue<Integer> pq = new PriorityQueue<>(new Comparator<Integer>() {
 
            @Override
            public int compare(Integer o1, Integer o2) {
                // TODO Auto-generated method stub
                return Integer.compare(o1, o2) * -1;
            }
        });
        
        int N = Integer.parseInt(br.readLine());
        
        for(int i=0;i<N;i++) {
            st = new StringTokenizer(br.readLine()," ");
            for(int j=0;j<N;j++) {
                pq.add(Integer.parseInt(st.nextToken()));
            }
        }
        
        for(int i=0;i<N-1;i++) pq.poll();
        
        System.out.println(pq.poll());
    }
 
}
 
cs

 

구현방법

첫 번째로는 PriorityQueue에 값을 넣어서 while()문을 통해 N번째 큰 수를 찾는 방법으로 구현했습니다.

 

두 번째로는 N번째로 큰 수를 찾는 문제이니 PriorityQueue에 오름차순이 아닌 내림차순으로 정렬하여 값을 넣은 뒤, N번째 큰 수를 찾는 방법으로 구현했습니다.

문제

이중 우선순위 큐(dual priority queue)는 전형적인 우선순위 큐처럼 데이터를 삽입, 삭제할 수 있는 자료 구조이다. 전형적인 큐와의 차이점은 데이터를 삭제할 때 연산(operation) 명령에 따라 우선순위가 가장 높은 데이터 또는 가장 낮은 데이터 중 하나를 삭제하는 점이다. 이중 우선순위 큐를 위해선 두 가지 연산이 사용되는데, 하나는 데이터를 삽입하는 연산이고 다른 하나는 데이터를 삭제하는 연산이다. 데이터를 삭제하는 연산은 또 두 가지로 구분되는데 하나는 우선순위가 가장 높은 것을 삭제하기 위한 것이고 다른 하나는 우선순위가 가장 낮은 것을 삭제하기 위한 것이다.

정수만 저장하는 이중 우선순위 큐 Q가 있다고 가정하자. Q에 저장된 각 정수의 값 자체를 우선순위라고 간주하자. 

Q에 적용될 일련의 연산이 주어질 때 이를 처리한 후 최종적으로 Q에 저장된 데이터 중 최댓값과 최솟값을 출력하는 프로그램을 작성하라.

 

입력

입력 데이터는 표준입력을 사용한다. 입력은 T개의 테스트 데이터로 구성된다. 입력의 첫 번째 줄에는 입력 데이터의 수를 나타내는 정수 T가 주어진다. 각 테스트 데이터의 첫째 줄에는 Q에 적용할 연산의 개수를 나타내는 정수 k (k ≤ 1,000,000)가 주어진다. 이어지는 k 줄 각각엔 연산을 나타내는 문자(‘D’ 또는 ‘I’)와 정수 n이 주어진다. ‘I n’은 정수 n을 Q에 삽입하는 연산을 의미한다. 동일한 정수가 삽입될 수 있음을 참고하기 바란다. ‘D 1’는 Q에서 최댓값을 삭제하는 연산을 의미하며, ‘D -1’는 Q 에서 최솟값을 삭제하는 연산을 의미한다. 최댓값(최솟값)을 삭제하는 연산에서 최댓값(최솟값)이 둘 이상인 경우, 하나만 삭제됨을 유념하기 바란다.

만약 Q가 비어있는데 적용할 연산이 ‘D’라면 이 연산은 무시해도 좋다. Q에 저장될 모든 정수는 32-비트 정수이다.

 

출력

출력은 표준출력을 사용한다. 각 테스트 데이터에 대해, 모든 연산을 처리한 후 Q에 남아 있는 값 중 최댓값과 최솟값을 출력하라. 두 값은 한 줄에 출력하되 하나의 공백으로 구분하라. 만약 Q가 비어있다면 ‘EMPTY’를 출력하라.

 

예제

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
import java.util.TreeMap;
 
public class Main_BJ_7662_이중우선순위큐 {
    static TreeMap<Long, Long> tm = new TreeMap<>();
    static StringBuilder sb = new StringBuilder();
    
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        
        int T = Integer.parseInt(br.readLine());
        
        for(int tc=0;tc<T;tc++) {
            int k = Integer.parseInt(br.readLine());
            for(int i=0;i<k;i++) {
                st = new StringTokenizer(br.readLine()," ");
                String command = st.nextToken();
                long num = Long.parseLong(st.nextToken());
                
                if(command.equals("I")) {
                    if(tm.containsKey(num)) {
                        tm.put(num, tm.get(num)+1);
                    }else tm.put(num, 1L);
                }
                else copy(num);
            }
            result();
        }
        System.out.println(sb.toString());
    }
    
    public static void copy(long num) {
        
        if(num==1) { //최대값 삭제 
            if(!tm.isEmpty()) {
                long value = tm.get(tm.lastKey())-1;
                if(value==0) tm.remove(tm.lastKey());
                else tm.put(tm.lastKey(), value);
            }
        }
        else {
            if(!tm.isEmpty()) {
                long value = tm.get(tm.firstKey())-1;
                if(value==0) tm.remove(tm.firstKey());
                else tm.put(tm.firstKey(), value);
            }
        }
    }
    
    public static void result() {
        if(tm.isEmpty()) sb.append("EMPTY\n");
        else sb.append(tm.lastKey()+" "+tm.firstKey()+"\n");
        tm.clear();
    }
}
cs

 

한줄평

처음에는 PriorityQueue와 ArrayList를 사용해서 command가 D일 때 copy()함수에서 큐에 있는 수들을 list에 담아 최솟값과 최댓값을 삭제하도록 구현했는데 시간 초과가 발생했습니다.

그래서 두 번째로 구현한 방법이 TreeMap을 사용해서 key값에 따라 key값이 벌써 TreeMap에 존재하면 value값을 +1 증가시켜 동일한 수가 몇 개가 존재하는지 확인할 수 있도록 했습니다. command가 D일때 copy()함수에서 최솟값과 최댓값의 value를 확인해 동일한 값이 없을 때는 삭제를 하고 동일한 값이 존재할 때는 key의 value 값은 -1 한 뒤, 다시 저장해 주었습니다. 그리고 마지막에는 result()함수에서 TreeMap에 값이 존재하는지 확인해 없으면 EMPTY를 출력해 주고 존재할 경우 최솟값과 최댓값을 출력해 주었습니다.

 

+ Recent posts