ybtoj·虫食算【DFS】

Description–

高效进阶「基础算法」第4章 深度搜索课堂过关 例题3

---

解题思路–

剪枝：（加数x,y，和z）

• 若当前位右边的数都确定了，判断当前位是否符合加法
• 若当前位右边的数没有确定，判断当前位是否符合加法（进位可为0或1）
• 判断和的位数是否大于n

---

代码–

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

int n, a[30], c[30];
bool gg, q[30], dis[30];
string s[5];

bool check()
{
	int g = 0, x, y, z;
	for (int i = n; i >= 1; --i)
	{
		x = a[s[1][i - 1] - 'A' + 1];
		y = a[s[2][i - 1] - 'A' + 1];
		z = a[s[3][i - 1] - 'A' + 1];
		if (x != -1 && y != -1 && z != -1)
	    {
	    	if (g != -1)
	    	{
	    		if ((x + y + g) % n != z) return 0;
	    		if (i == 1 && x + y + g >= n) return 0;
	    		g = (x + y + g) / n;
			}
			else
			{
				if ((x + y) % n != z && (x + y + 1) % n != z) return 0;
				if (i == 1 && x + y >= n) return 0;
			}
		}
		else g = -1;
	}
	if (g > 0) return 0;
	return 1;
}

void dfs(int lev)
{
	if (gg) return ;
	if (lev > n)
	{
		gg = 1;
		for (int i = 1; i <= n; ++i)
	      printf("%d ", a[i]);
	    return ;
	}
	for (int i = 0; i < n; ++i)
	  if (!q[i])
	  {
	  	  a[c[lev]] = i;
			 q[i] = 1;
	  	  if (check()) dfs(lev + 1);
	  	  a[c[lev]] = -1, q[i] = 0;
	  }
}

int main()
{
	scanf("%d", &n);
	for (int i = 1; i <= 3; ++i)
	  cin >> s[i];
	for (int i = n - 1; i >= 0; --i)
	{
		for (int j = 1; j <= 3; ++j)
	      if (!dis[s[j][i] - 'A' + 1])
	      {
	      	  dis[s[j][i] - 'A' + 1] = 1;
			  c[++c[0]] = s[j][i] - 'A' + 1;
		  }
	    a[i + 1] = -1;
	}
	gg = 0;
	dfs(1);
	
	return 0;
}