深度优先搜索dfs（poj 1655）

最新推荐文章于 2020-10-12 10:19:35 发布

原创最新推荐文章于 2020-10-12 10:19:35 发布 · 595 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#dfs

××××××算法学习×××××× 同时被 3 个专栏收录

50 篇文章

订阅专栏

搜索-------dfs深度优先

3 篇文章

订阅专栏

------搜索------

2 篇文章

订阅专栏

本文介绍了一道名为POJ1655的问题，该问题要求在给定的树状图中移除一个节点以使形成的子树中最大的节点数尽可能小，并输出这个最小值及对应的节点编号。通过构建树形结构并使用深度优先搜索算法来解决此问题。

题目：poj 1655

题意：给定一个树状图，要求去掉一个节点，得到几个子树，并且规定这些子树中拥有的最大节点的子树的节点数为去掉该节点的平衡值，求该平衡值最小为多少，并且求出此时去掉的是哪个点，当平衡值相当时要求去掉的点的序号最小

题解：建树，再进行一次深度优先搜索，找出该点的各个子树中有多少节点，取最大值，当然题目中是无向连接，所以要考虑双向，要用一个数组表示该点事前有没考虑

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>

using namespace std;
int minn;
vector<int> son[20050];
int vis[20050];
int go[20050]; //记录是否访问过该点
int dfs(int root,int all)
{
    go[root]=1;//记录该点已经访问过
    int maxx=0;
    int sum=1;
    for(int i=0;i<son[root].size();i++)
    {
        int id=son[root][i];
        if(go[id]) continue;
        int d=dfs(id,all);
        maxx=max(maxx,d);
        sum+=d;
    }
    son[root].clear();
    maxx=max(maxx,all-sum);
    if(maxx<vis[minn] || maxx==vis[minn] && minn>root){vis[root]=maxx;minn=root;}
    return sum;
}
int main()
{
    int t,n;
    scanf("%d",&t);
    int f,s;
    while(t--)
    {
        minn=1;
        memset(vis,0,sizeof(vis));
        memset(go,0,sizeof(go));
        scanf("%d",&n);
        for(int i=1;i<n;i++)
        {
            scanf("%d %d",&f,&s);
            son[f].push_back(s);
            son[s].push_back(f);
            vis[i]=259999;
        }
        vis[n]=259999;
        dfs(1,n);
        printf("%d %d\n",minn,vis[minn]);
    }
    return 0;
}