LeetCode 37 解数独

编写一个程序,通过已填充的空格来解决数独问题。

一个数独的解法需遵循如下规则:

数字 1-9 在每一行只能出现一次。
数字 1-9 在每一列只能出现一次。
数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。
空白格用 '.' 表示。

直接莽,回溯遍历:

from typing import *
from collections import defaultdict


class Solution:

    def solveSudoku(self, board: List[List[str]]) -> None:
        """
        Do not return anything, modify board in-place instead.
        """
        found = False

        def find_next(idx_i, idx_j):
            next_i, next_j = None, None
            for i in range(idx_i, 9):
                if next_i is not None:
                    break
                for j in range(9):
                    if i == idx_i and j <= idx_j:
                        continue
                    if board[i][j] == '.':
                        next_i, next_j = i, j
                        break
            return next_i, next_j

        def back(idx_i, idx_j, val):
            nonlocal found
            board[idx_i][idx_j] = val
            row_sets[idx_i].add(val)
            col_sets[idx_j].add(val)
            box_sets[idx_i // 3, idx_j // 3].add(val)
            next_i, next_j = find_next(idx_i, idx_j)
            if next_i is None:
                found = True
                return
            box_next_i, box_next_j = next_i // 3, next_j // 3
            for next_val in range(1, 10):
                next_val = str(next_val)
                if next_val not in row_sets[next_i] and next_val not in col_sets[next_j] and next_val not in box_sets[box_next_i, box_next_j]:
                    back(next_i, next_j, next_val)
            if not found:
                board[idx_i][idx_j] = '.'
                row_sets[idx_i].remove(val)
                col_sets[idx_j].remove(val)
                box_sets[idx_i // 3, idx_j // 3].remove(val)

        row_sets = defaultdict(lambda: set())
        col_sets = defaultdict(lambda: set())
        box_sets = defaultdict(lambda: set())
        for i in range(len(board)):
            for j in range(len(board[0])):
                if board[i][j] != '.':
                    row_sets[i].add(board[i][j])
                    col_sets[j].add(board[i][j])
                    box_i, box_j = i // 3, j // 3
                    box_sets[box_i, box_j].add(board[i][j])
        idx_i, idx_j = find_next(0, -1)
        for val in range(1, 10):
            val = str(val)
            if found:
                break
            if val not in row_sets[idx_i] and val not in col_sets[idx_j] and val not in box_sets[
                idx_i // 3, idx_j // 3]:
                back(idx_i, idx_j, val)

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值