AcWing: 1432. Défi échiquier
Mettez cette condition au début
- Il peut y avoir au plus une pièce sur chaque diagonale
Je me suis trompé, je pensais que c'était juste pour limiter la diagonale principale et la sous-diagonale de toute la matrice. Une seule pièce peut être placée au plus
En fait, il s'agit de la diagonale principale et de la sous-diagonale correspondant à chaque pièce d'échecs placée à la position actuelle .
Déformation de huit reines, l'énumération des dfs suffit
Position actuelle (ligne, col)
Toutes les positions correspondant à la ligne sous-diagonale: i + j == ligne + col
Positions correspondantes sur la diagonale principale: regardez le code des autres comme n + (ligne-j) // J'ai beaucoup appris
Code AC
import java.util.*;
import static java.lang.System.out;
public class Main{
static int ans = 0;
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int[][] map = new int[n][n];
// cols 记录当前列是否被访问过、 dia 记录当前副对角线是否被访问过
boolean[] cols = new boolean[n], dia = new boolean[2 * n];
dfs(map, 0, cols, dia, "");
out.println(ans);
}
public static void dfs(int[][] map, int row, boolean[] cols, boolean[] dia, String s){
if(row >= map.length) {
//out.println(s);
ans++;
if(ans <= 3) out.println(s);
return ;
}
int len = map[0].length;
for(int j = 0; j < len; j++) {
// 当前列 没有棋子 && 副对角线 没有棋子 && 当前主对角线没有棋子
if(!cols[j] && !dia[row + j] && map[row][j] == 0) {
// 对角线的
getset(map, row, j, 1);
cols[j] = true; dia[row + j] = true;
dfs(map, row + 1, cols, dia, s + (j + 1) + " ");
// 回溯
getset(map, row, j, -1);
cols[j] = false; dia[row + j] = false;
}
}
}
// 对角线
public static void getset(int[][] map, int row, int col, int val){
//主对角线
int mn = Math.min(row, col), rlen = map.length, clen = map[0].length;
int r = row - mn, c = col - mn;
while(r < rlen && c < clen) {
map[r++][c++] += val;
}
// 副对角线
//for(int i = 0; i < rlen; i++) {
// for(int j = 0; j < clen; j++) {
// if(i + j == row + col) map[i][j] += val;
// }
//}
}
}
Optimiser la partie diagonale principale
simplifier
Code AC
import java.util.*;
import static java.lang.System.out;
public class Main{
static int ans = 0;
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int[][] map = new int[n][n];
// cols 记录当前列是否被访问过、 dia 记录当前副对角线是否被访问过、master 记录当前主对角线是否被访问过
boolean[] cols = new boolean[n], dia = new boolean[2 * n], master = new boolean[2 * n];
dfs(map, 0, cols, dia, master, "");
out.println(ans);
}
public static void dfs(int[][] map, int row,
boolean[] cols, boolean[] dia, boolean[] master, String s){
if(row >= map.length) {
//out.println(s);
ans++;
if(ans <= 3) out.println(s);
return ;
}
int len = map[0].length;
for(int j = 0; j < len; j++) {
// 当前列 没有棋子 && 副对角线 没有棋子 && 当前主对角线没有棋子
if(!cols[j] && !dia[row + j] && !master[map.length + (row - j)]) {
// 对角线的
cols[j] = true; dia[row + j] = true; master[map.length + (row - j)] = true;
dfs(map, row + 1, cols, dia, master, s + (j + 1) + " ");
// 回溯
cols[j] = false; dia[row + j] = false; master[map.length + (row - j)] = false;
}
}
}
/**
* @Discard 废弃
*/
// 对角线
//public static void getset(int[][] map, int row, int col, int val){
// //主对角线
// int mn = Math.min(row, col), rlen = map.length, clen = map[0].length;
// int r = row - mn, c = col - mn;
// while(r < rlen && c < clen) {
// map[r++][c++] += val;
// }
//}
}