习题7-2:黄金图形
题意:
一个探险队要去参观城市,走n次,每次走1,2,3,4,5,…n步,参观停留的城市,每次方向转90度。路途中有一些障碍物是不能走的。求所有走法,按字典序输出。注意,题目没说每个城市只能参观一次,但是实际上每个城市就是只能参观一次不然会WA
思路:
深搜回溯,可以加一个剪枝,如果走到一个地方剩下的步数回不到原点了,直接回溯。
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<string>
#include<cmath>
#include<set>
#include<queue>
#include<map>
#include<stack>
#include<vector>
#include<list>
#include<deque>
using namespace std;
typedef long long ll;
const int tmp = 105;
const int maxn = 250;
const double eps = 1e-6;
const int INF = 1 << 30;
int T, n, k;
int G[maxn][maxn];///-1表示障碍点,1表示该次路径已经过了的点,0表示未过点
char ans[] = "ensw";
const int dir[4][2] = {1,0,0,1,0,-1,-1,0};
int sum[50], ansn, vv[250];
void init()
{
ansn = 0;
memset(G, 0, sizeof(G));
scanf("%d%d", &n, &k);
int a, b;
while(k--)
{
scanf("%d%d",&a,&b);
if(abs(a) > tmp || abs(b) > tmp)continue;///如果直接大于tmp,忽略,因为20步之内不可能超过tmp
G[a + tmp][b + tmp] = -1;
}
}
bool judge(int x, int y, int d, int v)
{
for(int i = 1; i <= d; i++)
{
x += dir[v][0];
y += dir[v][1];
if(abs(x) > tmp || abs(y) > tmp)continue;
if(G[x + tmp][y + tmp] == -1)return true;
}
if(abs(x) + abs(y) > sum[20] - sum[d])return true;///剪枝,当前可以走的最大步数小于到原点的距离直接回溯
return false;
}
void dfs(int x, int y, int d, int v)
{
if(d == n)
{
if(!x && !y)
{
for(int i = 0; i < n; i++)printf("%c", ans[vv[i]]);
cout<<endl;
ansn++;
}
return;
}
for(int i = 0; i < 4; i++)
{
if(i == v || i + v == 3)continue;///不能走同一方向或者相反方向
if(judge(x, y, d, i))continue;
int xx = x + dir[i][0] * (d + 1);
int yy = y + dir[i][1] * (d + 1);
if(G[xx + tmp][yy + tmp])continue;
vv[d] = i;
G[xx + tmp][yy + tmp] = 1;
dfs(xx, yy, d + 1, i);
G[xx + tmp][yy + tmp] = 0;
}
}
void solve()
{
dfs(0, 0, 0, -1);
printf("Found %d golygon(s).\n\n", ansn);
}
int main()
{
sum[0] = 0;
for(int i = 1; i < 21; i++)sum[i] = sum[i - 1] + i;
scanf("%d",&T);
while(T--)
{
init();
solve();
}
return 0;
}