P8642 [蓝桥杯 2016 国 AC] 路径之谜

P8642

独立完成。

搜索。

从 $y$ 方向射向 $x$ 轴记为 $liney$，表示这一列应该射多少次。

从 $x$ 射向 $y$ 轴同理。

主要是这个顺序把一开始绕晕了。

所以先输入的其实是 $y$ 射向 $x$ 轴，也就是 $liney$，然后才是 $linex$。

然后要加一个剪枝，如果当前的答案已经超出了限制，就可以不用继续搜下去了。这样可以大大提高效率。

#include<bits/stdc++.h>
using namespace std;
const int N =3000;
struct node{
	int x,y;
}ans[N][N];
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
int vis[N][N];
int linex[N],liney[N],lastx[N],lasty[N],n;
bool check(int x,int y){
	return (x>=1&&x<=n&&y>=1&&y<=n&&vis[x][y]==0);
}
bool checkans(){
	for(int i=1;i<=n;i++)	if(lastx[i]!=linex[i])	return false;
	for(int i=1;i<=n;i++)	if(lasty[i]!=liney[i])	return false;
	return true;
}
void Print(int x,int y){
	if(x==1&&y==1){
		cout<<0<<" ";
	}
	else{
		Print(ans[x][y].x,ans[x][y].y);
		cout<<(x-1)*n+y-1<<" ";
	}
}
void dfs(int x,int y){
	if(lastx[x]>linex[x])	return ;
	if(lasty[y]>liney[y])	return ;
	if(x==n&&y==n&&checkans()){
		Print(n,n);
		return ;
	}
	for(int i=0;i<=3;i++){
		int xx=x+dx[i],yy=y+dy[i];
		if(!check(xx,yy))	continue;
		lasty[yy]++,lastx[xx]++;
		ans[xx][yy].x=x,ans[xx][yy].y=y;
		vis[xx][yy]=1;
		dfs(xx,yy);
		lasty[yy]--,lastx[xx]--;
		vis[xx][yy]=0;
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++)	cin>>liney[i];
	for(int i=1;i<=n;i++)	cin>>linex[i];
	lastx[1]=1,lasty[1]=1,vis[1][1]=1;
	dfs(1,1);
	cout<<endl;
	return 0;
}