sudoku

main.cc

@@ -0,0 +1,97 @@
+/*
+ * @lc app=leetcode.cn id=37 lang=cpp
+ *
+ * [37] 解数独
+ */
+#include <bits/stdc++.h>
+using namespace std;
+// @lc code=start
+
+void print(const vector<vector<char>>& board) {
+  for (int i = 0; i < 9; ++i) {
+    for (int j = 0; j < 9; ++j) {
+      cout << board[i][j] << ",";
+    }
+    cout << endl;
+  }
+}
+
+class Solution {
+ public:
+  void solveSudoku(vector<vector<char>>& board) {
+    vector<pair<int, int>> empty;
+    for (int i = 0; i < 9; ++i) {
+      for (int j = 0; j < 9; ++j) {
+        if (board[i][j] == '.') {
+          empty.emplace_back(i, j);
+        }
+      }
+    }
+    int n = empty.size();
+    function<bool(int, int)> vaild = [&](int i, int j) -> bool {
+      array<int, 9> exist1 = {0, 0, 0, 0, 0, 0, 0, 0, 0};
+      array<int, 9> exist2 = {0, 0, 0, 0, 0, 0, 0, 0, 0};
+      array<int, 9> exist3 = {0, 0, 0, 0, 0, 0, 0, 0, 0};
+      for (int k = 0; k < 9; ++k) {
+        if (board[i][k] != '.') {
+          if (exist1[board[i][k] - '1']) {
+            return false;
+          }
+          exist1[board[i][k] - '1'] = 1;
+        }
+        if (board[k][j] != '.') {
+          if (exist2[board[k][j] - '1']) {
+            return false;
+          }
+          exist2[board[k][j] - '1'] = 1;
+        }
+      }
+      int r = i / 3 * 3;
+      int c = j / 3 * 3;
+      for (int k = 0; k < 3; ++k) {
+        for (int l = 0; l < 3; ++l) {
+          if (board[r + k][c + l] != '.') {
+            if (exist3[board[r + k][c + l] - '1']) {
+              return false;
+            }
+            exist3[board[r + k][c + l] - '1'] = 1;
+          }
+        }
+      }
+      return true;
+    };
+
+    function<void(int)> f = [&](int i) {
+      if (i == n) {
+        throw exception();
+      }
+      for (int j = 0; j < 9; ++j) {
+        board[empty[i].first][empty[i].second] = j + '1';
+        if (vaild(empty[i].first, empty[i].second)) {
+          f(i + 1);
+        }
+        board[empty[i].first][empty[i].second] = '.';
+      }
+    };
+    try {
+      f(0);
+    } catch (const exception& e) {
+    }
+    // cant be solved
+  }
+};
+// @lc code=end
+int main() {
+  Solution s;
+  vector<vector<char>> b = {{'5', '3', '.', '.', '7', '.', '.', '.', '.'},
+                            {'6', '.', '.', '1', '9', '5', '.', '.', '.'},
+                            {'.', '9', '8', '.', '.', '.', '.', '6', '.'},
+                            {'8', '.', '.', '.', '6', '.', '.', '.', '3'},
+                            {'4', '.', '.', '8', '.', '3', '.', '.', '1'},
+                            {'7', '.', '.', '.', '2', '.', '.', '.', '6'},
+                            {'.', '6', '.', '.', '.', '.', '2', '8', '.'},
+                            {'.', '.', '.', '4', '1', '9', '.', '.', '5'},
+                            {'.', '.', '.', '.', '8', '.', '.', '7', '9'}};
+  s.solveSudoku(b);
+  print(b);
+}