leetcode 2359.找到离给定两个节点最近的节点

首先说一下超时的做法,其实也能过66个样例:

就是用BFS遍历存储路径,因为题目中已经说了至多一条边,所以对于每一点来说,不可能出现多路径问题,要么只有一条路径,要么就没有路,所以顺序存储的节点的距离是依次按1递增的(从自己的距离0开始)。

由于有两个出发点,所以我们用了两个数组存储这两个点的路径,然后判断这里面的哪些点是遇到的点相同的,记录一下它们之间的的最大距离,存储到最终的结果数组中。

在结果数组中筛选出距离最小的那个对应的节点就是答案了。

因为时间复杂度在判断相同结点的时候变高了,所以这个做法有一部分大数据过不了。

复制代码
class Solution {
public:
queue<int>q;
void bfs(int u,vector<vector<int>>&s,vector<bool>&st,vector<int>&res){
    q.push(u);
    res.push_back(u);
    st[u]=true;
    while(!q.empty()){
        int tmp=q.front();
        q.pop();
        for(int i=0;i<s[tmp].size();i++){
            if(!st[s[tmp][i]]){
                st[s[tmp][i]]=true;
                res.push_back(s[tmp][i]);
                q.push(s[tmp][i]);
            }
        }
    }
}
    int closestMeetingNode(vector<int>& edges, int node1, int node2) {
        vector<int>res1;
        vector<int>res2;
        vector<vector<int>>s(edges.size());
        vector<bool>st(edges.size(),false);
        for(int i=0;i<edges.size();i++){
            if(edges[i]!=-1){
                s[i].push_back(edges[i]);
            }
        }
        bfs(node1,s,st,res1);
        q=queue<int>();
        for(int i=0;i<st.size();i++)
        st[i]=false;
        bfs(node2,s,st,res2);
        vector<int>res(edges.size(),edges.size()+1);
        int mins=INT_MAX;
        for(int i=0;i<res1.size();i++){
            int tmp=res1[i];
            for(int j=0;j<res2.size();j++){
                if(tmp==res2[j]){
                    res[tmp]=max(j,i);
                }
            }
        }
            for(int i=0;i<res.size();i++){
                mins=min(mins,res[i]);
            }
            int index=-1;
            for(int i=0;i<res.size();i++){
                if(mins==edges.size()+1)
                break;
                else{
                if(res[i]==mins)
                {
                    index=i;
                    break;
                }
                }
            }
        return index;
    }
};

然后作者做了一些改进,就是用了哈希表映射,让程序看起来更长了,其实就是空间换时间而已。

把二重循环换掉了,匹配相同路径的做法,换成了用一个共同的数组计数,为2的时候说明这个是两个起点开始经过的共同点。

复制代码
class Solution {
public:
queue<int>q;
void bfs(int u,vector<vector<int>>&s,vector<bool>&st,vector<int>&res){
    q.push(u);
    res.push_back(u);
    st[u]=true;
    while(!q.empty()){
        int tmp=q.front();
        q.pop();
        for(int i=0;i<s[tmp].size();i++){
            if(!st[s[tmp][i]]){
                st[s[tmp][i]]=true;
                res.push_back(s[tmp][i]);
                q.push(s[tmp][i]);
            }
        }
    }
}
    int closestMeetingNode(vector<int>& edges, int node1, int node2) {
        vector<int>res1;
        vector<int>res2;
        vector<vector<int>>s(edges.size());
        vector<bool>st(edges.size(),false);
        for(int i=0;i<edges.size();i++){
            if(edges[i]!=-1){
                s[i].push_back(edges[i]);
            }
        }
        bfs(node1,s,st,res1);
        q=queue<int>();
        for(int i=0;i<st.size();i++)
        st[i]=false;
        bfs(node2,s,st,res2);
        vector<int>res(edges.size(),edges.size()+1);
        vector<int>counts(edges.size(),0);
        vector<int>m1(edges.size(),0);
        vector<int>m2(edges.size(),0);
        for(int i=0;i<res1.size();i++){
            counts[res1[i]]++;
            m1[res1[i]]=i;
        }
        for(int i=0;i<res2.size();i++){
            counts[res2[i]]++;
            m2[res2[i]]=i;
        }
        for(int i=0;i<counts.size();i++){
            if(counts[i]==2){
                res[i]=max(m2[i],m1[i]);
            }
            
        }
        int index=-1;
        int mins=INT_MAX;
        for(int i=0;i<res.size();i++){
            mins=min(mins,res[i]);
        }
        for(int i=0;i<res.size();i++){
            if(mins==edges.size()+1)
            break;
            else{
                if(mins==res[i]){
                    index=i;
                    break;
                }
            }
        }
        return index;
    }
};
相关推荐
凌肖战2 小时前
力扣网编程55题:跳跃游戏之逆向思维
算法·leetcode
黑听人2 小时前
【力扣 简单 C】70. 爬楼梯
c语言·leetcode
88号技师3 小时前
2025年6月一区-田忌赛马优化算法Tianji’s horse racing optimization-附Matlab免费代码
开发语言·算法·matlab·优化算法
ゞ 正在缓冲99%…3 小时前
leetcode918.环形子数组的最大和
数据结构·算法·leetcode·动态规划
Kaltistss4 小时前
98.验证二叉搜索树
算法·leetcode·职场和发展
知己如祭4 小时前
图论基础(DFS、BFS、拓扑排序)
算法
mit6.8244 小时前
[Cyclone] 哈希算法 | SIMD优化哈希计算 | 大数运算 (Int类)
算法·哈希算法
c++bug4 小时前
动态规划VS记忆化搜索(2)
算法·动态规划
哪 吒4 小时前
2025B卷 - 华为OD机试七日集训第5期 - 按算法分类,由易到难,循序渐进,玩转OD(Python/JS/C/C++)
python·算法·华为od·华为od机试·2025b卷
努力写代码的熊大5 小时前
单链表和双向链表
数据结构·链表