bzoj3109: [cqoi2013]新数独

最新推荐文章于 2022-05-08 10:53:54 发布

原创最新推荐文章于 2022-05-08 10:53:54 发布 · 529 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

bzoj 同时被 2 个专栏收录

74 篇文章

订阅专栏

搜索

6 篇文章

订阅专栏

本文介绍了一种使用C++解决新数独问题的方法，通过搜索加剪枝策略求解给定相邻格子数大小关系的数独谜题。文章详细展示了如何利用递归深度优先搜索，并结合行、列及九宫格内的数字使用情况实现有效的剪枝操作。

题面在这里

题意：

给一个空的数独，告诉你每个9宫格内相邻格子数的大小关系，求这个数独。

做法：

搜索+剪枝。
剪枝就是最显然的剪枝，碰到不符合条件的就不搜下去。

易错点：

cmp函数返回的是1和-1，我原来定义了bool，然后就gg了。

代码：

/*************************************************************
    Problem: bzoj 3109 [cqoi2013]新数独
    User: fengyuan
    Language: C++
    Result: Accepted
    Time: 2576 ms
    Memory: 1300 kb
    Submit_Time: 2018-01-17 12:02:52
*************************************************************/

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cctype>
#include<cmath>
using namespace std;

const int N = 20;
int ans[N][N], v1[N][N], v2[N][N], v3[N][N], a[N][N], b[N][N];
//v1[i][j]表示第i行j数字是否使用过
//v2[i][j]表示第i列j数字是否使用过
//v3[i][j]表示第i个九宫格j数字是否使用过

inline int id(int x, int y)
{
    int xx = (x-1)/3+1, yy = (y-1)/3+1;
    return (xx-1)*3+yy;
}
inline int cmp(int x, int y) { return x > y ? 1 : -1; }//不要写成bool！！ 
inline bool check(int x, int y, int v)
{
    if(v1[x][v] || v2[y][v] || v3[id(x, y)][v]) return 0;
    if(cmp(v, ans[x][y-1]) != a[x][y] && a[x][y]) return 0;
    if(cmp(v, ans[x-1][y]) != b[x][y] && b[x][y]) return 0;
    return 1;
}
inline void printmatrix()
{
    for(int i = 1; i <= 9; i ++)
        for(int j = 1; j <= 9; j ++) {
            printf("%d", ans[i][j]);
            putchar(j == 9 ? '\n' : ' ');
        }
}
inline void dfs(int x, int y)
{
    if(x > 9) { printmatrix(); exit(0); }
    for(int i = 1; i <= 9; i ++)
        if(check(x, y, i)) {
            ans[x][y] = i;
            v1[x][i] = v2[y][i] = v3[id(x, y)][i] = 1;
            if(y == 9) dfs(x+1, 1);
            else dfs(x, y+1);
            ans[x][y] = 0;
            v1[x][i] = v2[y][i] = v3[id(x, y)][i] = 0;
        }
}
int main()
{
    //freopen("sample.in", "r", stdin);
    //freopen("test.out", "w", stdout);
    char ch[5];
    for(int i = 1; i <= 9; i ++) {
        for(int j = 1; j <= 9; j ++)
            if(j%3) {
                scanf("%s", ch);
                if(ch[0] == '>') a[i][j+1] = -1;
                else a[i][j+1] = 1;
            }
        if(i%3) {
            for(int j = 1; j <= 9; j ++) {
                scanf("%s", ch);
                if(ch[0] == 'v') b[i+1][j] = -1;
                else b[i+1][j] = 1;
            }
        }
    }
    dfs(1, 1);
    return 0;
}