(UVA) 729 - The Hamming Distance Problem - Solução 2

Solução usando recursão.
(fica mais rápido)

import java.io.*;
import java.util.*;

class Main {
    public static void main(String[] args) throws NumberFormatException, IOException {
        Main processando = new Main();
        processando.processa();
       
        System.exit(0);
    }

    static int leitor(BufferedReader br) throws NumberFormatException, IOException {
        int n;
        int resp = 0;
        
        while (true) {
            n = br.read();
            if (n >= '0' && n <= '9') break;
        }
        while (true) {
            resp = resp*10 + n-'0';
            n = br.read();
            if (n < '0' || n > '9') break;
        }
        
        return resp;
    }
   
    void funcao(int[] digito, int p, int comprimento, int distanciaHamming, BufferedWriter bw, int qteZero, int qteUm) throws NumberFormatException, IOException {
        if (p == comprimento) {
            if (qteUm == distanciaHamming) {
                for (int i = 0; i < comprimento; i++) {
                    bw.write(digito[i]);
                }
                bw.write("\n");
            }
           
            return;
        }
       
        digito[p] = '0';
        qteZero++;
        if ((comprimento-qteZero) < distanciaHamming) {
            qteZero--;
        }
        else {
            funcao(digito, p+1, comprimento, distanciaHamming, bw, qteZero, qteUm);
            qteZero--;
        }    
       
        digito[p] = '1';
        qteUm++;
        if (qteUm > distanciaHamming) {
            qteUm--;
        }
        else {
            funcao(digito, p+1, comprimento, distanciaHamming, bw, qteZero, qteUm);
            qteUm--;
        }
       
        return;
    }
 
    void processa() throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int qteCasos = leitor(br);
       
        boolean pula = false;
       
        while (qteCasos > 0) {
            qteCasos--;
           
            String line = br.readLine();
           
            int comprimento = leitor(br);
            int distanciaHamming = leitor(br);

            if (pula) {
                bw.write("\n");
            }
           
            pula = true;
           
            if (distanciaHamming != 0) {
                int[] digito = new int[comprimento];
                funcao(digito, 0, comprimento, distanciaHamming, bw, 0, 0);
            }
            else {
                for (int i = 0; i < comprimento; i++) {
                    bw.write("0");
                }        
                bw.write("\n");  
            }
           
            bw.flush();

        }
       
        bw.close();
       
        return;
    }
}

Comments

Popular posts from this blog

(Coderbyte) Dash Insert II - Solução

(Coderbyte) Run Length - Solução

(Coderbyte) Counting Minutes I - Solução