洛谷 P3958 奶酪

zhaowangji

已于 2023-08-31 21:35:50 修改

阅读量165

点赞数

文章标签：题解

于 2020-12-04 19:58:37 首次发布

本文链接：https://blog.csdn.net/qq_42636949/article/details/110671618

版权

读题，不难想到深搜。
然后加上一个 vis 数组记录这个地方有没有来过，如果来过肯定就不再来了
同时，你还需要注意“坐标的绝对值不超过 1,000,000,000”，记得算两点距离的时候要改成 long long 类型

#include<bits/stdc++.h>
#define reg register
#define ll long long
using namespace std;
int t;
struct node{
	int x;
	int y;
	int z;
}che[1007];
int vis[1007],Flag;
int n,h,r;
int Read();
double dis(int a,int b){
	return sqrt(((ll)che[a].x-(ll)che[b].x)*((ll)che[a].x-(ll)che[b].x)+((ll)che[a].y-(ll)che[b].y)*((ll)che[a].y-(ll)che[b].y)+((ll)che[a].z-(ll)che[b].z)*((ll)che[a].z-(ll)che[b].z));
}
void dfs(int k){
	if(che[k].z+r>=h){
		Flag=1;
		return;
	}
	for(reg int i=1;i<=n;++i)
		if(!vis[i]&&dis(k,i)<=2.0*r&&!Flag){//记得是2.0
			vis[i]=1;
			dfs(i);
			vis[i]=0;
		}
	return;
}
int main(){
	t=Read();
	while(t--){
		memset(vis,0,sizeof(vis));
		Flag=0;
		n=Read();h=Read();r=Read();
		for(reg int i=1;i<=n;++i){
			che[i].x=Read();
			che[i].y=Read();
			che[i].z=Read();
		}
		for(reg int i=1;i<=n;++i)
			if(!Flag&&(che[i].z-r)<=0){
				vis[i]=1;
				dfs(i);
				vis[i]=0;
			}
		Flag==1?cout<<"Yes\n":cout<<"No\n";
	}
	return 0;
}
int Read(){
	char ch=getchar();
	int res=0,flag=1;
	while(ch<'0'||ch>'9'){
		if(ch=='-')flag=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9'){
		res=(res<<3)+(res<<1)+ch-'0';
		ch=getchar();
	}
	return res*flag;
}

但是，有 30 分是会 TLE 的
那么，再进行思考：
一般我们写深搜时，总是要回溯，尤其是新学深搜的时候，老师一定会强调又强调

但是，这道题，真的有必要吗？

只要来过了，就一定会走下去，就一定有结果

那么，如果第一次就已经完成目标，还会有第二次吗？

不会。所以，无需回溯

所以，把上文代码中的

vis[i]=0;

全都去掉就行了

#include<bits/stdc++.h>
#define reg register
#define ll long long
using namespace std;
int t;
struct node{
	int x;
	int y;
	int z;
}che[1007];
int vis[1007],Flag;
int n,h,r;
int Read();
double dis(int a,int b){
	return sqrt(((ll)che[a].x-(ll)che[b].x)*((ll)che[a].x-(ll)che[b].x)+((ll)che[a].y-(ll)che[b].y)*((ll)che[a].y-(ll)che[b].y)+((ll)che[a].z-(ll)che[b].z)*((ll)che[a].z-(ll)che[b].z));
}
void dfs(int k){
	if(che[k].z+r>=h){
		Flag=1;
		return;
	}
	for(reg int i=1;i<=n;++i)
		if(!vis[i]&&dis(k,i)<=2.0*r&&!Flag){
			vis[i]=1;
			dfs(i);
		}
	return;
}
int main(){
	t=Read();
	while(t--){
		memset(vis,0,sizeof(vis));
		Flag=0;
		n=Read();h=Read();r=Read();
		for(reg int i=1;i<=n;++i){
			che[i].x=Read();
			che[i].y=Read();
			che[i].z=Read();
		}
		for(reg int i=1;i<=n;++i)
			if(!Flag&&(che[i].z-r)<=0){
				vis[i]=1;
				dfs(i);
			}
		Flag==1?cout<<"Yes\n":cout<<"No\n";
	}
	return 0;
}
int Read(){
	char ch=getchar();
	int res=0,flag=1;
	while(ch<'0'||ch>'9'){
		if(ch=='-')flag=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9'){
		res=(res<<3)+(res<<1)+ch-'0';
		ch=getchar();
	}
	return res*flag;
}