WEEK4周记 模拟赛题C题———宇宙射线
一、题意
样例解释
二、算法
1.思路
首先提出一个粗糙的也是错误的思路
该题很自然而然的一个思路是,利用dfs或者bfs进行搜索遍历,直到将所有的点都搜索完。写的很爽,但是写完才发现总的时间复杂度是
O
(
2
n
)
O(2^n)
O(2n),显然超时了,只能过第一个点。
实际上,原版的dfs也不是这样写的,dfs在遇到已经走过的点之后会停止继续往下递归,也就是记忆性搜索,并进行恰当的剪枝操作。
在原有错误思路上进行改进
所以我们这里也要进行剪枝操作。
不难想到,题中的宇宙射线的分裂方式很容易产生重合的情况。这里所说的重合指的是,在同一深度,向同一方向,分裂出宇宙射线移动相同的格数。事实上,由题意可知在同一深度分裂出的宇宙射线移动的格数必然相同,所以我们只需要考虑深度和方向。
我们的剪枝条件是:如果递归到深度i,并向d方向分裂宇宙射线时,经过的第一个点将被标记。换句话说,如果再次以深度i方向d的“身份”到达这个点(并且这个点还是第一个被经过的点),那么直接停止向下递归。因为显然前面已经发生过与此时完全相同的情况,此时再往下递归不论多少层也都是前面走过的老路。
下面的代码就是dfs递归函数中的剪枝部分:
if(a[i]>0)//a[i]是第i次分裂前进的格数
{//单独处理第一个点
cur=cc(d,cur);//在该方向上前进一格
if(pos[i][d][cur.first][cur.second]==1)return;
//一模一样的分裂在此处已经发生过了
pos[i][d][cur.first][cur.second]=1;
if(pos[30][8][cur.first][cur.second]==0)
{//标记这个点被经过,防止重复记录格数(与剪枝无关)
ans++;
pos[30][8][cur.first][cur.second]=1;
}
}
需要注意的几个点
根据剪枝条件,只标记深度i方向d经过的第1个点,在这条路线上的第2个点……就都不再标记了。
初始出发的点并不需要被在进行dfs之前标记。比如我们从(150,150)这个点出发,实际上(150,150)并没有被经过,分裂后到达的第一个点是(150,151)。
三、代码
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
int a[31];
bool pos[31][9][300][300];
//int dir[300][300];
pair<int,int> cc(int i,pair<int,int> c)
{//在i方向上前进一格,返回下一个点
if(i==0) c.second++;
else if(i==1) c.first++,c.second++;
else if(i==2) c.first++;
else if(i==3) c.first++,c.second--;
else if(i==4) c.second--;
else if(i==5) c.first--,c.second--;
else if(i==6) c.first--;
else if(i==7) c.first--,c.second++;
return c;
}
int ans=0;
int n;
void dfs(int i,int d,pair<int,int> c)
{//i是深度,d是方向,c是初始点
if(i>=n) return;
//ans+=a[i];
pair<int,int> cur=c;
if(a[i]>0)
{//单独处理第一个点
cur=cc(d,cur);//在该方向上前进一格
if(pos[i][d][cur.first][cur.second]==1)return;
pos[i][d][cur.first][cur.second]=1;
if(pos[30][8][cur.first][cur.second]==0)
{
ans++;
pos[30][8][cur.first][cur.second]=1;
}
}
int j;
for(j=1;j<a[i];j++)
{//除第一个点之外该线路上剩余的点并不需要pos[i][d][cur.first][cur.second]=1;
cur=cc(d,cur);
if(pos[30][8][cur.first][cur.second]==0)
{
ans++;
//pos[i][d][cur.first][cur.second]=1;
pos[30][8][cur.first][cur.second]=1;
}
}
dfs(i+1,(d+1)%8,cur);
dfs(i+1,(d-1+8)%8,cur);//(d-1+8)%8才是7!!
}
int main()
{
scanf("%d",&n);
int i,j;
for(i=0;i<n;i++)
scanf("%d",&a[i]);
pair<int,int> cur(150,150);
//pos[0][0][150][150]=1;
//pos[30][8][150][150]=1;
//初始点并不需要初始化为1,第一次分裂会在第一层dfs中进行
dfs(0,0,cur);
//queue<int> q
//while()
printf("%d",ans);
return 0;
}