牛客OI周赛15-提高组 A 环球旅行（树形dp）

牛客OI周赛15-提高组 A 环球旅行

树形dp+树的直径

题目大意:删除一条边后剩下的子树中树的直径最大值最小.求此最小值
现场赛差点搞出来 ，呜呜呜
思路：可证删除的这条边肯定在原树的直径上，两遍dfs确定树的直径的起点和终点，这样问题转化为删除一条边剩下的求俩个子树中直径的，可以两遍dfs分别从st，en开始
ans[u][0]表示从st开始向下遍历后 如果u为断点的直径的最大值
ans[u][1]表示从en开始像下遍历后，如果u为断点的直径的最大值
考虑转移 ans[u]包含所有以u为根节点的子树中直径的最大值和含u节点本身产生的最大值 树形dp[][]很好解决
最后处理答案，遍历之前的直径枚举u,fa[u]，答案就是 res=min(res,max(ans[u][0],ans[fa][1]));

#pragma GCC optimize(3,"Ofast","inline")  	//G++
#include<bits/stdc++.h>
#define mem(a,x) memset(a,x,sizeof(a))
#define debug(x) cout << #x << ": " << x << endl;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define fcout cout<<setprecision(4)<<fixed
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;

const int inf=0x3f3f3f3f;
const int mod=1e9+7;
const int maxn = 1e6+5;

template<typename T> void read(T &x){
    x = 0;char ch = getchar();ll f = 1;
    while(!isdigit(ch)){if(ch == '-')f*=-1;ch=getchar();}
    while(isdigit(ch)){x = x*10+ch-48;ch=getchar();}x*=f;
}
template<typename T, typename... Args> void read(T &first, Args& ... args) {
    read(first);
    read(args...);
}
vector<pii>edge[maxn];
int dp[maxn][2],ans[maxn][2];
int st,en,f[maxn],sum,dis[maxn];
void add(int x,int y,int val){
    edge[x].push_back({y,val});
    edge[y].push_back({x,val});
}
void dfs_root(int u,int fa){
//    cout<<u;
    f[u]=fa;
    for(auto it:edge[u]){
        int v=it.first,val=it.second;
        if(v==fa) continue;
        dis[v]=dis[u]+val;
        dfs_root(v,u);
    }
    if(dis[u]>sum) sum=dis[u],en=u;
}
void dfs(int u,int fa,int _){

    dp[u][0]=dp[u][1]=0;
    for(auto it:edge[u]){
        int v=it.first,val=it.second;
        if(v==fa) continue;
        dfs(v,u,_);
        if(dp[v][0]+val>dp[u][0]) dp[u][1]=dp[u][0],dp[u][0]=dp[v][0]+val;
        else if(dp[v][0]+val>dp[u][1]) dp[u][1]=dp[v][0]+val;
        ans[u][_]=max(ans[u][_],ans[v][_]);
    }
    ans[u][_]=dp[u][0]+dp[u][1];
}
void get_ans(){
    int fa=f[en],u=en,res=inf;
    while(fa){
        res=min(res,max(ans[u][0],ans[fa][1]));
        u=f[u],fa=f[u];
    }
    cout<<res<<"\n";
}
int main()
{
    int n;
    read(n);
    for(int i=1,x,y,z;i<=n-1;i++){
        read(x,y,z);
        add(x,y,z);
    }
    dfs_root(1,0);
    st=en,sum=0,dis[st]=0;
    dfs_root(st,0);
    //debug(st) debug(en)
    dfs(st,0,0);
    dfs(en,0,1);
//    for(int i=1;i<=n;i++){
//        cout<<ans[i][0]<<" "<<ans[i][1]<<"\n";
//    }
    get_ans();
}

当时dp转移没考虑周全，呜呜呜