L2-043 龙龙送外卖

YuFei Zhou

已于 2024-03-28 21:48:24 修改

阅读量152

点赞数 3

分类专栏：刷题记录文章标签：算法数据结构 dfs

于 2024-03-27 22:01:18 首次发布

本文链接：https://blog.youkuaiyun.com/m0_74237658/article/details/137091148

版权

刷题记录专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一段C++代码，使用深度优先搜索（DFS）算法在给定的树结构中处理查询问题，涉及向量（vector）构建树，以及如何在树中追踪节点的深度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个题主要考察树的dfs......，我们使用vector建树

#include<bits/stdc++.h>
typedef unsigned long long ull;
typedef long long ll;
using namespace std;
const int N = 2e5 + 10;
const int M = 2e4 + 10;
const int mod  = 1e9 + 7;

vector<int> g[N];
int d[N],f[N];
bool st[N];
void dfs(int u){
    for (auto &j : g[u]) {
        d[j] = d[u] + 1;
        dfs(j);
    }
}
void solve(){
    int m,n;
    cin>>n>>m;
    int r;
    for (int i = 1; i <= n; ++i) {
        cin>>f[i];
        if(f[i] == -1) r = i;
        else{
            g[f[i]].push_back(i);
        }
    }
    dfs(r);
    int v = 0,maxx = 0;
    for (int i = 1; i <= m; ++i) {
        int x;
        cin>>x;
        maxx = max(maxx,d[x]);
        while (!st[x] && x != r){
            st[x] = true;
            v += 2;
            x = f[x];
        }
        cout<<v - maxx<<endl;
    }
}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL),cout.tie(NULL);
    int t = 1;
    //cin>>t;
    while(t--){
        solve();//1 2 3 4 5
    }
    return 0 ;
}