題目描述
小明冒充X星球的騎士,進入了一個奇怪的城堡。城堡裏邊什麼都沒有,只有方形石頭鋪成的地面。
假設城堡地面是 n x n 個方格。
按習俗,騎士要從西北角走到東南角。可以橫向或縱向移動,但不能斜着走,也不能跳躍。
每走到一個新方格,就要向正北方和正西方各射一箭。(城堡的西牆和北牆內各有 n 個靶子)
同一個方格只允許經過一次。但不必走完所有的方格。
如果只給出靶子上箭的數目,你能推斷出騎士的行走路線嗎?
有時是可以的,比如圖中的例子。
本題的要求就是已知箭靶數字,求騎士的行走路徑(測試數據保證路徑唯一)
輸入
第一行一個整數N(0<N<20),表示地面有 N x N 個方格
第二行N個整數,空格分開,表示北邊的箭靶上的數字(自西向東)
第三行N個整數,空格分開,表示西邊的箭靶上的數字(自北向南)
輸出
一行若干個整數,表示騎士路徑
爲了方便表示,我們約定每個小格子用一個數字代表,從西北角開始編號: 0,1,2,3…
比如,上圖中的方塊編號爲:
0 1 2 3
4 5 6 7
8 9 10 11
12 13 14 15
樣例輸入
4
2 4 3 4
4 3 3 3
樣例輸出
0 4 5 1 2 3 7 11 10 9 13 14 15
AC代碼
#include<iostream>
#include<vector>
using namespace std;
const int N=25;
int top[N];
int L[N];
bool visit[N][N];
int base[4][2]={{-1,0},{1,0},{0,-1},{0,1}};
vector<int> v;
int n;
bool empty(){
for(int i=0;i<n;i++){
if(top[i]!=0){
return false;
}
}
for(int i=0;i<n;i++){
if(L[i]!=0){
return false;
}
}
return true;
}
bool judge(int r,int c){
if(r<0||c<0){
return false;
}
if(r>=n||c>=n){
return false;
}
if(visit[r][c]){
return false;
}
if(L[r]<=0||top[c]<=0){
return false;
}
return true;
}
void dfs(int row,int col){
L[row]--;
top[col]--;
visit[row][col]=true;
v.push_back(row*n+col);
//cout<<row*n+col<<endl;
if(row==n-1&&col==n-1){
if(empty()){
for(int i=0;i<v.size();i++){
cout<<v[i]<<" ";
}
}
return ;
}
for(int i=0;i<4;i++){
int r=row+base[i][0];
int c=col+base[i][1];
if(judge(r,c)){
dfs(r,c);
L[r]++;
top[c]++;
visit[r][c]=false;
v.pop_back();
}
}
}
int main(){
cin>>n;
for(int i=0;i<n;i++){
cin>>top[i];
}
for(int i=0;i<n;i++){
cin>>L[i];
}
dfs(0,0);
return 0;
}