P1406 方格填数

暴力搜索

package dfs;

import java.util.Arrays;
import java.util.Scanner;

public class 方格填数 {
    static Scanner s = new Scanner(System.in);
    static int n = s.nextInt(),sum = 0;
    static int[] ghy = new int[n*n];//读取输入数组
    static int[][] ans = new int[n][n];//记录答案
    static int[] hang = new int[n];//行
    static int[] lie = new int[n];//列
    static int count = 0;//左对角线
    static int count2 = 0;//右对角线
    static boolean[] flag = new boolean[n*n];//标记是否走过
    public static void main(String[] args) {
        for (int i = 0; i < n * n; i++) {
            ghy[i] = s.nextInt();
            sum += ghy[i];
        }
        sum = sum/n;
        Arrays.sort(ghy);
        dfs(0);
    }
    static void dfs(int u){
        if (u == n * n) {
            for (int i = 0; i < n; i++) {
                if (!(hang[i] == sum && lie[i] == sum)) {//判断行和列
                    return;
                }
            }
            //如果行和列都通过了在判断对角线,这里要注意,判断完要把count重置为0
            for (int i = 0; i < n; i++) {
                for (int j = 0; j < n; j++) {
                    if (i == j) {
                        count += ans[i][j];
                    }
                    if ((i + j) == n - 1) {
                        count2 += ans[i][j];
                    }
                }
            }
            if(count == sum && count2 ==sum) {
            System.out.println(sum);
                for (int i = 0; i < n; i++) {
                    for (int j = 0; j < n; j++) {
                        System.out.print(ans[i][j] + " ");
                    }
                    System.out.println();
                }
                System.exit(0);//直接终止程序
            }
            count2 =count=0;//重置count,不然不正确的答案的count会累加
            return;
        }
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if(!flag[i*n + j]) {
                    flag[i*n + j] = true;//标记为走过
                    ans[u/n][u%n] = ghy[i * n + j];//i*n + j是代表第几个,u/n得出的是第几行,不可以直接用i
                    hang[u/n] += ghy[i * n + j];
                    lie[u%n] += ghy[i * n + j];
                    dfs(u + 1);
                    回溯
                    flag[i*n + j] = false;
                    ans[u/n][u%n] = 0;
                    hang[u/n] -= ghy[i * n + j];
                    lie[u%n] -= ghy[i * n + j];
                }
            }
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
方格数是一种经典的数学游戏,通过写数字使得每行、每列和每个小方格内的数字均不重复。在Python中可以通过回溯算法实现方格数游戏的求解。下面是一个简单的实现示例: ```python # 方格数游戏求解 # 定义游戏棋盘 board = [[0 for i in range(9)] for j in range(9)] # 定义判断是否可数字的函数 def is_valid(row, col, num): # 判断行和列是否重复 for i in range(9): if board[row][i] == num or board[i][col] == num: return False # 判断小方格是否重复 start_row = (row // 3) * 3 start_col = (col // 3) * 3 for i in range(start_row, start_row + 3): for j in range(start_col, start_col + 3): if board[i][j] == num: return False return True # 定义回溯函数 def backtrack(row, col): # 判断是否满 if row == 9: return True # 计算下一个位置 next_row = row if col < 8 else row + 1 next_col = (col + 1) % 9 # 如果已经过,则跳过 if board[row][col] != 0: return backtrack(next_row, next_col) # 尝试数字 for num in range(1, 10): if is_valid(row, col, num): board[row][col] = num if backtrack(next_row, next_col): return True board[row][col] = 0 return False # 求解方格数游戏 backtrack(0, 0) # 输出结果 for row in board: print(row) ``` 可以看到,该算法通过递归实现对棋盘的深度优先搜索,并在每个格子中尝试入数字,如果符合要求就继续搜索下一步。如果搜索到最后,即满了整个棋盘,则返回True,否则返回False。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值