洛谷 p1092 虫食算

时间:2022-04-29 00:49:32

题目链接: https://www.luogu.org/problemnew/show/P1092

这个题折腾了我好久 这其实本质上是一道凑算式的题目 ,让一个二维数组存算式,一个一位数组存字母分别代表哪个数字。需要注意的是数位还有位置的标号的问题 究竟是0-n还是1-n还是0-(n-1),这是需要仔细考虑的问题。

直接上代码吧

#include<bits/stdc++.h>

using namespace std;

char ss[][];
int n,flag[],s[][],vis[]; void dfs(int x,int y,int t)
{
if(x==)
{
if(t==)
{
for(int i=;i<n;i++)
cout<<flag[i]<<" ";
cout<<flag[n];
exit();
}
return;
}
for(int i=x-;i>=;i--) //剪枝,一旦不适合就返回
{
int a=flag[s[][i]];
int b=flag[s[][i]];
int c=flag[s[][i]];
if(a==-||b==-||c==-) //还没有赋值的先不管他
continue;
if((a+b)%n!=c&&(a+b+)%n!=c)
return;
}
if(flag[s[y][x]]==-) //如果这一位的字母还没有赋值
{
for(int i=n-;i>=;i--) //要注意 取值是0到n-1
if(vis[i]==)
{
if(y!=) //最后一行和其他的有不同 要分开讨论
{
flag[s[y][x]]=i;vis[i]=;
dfs(x,y+,t);
flag[s[y][x]]=-;vis[i]=;
} else
{
int w=flag[s[][x]]+flag[s[][x]]+t;
if(w%n!=i)
continue;
flag[s[][x]]=i;vis[i]=;
dfs(x-,,w/n);
flag[s[][x]]=-; vis[i]=;
}
} }
else //如果已经赋值
{
if(y!=)
dfs(x,y+,t);
else
{
int w=flag[s[][x]]+flag[s[][x]]+t;
if (w%n!=flag[s[][x]]) //剪枝
return;
dfs(x-,,w/n);
}
} } int main()
{ cin>>n;
int i,j;
for(i=;i<;i++)
{
cin>>ss[i];
}
for(i=;i<=;i++)
for(j=;j<=n;j++)
{
s[i][j]=ss[i-][j-]-'A'+; //为了方便使用 我直接让角标从1开始
}
memset(flag,-,sizeof(flag));
dfs(n,,); }