题解 | 2023 年牛客多校第三场 E 题

Koraidon, Miraidon and DFS Shortest Path

https://ac.nowcoder.com/acm/contest/57357/E

题意:给定一张 G(n,m)G(n,m) 的有向图,使用 dfs 算法求解从 11 开始的单源最短路,问给定的图能否在任何边遍历顺序下都正确输出。1n,m1051 \le n,m\le 10^5

解法:为什么我们要找支配树?可以考虑以下三个例子:

alt

基本错误型:一个点(22)有多种到达方式。

alt

一个环:多次遍历到的点不一定是有重复的。

破环:多种遍历方式会导致破环方式不同(如 646 \to 4424 \to 2535 \to 3),从而影响答案。

首先我们依然可以建立一个 bfs 树,找到从 11 出发到每个点的最短距离。然后考虑原图中的每一条边:

  1. 如果当前一条边 (u,v)(u,v) 连接的两点是同层的——必然出现了图 1 基本型,因而输出 No
  2. 如果当前一条边 (u,v)(u,v) 连接的两点是 uuvv 深的——该边会在 bfs 树中用到,且一定满足 depu+1=depv{\rm dep}_u+1={\rm dep}_v。不关心它对答案的影响。
  3. 如果当前一条边 (u,v)(u,v) 连接了不同层的两点(uuvv 浅)——可能出现图 2(正确)或者图 3(错误)的情况。这时要分析从 11 出发是不是必须经过 vvuu。如果必须经过,则是图 2 情况。否则,一定可以经过一条不经过 vv 的道路到达 uu(支配关系定义),这时更新 vv 的答案会出错(depu+1>depu>depv{\rm dep}_u+1 >{\rm dep}_u > {\rm dep}_v)。因而这个在支配树上维护支配关系即可。
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 5e5 + 10;
namespace dtree // 支配树模板
{
    const int MAXN = 500020;
    vector<int> E[MAXN], RE[MAXN], rdom[MAXN];

    int S[MAXN], RS[MAXN], cs;
    int par[MAXN], val[MAXN], sdom[MAXN], rp[MAXN], dom[MAXN];

    void clear(int n)
    {
        cs = 0;
        for (int i = 0; i <= n; i++)
        {
            par[i] = val[i] = sdom[i] = rp[i] = dom[i] = S[i] = RS[i] = 0;
            E[i].clear();
            RE[i].clear();
            rdom[i].clear();
        }
    }
    void add_edge(int x, int y) { E[x].push_back(y); }
    void Union(int x, int y) { par[x] = y; }
    int Find(int x, int c = 0)
    {
        if (par[x] == x)
            return c ? -1 : x;
        int p = Find(par[x], 1);
        if (p == -1)
            return c ? par[x] : val[x];
        if (sdom[val[x]] > sdom[val[par[x]]])
            val[x] = val[par[x]];
        par[x] = p;
        return c ? p : val[x];
    }
    void dfs(int x)
    {
        RS[S[x] = ++cs] = x;
        par[cs] = sdom[cs] = val[cs] = cs;
        for (int e : E[x])
        {
            if (S[e] == 0)
                dfs(e), rp[S[e]] = S[x];
            RE[S[e]].push_back(S[x]);
        }
    }
    int solve(int s, int *up)
    {
        dfs(s);
        for (int i = cs; i; i--)
        {
            for (int e : RE[i])
                sdom[i] = min(sdom[i], sdom[Find(e)]);
            if (i > 1)
                rdom[sdom[i]].push_back(i);
            for (int e : rdom[i])
            {
                int p = Find(e);
                if (sdom[p] == i)
                    dom[e] = i;
                else
                    dom[e] = p;
            }
            if (i > 1)
                Union(i, rp[i]);
        }
        for (int i = 2; i <= cs; i++)
            if (sdom[i] != dom[i])
                dom[i] = dom[dom[i]];
        for (int i = 2; i <= cs; i++)
            up[RS[i]] = RS[dom[i]];
        return cs;
    }
}
int up[MAXN];
vector<int> G[MAXN];
int dep[MAXN], f[21][MAXN];
void dfs(int x, int fa)
{
    dep[x] = dep[fa] + 1;
    for (int i = 0; i <= 19; i++)
        f[i + 1][x] = f[i][f[i][x]];
    for (auto it : G[x])
    {
        if (it == fa)
            continue;
        f[0][it] = x;
        dfs(it, x);
    }
}
int lca(int x, int y)
{
    if (dep[x] < dep[y])
        swap(x, y);
    for (int i = 20; i >= 0; i--)
    {
        if (dep[f[i][x]] >= dep[y])
            x = f[i][x];
        if (x == y)
            return x;
    }
    for (int i = 20; i >= 0; i--)
        if (f[i][x] != f[i][y])
            x = f[i][x], y = f[i][y];
    return f[0][x];
}
string solve()
{
    int n, m;
    cin >> n >> m;
    dtree::clear(n);
    for (int i = 1; i <= n; i++)
        G[i].clear();
    for (int i = 1; i <= m; i++)
    {
        int x, y;
        cin >> x >> y;
        dtree::E[x].push_back(y);
    }
    dtree::solve(1, up);
    for (int i = 2; i <= n; i++)
        G[up[i]].push_back(i);
    dfs(1, 0);
    const int inf = 1e9;
    vector<int> dis(n + 1, inf);
    dis[1] = 0;
    queue<int> q;
    q.push(1);
    bool flag = true;
    while (!q.empty())
    {
        int x = q.front();
        q.pop();
        for (auto it : dtree::E[x])
            if (dis[it] == inf)
            {
                q.push(it);
                dis[it] = dis[x] + 1;
                continue;
            }
            else if (dis[it] != dis[x] + 1)
            {
                if (lca(it, x) != it)
                    flag = false;
            }
    }
    if (flag)
        return "Yes";
    else
        return "No";
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T;
    cin >> T;
    while (T--)
        cout << solve() << "\n";
    return 0;
}
全部评论

相关推荐

昨天 20:22
重庆大学 Java
字节暑期刚入职四天,因为是年前,所以很多正职都放假走了,也就没有给我分配mt,然后有一个老哥在我来的时候给我发了一个landing手册,然后还有关于部门业务的白皮书,还有一些业务代码。然后本人是java面的,进来第一次接触go语言&nbsp;前面几天熟悉了一下go的语法和go的框架,可以读但是还不太会写,然后业务白皮书也看的很头疼,包括landing手册里要了解的很多东西说实话我看文档真的快看死了,一个嵌套一个,问题是我还完全不知道咋用这个我了解的东西,还有就是那个项目代码,那个老哥喊我去写写单测,熟悉一下go的语法,但也进行的很困难(这是我第一段实习,之前都是springboot那一套,真不太熟悉这个)想问问大家的建议,就是我从现在开始到在开年回来之前应该做些什么,我目前就一个想法&nbsp;就是复现一个landing手册上的go框架小项目&nbsp;就是相当于帮自己锻炼锻炼怎么写go&nbsp;或者各位大佬有没有更好的锻炼go语法的建议还有就是大家都在说vibe&nbsp;coding,那我应该怎么锻炼自己使用ai的能力,感觉我除了给一些需求然后它给我生成代码,好像就没别的用法了,那些什么工作流、拆解、skill啥的都不知道从哪一个地方开始,包括我现在正在实习,不知道精力该怎么分配,去网上想找找关于agent开发的一些学习流程,说实话,众说纷纭,有的是从python开始打基础然后系统学那些rag&nbsp;prompt&nbsp;langchain&nbsp;mcp等等,有的是说直接找一个github上的ai项目然后反复问ai,我确实有点迷茫,恳求各位大佬能留下你们宝贵的建议,我一定认真反复深刻学习有一说一&nbsp;我觉得字节饭挺好吃的!
Jasonnnnnn...:直接把项目代码喂给AI然后让它帮你分析,如果组里已经有一些流程图总结的话最好,没有的话自己画一个 Go的话其实只要把基础语法搞明白就行了,项目里很多都是直接让ai帮你写好然后自己稍微改下,不用学的特别深 ai的话,可以自己写一些md文件来搞点小东西,但除非你打算转算法,否则不用把rag langchain学的特别深,了解下就行了
字节跳动公司福利 1371人发布
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务