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;
}