forked from MiYazJE/Acepta-el-reto
-
Notifications
You must be signed in to change notification settings - Fork 0
/
p137.java
94 lines (74 loc) · 2.72 KB
/
p137.java
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
import java.util.ArrayList;
import java.util.Scanner;
public class p137 {
public static void obtenerSize(int[][] m, int i, int j) {
if (!safe(m, i, j)) {
}
else if (m[i][j] == 1) {
if (contieneDiagonal(m, i, j)) {
correcto = false;
return;
}
size++;
m[i][j] = 2;
obtenerSize(m, i+1, j);
obtenerSize(m, i-1, j);
obtenerSize(m, i, j+1);
obtenerSize(m, i, j-1);
}
}
public static boolean contieneDiagonal(int[][] m, int i, int j) {
return ( (safe(m, i-1, j-1) && m[i-1][j-1] != 0) || // arriba izq
(safe(m, i+1, j+1) && m[i+1][j+1] != 0) || // abajo der
(safe(m, i+1, j-1) && m[i+1][j-1] != 0) || // abajo izq
(safe(m, i-1, j+1) && m[i-1][j+1] != 0) ); // arriba der
}
public static boolean safe(int[][] m, int i, int j) {
return (i >= 0 && j >= 0 && i < m.length && j < m[0].length);
}
public static int size;
public static boolean correcto;
public static void main(String[] args) {
Scanner s = new Scanner(System.in);
int numBarcos, dim;
ArrayList<Integer> tamanyos;
int[][] m;
boolean noBarcos;
while (true) {
numBarcos = s.nextInt();
if (numBarcos == 0) break;
tamanyos = new ArrayList<>();
correcto = noBarcos = true;
// LECTURA DE LOS BARCOS
for (int i = 0; i < numBarcos; i++) tamanyos.add(s.nextInt());
// dimensiones del tablero
dim = s.nextInt();
m = new int[dim][dim];
// Lectura del tablero
for (int i = 0; i < dim; i++)
for (int j = 0; j < dim; j++) {
m[i][j] = s.nextInt();
if (m[i][j] == 1) noBarcos = false;
}
if (noBarcos) System.out.println("NO");
else {
for (int i = 0; i < dim; i++) {
for (int j = 0; j < dim; j++) {
size = 0;
if (m[i][j] == 1) {
obtenerSize(m, i, j);
if (correcto && tamanyos.contains(size)) {
tamanyos.remove(Integer.valueOf(size));
} else {
correcto = false;
break;
}
}
}
if (!correcto) break;
}
System.out.println( (correcto && tamanyos.isEmpty()) ? "SI" : "NO" );
}
}
}
}