백준 1195번 : 킥다운

전에 실패했던 문제였는데 그리디에 자신감이 붙어서 다시 한번 시도해봤다!

if문의 순서만 바꾸면 되는 간단한 문제였다..

원리는 두개의 배열을 입력받고 더 긴 배열에 앞부분부터 더 작은 배열의 길이 - 1 만큼 0을 추가하는 것이다.

그 다음에 0을 제거해 나가면서 맞물리지 않는 부분이 있는지(같은 인덱스에 있는 두 배열의 값의 합이 3을 넘는지) 체크를 한 후 조건을 만족하면 현재의 길이를 리스트에 저장한 후 마지막에 가장 짧은 길이를 찾아서 출력한다.

import java.io.*;
import java.util.ArrayList;
import java.util.List;

class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuffer sb = new StringBuffer();
        char[] first = br.readLine().toCharArray();
        char[] second = br.readLine().toCharArray();
        List<Integer> maxList = new ArrayList<>();
        List<Integer> minList = new ArrayList<>();
        boolean firstMax = first.length > second.length;

        if (firstMax) {
            for (char c : first) {
                maxList.add(Integer.parseInt(String.valueOf(c)));
            }
            for (char c : second) {
                minList.add(Integer.parseInt(String.valueOf(c)));
            }
            for (int i = 0; i < second.length - 1; i++) {
                maxList.add(0, 0);
            }
        } else {
            for (char c : second) {
                maxList.add(Integer.parseInt(String.valueOf(c)));
            }
            for (char c : first) {
                minList.add(Integer.parseInt(String.valueOf(c)));
            }
            for (int i = 0; i < first.length - 1; i++) {
                maxList.add(0, 0);
            }
        }

        List<Integer> len = new ArrayList<>();
        int size = maxList.size();
        int flagCount = 0;
        for (int j = 0; j < size; j++) {
            boolean flag = true;
            for (int i = 0; i < Math.min(maxList.size(), minList.size()); i++) {
                if (maxList.get(i) != 0 && minList.get(i) + maxList.get(i) > 3) {
                    flag = false;
                    break;
                }
            }
            if (flag) {
                flagCount++;
                len.add(Math.max(minList.size(), maxList.size()));
            }
            if (maxList.get(0).equals(0)) {
                maxList.remove(0);
            } else {
                minList.add(0, 0);
            }
        }
        if (flagCount < 1) {
            writer.write(String.valueOf(first.length + second.length));
        } else {
            writer.write(String.valueOf(len.stream().mapToInt(value -> value).min().orElse(0)));
        }
        writer.flush();
        writer.close();
    }
}