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;
    }
};
相关推荐
Two_brushes.2 分钟前
【算法】宽度优先遍历BFS
算法·leetcode·哈希算法·宽度优先
森焱森2 小时前
水下航行器外形分类详解
c语言·单片机·算法·架构·无人机
QuantumStack4 小时前
【C++ 真题】P1104 生日
开发语言·c++·算法
写个博客5 小时前
暑假算法日记第一天
算法
绿皮的猪猪侠5 小时前
算法笔记上机训练实战指南刷题
笔记·算法·pta·上机·浙大
hie988945 小时前
MATLAB锂离子电池伪二维(P2D)模型实现
人工智能·算法·matlab
杰克尼6 小时前
BM5 合并k个已排序的链表
数据结构·算法·链表
.30-06Springfield6 小时前
决策树(Decision tree)算法详解(ID3、C4.5、CART)
人工智能·python·算法·决策树·机器学习
我不是哆啦A梦6 小时前
破解风电运维“百模大战”困局,机械版ChatGPT诞生?
运维·人工智能·python·算法·chatgpt
xiaolang_8616_wjl6 小时前
c++文字游戏_闯关打怪
开发语言·数据结构·c++·算法·c++20