(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;
}
}
(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
Post a Comment