L2拯救007

本题为判断是否能到达终点的dfs

#include <bits/stdc++.h>
#define endl '\n'
#define INF 0x3f3f3f3f
#define ll long long
#define Mirai ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
#define COUT cout << "###" << endl;
#define pb push_back
using namespace std;
typedef pair<int, int> PII;
const int N = 110;
vector<int> g[N];
struct NODE
{
    int x, y;
    int num;
} Node[N];
bool vis[N];//用于dfs
int start[N];//存储可以第一个跳上的鳄鱼编号
int idx;
int n, m;
bool flag;//判断最终是否能够逃出
void dfs(int u)
{
    for (auto v : g[u])
    {
        if(v==-1)//因为-1不可以作为vis的下标，所以要在进行下一轮dfs之前判断该临点是否为重点(-1)
        {
            flag=true;
            cout<<"Yes";
            return ;
        }
        if (!vis[v])
        {
            vis[v]=true;//更新状态为已遍历
            dfs(v);//下一层dfs
            vis[v]=false;//还原状态为未遍历
            if (flag)//如果结果已经得出，则回溯
                return;
        }
    }
}
int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
    {
        int x, y;
        cin >> x >> y;
        Node[i] = {x, y, i};
    }
    for (int i = 1; i <= n; i++)
    {
        if ((Node[i].x * Node[i].x) + (Node[i].y * Node[i].y) <=(m+7.5)*(m+7.5))
            start[idx++] = i;//如果该点到原点距离小于小岛半径加上最大跳跃距离，则可以作为第一个跳上的鳄鱼
        if (Node[i].x >= 50 - m || Node[i].x <= -1 * (50 - m) || Node[i].y >= 50 - m || Node[i].y <= -1 * (50 - m))
            g[i].push_back(-1);//如果该点可以直接跳出鳄鱼池，则将临点加入一个-1来作为终点标志
        for (int j = 1; j <= n; j++)
        {
            if (i != j)
            {
                double dist = sqrt(pow(abs(Node[i].x - Node[j].x), 2) + pow(abs(Node[i].y - Node[j].y), 2));
                //dist用来存储两个鳄鱼之间的距离
                if (dist <= m)//如果此距离小于等于最大跳跃距离，则可跳
                {
                    g[i].push_back(j);
                }
            }
        }
    }
    for (int i = 0; i < idx; i++)
    {
        vis[start[i]] = true;//先将此点更新为已遍历状态
        dfs(start[i]);//dfs该点
        vis[start[i]] = false;//还原该点状态
        if (flag)//如果结果已经判断为可以逃出，则跳出循环
            break;
    }
    if (!flag)
        cout << "No";
    return 0;
}