WEEK4 csp模拟 C-可怕的宇宙射线

C-可怕的宇宙射线

题目描述

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

Input

输入第一行包含一个正整数 ，表示宇宙射线会分裂 次 第二行包含n个正整数 ，第 i个数 ai表示第 i次分裂的宇宙射线会在它原方向上继续走多少个单位长度。

Output

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

Sample Input

4
4 2 2 3

Sample Output

39

数据点说明

数据点	n
10%	<=10
40%	<=20
100%	<=30

题解

• 结构体direction表示方向，其中po代表南北方向，ne代表东西方向，-1表示南和西，1表示北和东。结构体point表示平面上的某个点。bool型二维数组盘表示表示二维空间中的点。结构体record是为了方便记录我们走过的路线，减少重复，提高运算速度。定义一个set，存放record型数据，因为两条射线，如果分裂的次数、所在的点和要发射的方向都相同的话，那这两条射线后面所经过的点会完全相同，此时可据此进行剪枝。
• 将分裂次数和每步分裂后向前的格子数都输入后，运行count子函数。
• count函数中，运行一次count函数相当于射线分裂发射一次，参数a是存放每次射线分裂后前进几步的数组，dir是射线前进的方向，p是射线前进前的点，m指的是这次射线是第几次分裂后的，函数返回值是本次射线及其后续射线又占据了几个空格。若m等于n，则上一次已经是最后一次了，射线不能再前进，返回值0。若set中已经有一个record记录了，说明这条射线后面的点都已经被走过了，直接返回0；没有记录则将本次record插入set中。计算返回值ret时，检查本次射线经过路上的空格数（并将pan上的该空格置为1），然后算出下一次分裂出的两条射线的起始点和方向，调用count，将本次射线的经过空格数与两次count的返回值相加就是本次返回值。

代码

#include <iostream>
#include <cstdio>
#include <set>
#include <string.h>
using namespace std; 
struct direction{
	int po;// 北1，南-1，无0 
	int ne;// 东1，西-1，无0 
	bool operator>(const direction &di)const{
		return (po==di.po)?ne>di.ne:po>di.po;
	}
	bool operator<(const direction &di)const{
		return (po==di.po)?ne<di.ne:po<di.po;
	}
	bool operator==(const direction &di)const{
		return ne==di.ne&&po==di.po;
	}
}; 
struct point{
	int x,y;//当前运行到的点 
	bool operator>(const point &p)const{
		return (x==p.x)?y>p.y:x>p.x;
	}
	bool operator<(const point &p)const{
		return (x==p.x)?y<p.y:x<p.x;
	}
	bool operator==(const point &p)const{
		return y==p.y&&x==p.x;
	}
};
struct record{
	point pt;
	direction dt;
	int m;
	bool operator==(const record r)const{
		return pt==r.pt&&dt==r.dt&&m==r.m;
	}
	bool operator>(const record &r)const{
		if(m==r.m){
			if(pt==r.pt){
				return dt>r.dt;
			}else{return pt>r.pt;}
		}else{return m>r.m;}
	}
	bool operator<(const record &r)const{
		if(m==r.m){
			if(pt==r.pt){
				return dt<r.dt;
			}else{return pt<r.pt;}
		}else{return m<r.m;}
	}
};
set<record> st;
bool pan[400][400];
int n;

int count(int*a,direction dir,point p,int m){
	//m该运行a数组的第几个 
	if(m==n){return 0;}
	record r={p,dir,m};
	if(st.find(r)==st.end()){
		st.insert(r);
	}else{
	//	cout<<"p.x:"<<p.x<<" p.y:"<<p.y;
	//	cout<<" dir.po:"<<dir.po<<" dir.ne:";
	//	cout<<dir.ne<<" m:"<<m<<endl;
		return 0;
	}
	int ret=0; 
	for(int i=1;i<=a[m];i++){
		if(pan[p.x+i*dir.ne][p.y+i*dir.po]==0){
			ret++;
			pan[p.x+i*dir.ne][p.y+i*dir.po]=1;
		}
	}
	point pp={p.x+a[m]*dir.ne,p.y+a[m]*dir.po};
	direction dir1;
	direction dir2;
	if(dir.ne==0){
		dir1={dir.po,-1};
		dir2={dir.po,1};
	}else if(dir.po==0){
		dir1={-1,dir.ne};
		dir2={1,dir.ne};
	}
	else{
		dir1={0,dir.ne};
		dir2={dir.po,0};
	}
	return ret+count(a,dir1,pp,m+1)+count(a,dir2,pp,m+1);
}
int main(int argc, char** argv) {
	cin>>n; 
	int* a=new int[n];
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	memset(pan,0,sizeof(pan)); 
	direction dir={1,0};
	point p={200,199}; 
	int ans=count(a,dir,p,0);
	cout<<ans<<endl;
	return 0;
}