poj Cow Marathon树的直径

poj Cow Marathon树的直径_第1张图片

。。。。。。

在时间超限的边缘挣扎;

两次dfs,因为第一次找到最远的出口,但是前面还有没有搜的;

然后从这次出口进行dfs;

#pragma GCC optimize(2)

#include 

#include 

#include 

#define maxn 1000005

typedef long long ll;

using namespace std;

ll n,m,ma,fl;

ll flag[maxn];

vector < pair  > plan[maxn];

void bulid(ll a,ll b,ll c)
{
    plan[a].push_back(make_pair(b,c));
    plan[b].push_back(make_pair(a,c));
}

void dfs1(ll be,ll step)
{
    if(flag[be] == 0)
    {
        flag[be] = 1;
    }
    else
    {
        return ;
    }
    if(step > ma)
    {
        ma = step;
        fl = be;
    }
    for(int i = 0; i < plan[be].size(); i ++)
    {
        dfs1(plan[be][i].first,step+plan[be][i].second);
    }
}

void dfs2(ll be,ll step)
{
    if(flag[be] == 0)
    {
        flag[be] = 1;
    }
    else
    {
        return ;
    }
     ma = max(ma,step);
    for(int i = 0; i < plan[be].size(); i ++)
    {
        dfs2(plan[be][i].first,step+plan[be][i].second);
    }
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);

    cin >> n >> m;

    while(m--)
    {
        ll a,b,c;
        cin >> a >> b >> c;
        char z;
        cin >> z;
        bulid(a,b,c);
    }

    dfs1(1,0);
    memset(flag,0,sizeof(flag));
    ma = 0;
    dfs2(fl,0);
    cout << ma << endl;

    return 0;
}

 

你可能感兴趣的:(广搜和深搜)