글 읽기 - 자바 대충 짠 그리디

jeen0112 1년 전 0

생각해보면 간단하네요. 1은 앞에서 부터 절반만큼 제거해주고 0은 뒤에서부터 절반만큼 제거해주면 됩니다.

자바 스트링 제거는 어렵기 때문에 인덱스를 모두 담아서 문자열을 재배치하시면 되겠습니다.

import java.util.*;
class Sum{
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        String str = scan.next();
        int one = 0;
        int zero = 0;
        for(int i = 0; i < str.length(); i++){
            if(str.charAt(i) == '1')
                one++;
            else
                zero++;
        }
        one = one / 2;
        zero = zero / 2;
        ArrayList<Integer> list = new ArrayList<>();
        int count = 0;
        for(int i = 0; i < str.length(); i++){
            if(count == one){
                break;
            }
            if(str.charAt(i) == '1'){
                list.add(i);
                count++;
            }
        }
        count = 0;
        for(int i = str.length() - 1; i >= 0; i--){
            if(count == zero){
                break;
            }
            if(str.charAt(i) == '0'){
                list.add(i);
                count++;
            }
        }
        Collections.sort(list);
        String temp = "";
        int index = 0;
        for(int i = 0; i < str.length(); i++){
            if(index == list.size()){
                temp += str.charAt(i);
                continue;
            }
            int now = list.get(index);
            if(now == i) {
                index++;
                continue;
            }
            temp += str.charAt(i);
        }
        System.out.println(temp);
    }
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

자바 대충 짠 그리디

jeen0112 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말