SDU 程序思维与实践week4 可怕的宇宙射线

C 可怕的宇宙射线

题目描述

众所周知，瑞神已经达到了CS本科生的天花板，但殊不知天外有天，人外有苟。在浩瀚的宇宙中，存在着一种叫做苟狗的生物，这种生物天生就能达到人类研究生的知识水平，并且天生擅长CSP，甚至有全国第一的水平！但最可怕的是，它可以发出宇宙射线！宇宙射线可以摧毁人的智商，进行降智打击！
宇宙射线会在无限的二维平面上传播（可以看做一个二维网格图），初始方向默认向上。宇宙射线会在发射出一段距离后分裂，向该方向的左右45°方向分裂出两条宇宙射线，同时威力不变！宇宙射线会分裂 次，每次分裂后会在分裂方向前进ai个单位长度。现在瑞神要带着他的小弟们挑战苟狗，但是瑞神不想让自己的智商降到普通本科生zjm那么菜的水平，所以瑞神来请求你帮他计算出共有多少个位置会被"降智打击"。

时间内存限制

每个测试点 1000ms 262144KB

输入说明

输入第一行包含一个正整数 n（n<=30)，表示宇宙射线会分裂n 次
第二行包含n个正整数a1,a2···an ，第ai 个数表示第 i次分裂的宇宙射线会在它原方向上继续走多少个单位长度

输出说明

输出一个数ans ，表示有多少个位置会被降智打击

输入样例

4
4 2 2 3

输出样例

39

测试点说明

• 10% n<=10
• 20% n<=20
• 30% n<=30

思路分析

1、利用朴素的DFS/BFS时间复杂度以及空间复杂度均为指数增长，会出现TLE 或者 MLE，只能拿到40%的分数
2、本题具有明显的对称性。射线每次分裂均是沿着本次前进的方向左右对称，故只需考虑一侧即可。在实际操作中只需考虑不断向右分裂，然后沿对称轴对称。
3、越往后分裂的射线对称轴数越多，需要对称的次数也越多。本题采用DFS+回溯法，让射线不断向右分裂，然后从最后一步向前回溯。每次DFS步骤如下：

• 第i次向右分裂的过程中首先DFS第i+1次向右分裂，这样第i次分裂时就获得了第i+1（及其以后）次分裂到达的点。
• 对已获得的第i+1(及其以后)次分裂的点进行对称构造。
• 执行第i次（本次）分裂的前进步骤。

4、判断重复点的方法：利用STL的set自动去重

对称点的计算方法

由平面几何以及全等三角形知识可得出对称点的坐标。
（1）垂直对称：

（2）水平对称：

（3）与x轴正方形成45°对称：

（4）与x轴负方向成45°对称：

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
using namespace std;
struct point{
	int x,y;
	bool operator <(const point & p) const{
		if(x!=p.x){
			return x<p.x;
		}else {
			return y<p.y;
		}	
	}
};
set<point> s;
int n;
int dx[]={0,1,1,1,0,-1,-1,-1};
int dy[]={1,1,0,-1,-1,-1,0,1};
int arr[31];
void dfs(int x,int y,int dir,int ord){
	if(ord>n){
		return;
	}
	//向右分裂 
	dfs(x+dx[dir]*arr[ord],y+dy[dir]*arr[ord],(dir+1)%8,ord+1);
	//对称构造 
	set<point> temp; 
	for(auto &i:s){
		if(dir==0||dir==4){
			temp.insert({2*x-i.x,i.y});
		}
		else if(dir==1||dir==5){
			temp.insert({x+i.y-y,y+i.x-x});
		}
		else if(dir==2||dir==6){
			temp.insert({i.x,2*y-i.y});
		}
		else if(dir==3||dir==7){
			temp.insert({x+y-i.y,y-i.x+x});
		}
	} 
	s.insert(temp.begin(),temp.end());	
	//本次前进 
	for(int i=0;i<arr[ord];i++){		
		x=x+dx[dir];
		y=y+dy[dir];
		s.insert({x,y});			
	}

} 
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>arr[i];
	}
	dfs(0,0,0,1);	
	cout<<s.size()<<endl;	
	return 0;
}

官方解析