ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

[CF1294F] Three Paths on a Tree - 树的直径

2021-02-11 15:32:33  阅读:124  来源: 互联网

标签:Paths tar int Tree Three dfs back push path


[CF1294F] Three Paths on a Tree - 树的直径

Description

给定一棵含 \(n\ (3\leq n\leq2\cdot 10^5)\) 个结点的无权树,试找出三个结点 \(u\)、\(v\)、\(w\),\(\operatorname{s.t.}\) $$\operatorname{card}({u,v\text{ 间的路径}}\cup{v,w\text{ 间的路径}}\cup{w,u\text{ 间的路径}})$$ 最大。输出方案。

Solution

直径以及距离直径的最远点(反证法)

两次 DFS + BFS

#include <bits/stdc++.h>
using namespace std;
const int N = 1000005;

int n;
vector<int> g[N];
int d[N], v[N];

void dfs(int p, int from)
{
    for (int q : g[p])
        if (q != from)
        {
            d[q] = d[p] + 1;
            dfs(q, p);
        }
}

vector<int> path;

bool find_path(int p, int from, int tar)
{
    path.push_back(p);
    if (tar == p)
        return true;
    for (int q : g[p])
        if (q != from)
        {
            if (find_path(q, p, tar))
                return true;
        }
    path.pop_back();
    return false;
}

signed main()
{
    ios::sync_with_stdio(false);
    cin >> n;
    for (int i = 1; i < n; i++)
    {
        int x, y;
        cin >> x >> y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(1, 0);
    int p = 0;
    for (int i = 1; i <= n; i++)
        if (d[i] > d[p])
            p = i;
    d[p] = 0;
    dfs(p, 0);
    int q = 0;
    for (int i = 1; i <= n; i++)
        if (d[i] > d[q])
            q = i;

    int len1 = d[q];

    find_path(p, 0, q);

    queue<int> que;
    for (int i = 1; i <= n; i++)
        d[q] = n + 1;
    for (int i : path)
        que.push(i), d[i] = 0, v[i] = 1;

    int last = 0;
    while (que.size())
    {
        int now = que.front();
        que.pop();
        if (now != p && now != q)
            last = now;
        for (int q : g[now])
        {
            if (d[q] > d[now] + 1)
            {
                d[q] = d[now] + 1;
                if (!v[q])
                    v[q] = 1, que.push(q);
            }
        }
    }

    int len2 = d[last];

    cout << len1 + len2 << endl;
    cout << p << " " << q << " " << last << endl;
}

标签:Paths,tar,int,Tree,Three,dfs,back,push,path
来源: https://www.cnblogs.com/mollnn/p/14397778.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有