PTA 拯救007(升级版) (DFS,加最大的一个测试点)

拯救007(升级版)

思路:
首先找到从小岛能跳的鳄鱼,存储到优先队列中。因为题目中要求,如果最短路径不唯一,则输出第一跳最近的那个解。
如果能出去那就比较当前最小的,保存即可。但是需要用到回溯,不然最大的测试点过不去,理由如下:
最大的测试点可能的图
如图,(12345是输入顺序,竖线是岸边)假如说从湖中心小岛只能跳到1,那从1开始dfs,次序是 1 2 3 4 成功, 然后就不会有 1 5 4 成功了,所以要用回溯。
最后要特判能一步跳出。

代码如下:

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define SZ(x) (ll)x.size()
#define rep(i,a,b) for(ll i=(a);i<=(b);++i)
#define per(i,a,b) for(ll i=(a);i>=(b);--i)
#define mem(a,b) memset(a,b,sizeof a)
using namespace std;
typedef long long ll;
typedef vector<ll> vi;
typedef pair<int,int> pii;
typedef struct Node *node;
template<class T>
inline void read(T &x) {
	x=0;
	ll f=0;
	char c=getchar();
	while(!isdigit(c)) {
		f|=(c=='-');
		c=getchar();
	}
	while(isdigit(c)) {
		x=(x<<3)+(x<<1)+c-'0';
		c=getchar();
	}
	if(f) x=-x;
}
const int maxn=1e3+5,inf=0x3f3f3f3f;
struct Node {
	ll x=0,y=0;
} g[maxn];
struct mil {
	ll pos;
	double dis;
	bool operator<(const mil b)const {
		return dis>b.dis;
	}
};
ll n,d,vis[maxn];
ll mi=LLONG_MAX,ret[maxn],step[maxn];
double dist(ll x,ll y) {
	return sqrt((g[x].x-g[y].x)*(g[x].x-g[y].x)+(g[x].y-g[y].y)*(g[x].y-g[y].y));
}
void dfs(ll x,ll num) {
	if(!(abs(g[x].x)+d<50 && abs(g[x].y)+d<50)) {
		if(num<mi){
			mi=num;
			memcpy(ret,step,sizeof step);
		}
		return ;
	}
	vis[x]=1;
	rep(i,1,n) {
		double dis=dist(x,i);
		if(!vis[i] && dis<=d){
			step[num]=i;
			dfs(i,num+1);
		}
	}
	vis[x]=0;
}
int main() {
	priority_queue<mil> q;
	read(n);  //输入函数,相当于cin
	read(d);
	rep(i,1,n) {
		cin >> g[i].x >> g[i].y;
		double dis=dist(i,0);
		if(dis<=d+7.5) {
			q.push({i,dis});
		}
	}
	if(d>=50-7.5){
		cout << "1";
		return 0;
	}
	while(!q.empty()) {
		mil t=q.top();
		q.pop();
		step[0]=t.pos;
		dfs(t.pos,1);
	}
	if(mi!=LLONG_MAX) {
		cout << mi+1 << endl;
		rep(i,0,mi-1){
			cout << g[ret[i]].x << " " << g[ret[i]].y << endl;
		}
	}
	else
		puts("0");
	return 0;
}

发布了19 篇原创文章 · 获赞 15 · 访问量 2466

猜你喜欢

转载自blog.csdn.net/qq_41829380/article/details/104296000