ヒューリスティック探索 aoj 深さ優先探索 8クイーン バックトラック

ALDS_1_13_A 8 クイーン

ALDS_1_13_A 8 クイーン
Feb. 2, 2020, 1:52 p.m.

目次

問題

http://judge.u-aizu.ac.jp/onlinejudge/description.jsp?id=ALDS1_13_A

解説

計算量

解答

 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
#define MAX_K 
#define W 8
Int K;
bool B[W][W];

#define FREE -1
#define NOT_FREE 1

vector<Int> rows(W, FREE), cols(W, FREE), dpos(W*2-1, FREE), dneg(W*2-1, FREE);

void input() {
  Int r, c;
  cin >> K;

  loop(r,0,W) {
    loop(c,0,W) {
      B[r][c] = false;
    }
  }

  loop(k,0,K) {
    cin >> r >> c;
    B[r][c] = true;
  }
}

void dump() {
  // Filter the answer
  loop(c,0,W) {
    loop(r,0,W) {
      if (B[r][c]) {
        if (rows[c] != r) return;
      }
    }
  }

  loop(c,0,W) {
    loop(r,0,W) {
      if (rows[r] == c) cout << 'Q';
      else cout << '.';
    }
    cout << endl;
  }
}

bool is_free(Int col, Int row) {
  return cols[row] == FREE && dpos[col + row] == FREE && dneg[col - row + W - 1] == FREE;
}

void rec(Int col) {
  if (col == W) {
    dump(); return;
  }

  loop(row,0,W) {
    if (!is_free(col, row)) continue;
    rows[col] = row;
    cols[row] = dpos[col + row] = dneg[col - row + W - 1] = NOT_FREE;

    rec(col + 1);

    rows[col] = cols[row] = dpos[col + row] = dneg[col - row + W - 1] = FREE;
  }
}

void solve() {
  rec(0);
}

int main(void) {
  input();
  solve();
}