本蒟蒻在调试代码时发现当 n>10 时会 TLE,交上去评测只有 61 分,不知怎样优化。请求大佬援助。
#include<bits/stdc++.h>
using namespace std;
struct node
{
int x,y,cnt;
};
bool visx[15],visy[15],visleft[30],visright[30];
int ans[15],n,ansnum=0;
void print()
{
for (int i=1;i<=n;i++)
{
printf("%d ",ans[i]);
}
printf("\n");
}
void dfs(int x,int y,int cnt)
{
if (cnt==n)
{
ansnum++;
if (ansnum<=3)
{
print();
}
return;
}
if (x==n+1)
{
return;
}
if (y==n+1)
{
dfs(x+1,1,cnt);
return;
}
if (visx[x]==0&&visy[y]==0&&visleft[x+y]==0&&visright[x-y+13]==0)
{
visx[x]=1;
visy[y]=1;
visleft[x+y]=1;
visright[x-y+13]=1;
ans[cnt+1]=y;
dfs(x,y+1,cnt+1);
visx[x]=0;
visy[y]=0;
visleft[x+y]=0;
visright[x-y+13]=0;
}
dfs(x,y+1,cnt);
}
int main()
{
scanf("%d",&n);
dfs(1,1,0);
printf("%d",ansnum);
return 0;
}