剪枝:这里只是用到了一个小剪枝对于a[x],b[y] 都要判断一下是不是已经等于0了
还有对于dfs和bfs 我们一定要分清哪一个是x轴哪一个是y轴(竖着的是x,横着的是y),我们一定要处理要输入输出,否则存的图就是不一样的,这样就大意失荆州,见祖宗了。
这里用到vector数组进行路径的记录;
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std ;
const int N = 50 ;
int a[N] , b[N] ;
int n ;
bool v[N][N] ;
int d[4][2] = {{0,1},{0,-1},{1,0},{-1,0}} ;
vector<pair<int,int>> q ;
void dfs(int x, int y){
//if(a[x] < 0 || b[y] < 0) return;
if(x==n-1 && y==n-1){
bool flag = 1 ;
for(int i = 0; i < n ; i ++) if(a[i]) flag = 0 ;
for(int i = 0 ; i < n ; i ++) if(b[i]) flag = 0 ;
if(flag){
for(int i = 0 ; i < q.size() ; i ++){
cout << q[i].first * n + q[i].second <<" " ;
}
}
}
for(int i = 0 ; i < 4 ; i ++){
int tx = x + d[i][0] , ty = y + d[i][1] ;
if(tx < 0|| tx>=n || ty < 0 || ty >= n) continue ;
if(v[tx][ty]) continue ;
a[tx] -- ; b[ty] -- ;
v[tx][ty] = 1 ;
q.push_back({tx,ty}) ;
dfs(tx,ty) ;
a[tx] ++ ; b[ty] ++ ;
v[tx][ty] = 0 ;
q.pop_back() ;
}
}
int main(){
cin >> n ;
for(int i = 0 ; i < n ; i++) cin >> b[i] ;
for(int i = 0 ; i < n ; i++) cin >> a[i] ;
v[0][0] = 1 ;
a[0] -- ,b[0] -- ;
q.push_back({0,0}) ;
dfs(0,0) ;
}