[0x20]虫食算

青铜世纪

已于 2022-04-12 23:54:19 修改

阅读量285

点赞数

分类专栏： 0x20 搜索文章标签：算法

于 2022-04-12 23:53:43 首次发布

本文链接：https://blog.csdn.net/Joker15517/article/details/124137823

版权

0x20 搜索专栏收录该内容

2 篇文章 0 订阅

订阅专栏

虫食算

思路

搜索顺序：枚举每个字母代表哪个数字

剪枝：

按照从右到左的顺序从算式中枚举每个字母的选法
检查算式的合法性，如果不合法及时回溯
1. 从后往前枚举每一列，设当前列的加数分别为 $a, b, c$ 如果这一列右边的数可以确定，那到这一列的进位t也是可以确定的，如果 $a + b + t \neq = c$ 则当前方案不合法
2. 如果这一列右边的数有不能确定的，则进位不能确定，但有0和1两种取值，因此如果 $a + b + 0 \neq = c 或 a + b + 1 \neq = c$ ，则该方案不合法
3. 如果最高位有进位，则该方案不合法

code

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 30;
int n;
char e[3][N];
bool st[N];
int path[N], q[N];

inline bool check()
{
    int t = 0;
    for (int i = n-1; i >= 0; i --)
    {
        int a = e[0][i] - 'A', b = e[1][i] - 'A', c = e[2][i] - 'A';
        a = path[a], b = path[b], c = path[c];
        if(a != -1 && b != -1 && c != -1)
        {
            if(t == -1)
            {
                if((a + b + 0) % n != c && (a + b + 1) % n != c) return false;
                if(!i && a + b >= n) return false;
            }
            else
            {
                if((a + b + t) % n != c) return false;
                if(!i && a + b + t >= n) return false;
                t = (a + b + t) / n;
            }
        }
        else
        {
            t = -1;
        }
    }
    return true;
}

bool dfs(int u)
{
    if(u == n) return true;
    
    for (int i = 0; i < n; i ++)
    {
        if(!st[i])
        {
            st[i] = true;
            path[q[u]] = i;
            if(check() && dfs(u+1)) return true;
            st[i] = false;
            path[q[u]] = -1;
        }
    }
    return false;
}

int main()
{
    scanf("%d", &n);
    for (int i = 0; i < 3; i ++) scanf("%s", e[i]);
    for (int i = n - 1, k = 0; i >= 0; i --)
    {
        for (int j = 0; j < 3; j ++)
        {
            int t = e[j][i] - 'A';
            if(!st[t])
            {
                st[t] = true;
                q[k ++] = t;
            }
        }
    }
    memset(st, false, sizeof st);
    memset(path, -1, sizeof path);
    dfs(0);
    for (int i = 0; i < n; i ++) printf("%d ", path[i]);
    return 0;
}