Home 16967번(타노스)
Post
Cancel

16967번(타노스)

첫 시도

  • 투포인터로 가능할지 확인 -> 문자열이 그대로 사용되는 것이 아니라 숫자가 제거되어 기존 문자열에서 변경이 되기 때문에 투포인터로는 구현 불가

해결

  • 먼저 0과 1의 총 개수를 구하고 그 수를 각각 2로 나눈다. 나눈 수가 문자열에서 지워야 할 0과 1의 갯수다.
  • 문제 조건을 고려해 봤을 때 사전순으로 빠르기 위해서는 0은 앞에 있어야 하고 1은 뒤에 있어야 한다 -> 1을 앞에서부터 지우고 0을 뒤에서부터 지운다면 항상 사전순으로 가장 빠른 것을 구할 수 있다.
  • 문자열의 길이를 N이라고 했을 때 시간 복잡도는 O(3N)
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
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.StringTokenizer;

public class Main {

    public static void main(String[] args) throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        String[] input = br.readLine().split("");

        int zero = 0;
        int one = 0;

        for(int i = 0; i< input.length; i++){
            if(input[i].equals("1")) one++;
            else zero++;
        }

        zero /= 2;
        one /= 2;

        for(int i = 0; i<input.length; i++){ // 정방향, 1 제거
            if(one != 0 && input[i].equals("1")) {
                input[i] = "-1";
                one--;
            }
        }
        for(int i = input.length-1; i>=0; i--){ // 역방향, 0 제거
            if(zero != 0 && input[i].equals("0")) {
                input[i] = "-1";
                zero--;
            }
        }
        String result = "";
        for(int i = 0; i<input.length; i++){
            if(input[i].equals("0") || input[i].equals("1")){
                result += input[i];
            }
        }

        System.out.println(result);
    }
}

참고

This post is licensed under CC BY 4.0 by the author.