マルチフィールド補正第九HDU旅行で1007リッカイン - DP +ツリーのルート変更

トピックリンク:ポイント私はああ╭(╯^╰)╮

効果の件名:

    無根ツリーが定義されました L ( a , b ) L(A、B) の木からです a A b B ルートポイント
    算出 p a i r s ( l 1 , l 2 ) ペア(L_1、L_2) L ( a , b ) = l 1 , L ( c , d ) = l 2 L(A、B)= L_1、L(C、D)= L_2
    以下とから a A b B パスの c C d D パスばらばら

問題解決のアイデア:

    計算 p a i r s ( l 1 , l 2 ) ペア(L_1、L_2) 、IFが会うことが判明しました p a i r ( x , y ) 対(X、Y)
    も満足しています p a i r s ( i , y ) i ( 1 , x ) 対(I、Y)、I \(1×)で
    単純に各のためにそれを必要と x バツ 、最大 y でき
    、その後 m x [ i ] = m a x ( m x [ i ] , m x [ i + 1 ] ) MX [I] = MAX(MX [i]は、MX [I + 1]) とすることができます O ( n ) O(N) 計算の答え

    パスが交差し、エッジがツリーを分割するために、このエッジに、列挙することができない考える
    と2つのサブツリーの直径を計算します x バツ y 、上記に対応する
    直径直接サブツリーツリーを計算することができます。 D P DP d p [ r t ] [ 0 ] DP [RT] [0] を表し r t RT 最大深さ
     d p [ r t ] [ 1 ] DP [RT] [1] 表します r t RT 最大直径は、2つのサブツリーの最大値内で選択されます d p [ s o n ] [ 0 ] DP [息子] [0] に更新


    そして、ルートを変更することを考えて、エッジを列挙する方法を検討 d p DP 父親からで r t RT への変化 s o n 息子 時間は
    最初の削除 r t RT s o n 息子 側の更新 r t RT 列挙側の目的を達成するために、この時点での情報は、
    答えのルートを変更するプロセスにすることができます直接計算され
    ます。root同様の質問に変更-木DP +変更HDUマルチ学校の第8フィールド1006 Acesrcと旅行をルート

コア:DP +ツリールートトランスデューサ(列挙側)

#include<bits/stdc++.h>
#define rint register int
#define deb(x) cerr<<#x<<" = "<<(x)<<'\n';
#define x first
#define y second
using namespace std;
typedef long long ll;
using pii = pair <int,int>;
const int maxn = 1e5 + 5;
int T, n, dp[maxn][2], mx[maxn];
vector <int> g[maxn];
map <pii, int> m[maxn][2];
map <pii, int>::iterator it;

inline void add0(int u, int v){
    m[u][0][{dp[v][0], v}]++;
    while(m[u][0].size()>3) m[u][0].erase(m[u][0].begin());
}
inline void add1(int u, int v){
    m[u][1][{dp[v][1], v}]++;
    while(m[u][1].size()>3) m[u][1].erase(m[u][1].begin());
}

inline void get(int rt){
    dp[rt][0] = dp[rt][1] = 1;
    if(m[rt][0].size()){
        it = m[rt][0].end(), --it;
        dp[rt][0] = dp[rt][1] = it->x.x + 1;
        if(m[rt][0].size() > 1){
            int tmp = it->x.x;
            --it;
            dp[rt][1] = max(dp[rt][1], tmp + it->x.x + 1);
        }
    }
    if(m[rt][1].size()){
        it = m[rt][1].end(), --it;
        dp[rt][1] = max(dp[rt][1], it->x.x);
    }
}

inline void dfs(int u, int fa){
    m[u][0].clear(), m[u][1].clear();
    for(auto v : g[u]){
        if(v == fa) continue;
        dfs(v, u);
        add0(u, v);
        add1(u, v);
    }
    get(u);
}

inline void gao(int x, int y){
    mx[x] = max(mx[x], y);
    mx[y] = max(mx[y], x);
}

inline void move(int rt, int son){
    m[rt][0].erase( {dp[son][0], son} );
    m[rt][1].erase( {dp[son][1], son} );
    get(rt);
    gao(dp[rt][1], dp[son][1]);
    add0(son, rt);
    add1(son, rt);
    get(son);
}

inline void dfs1(int rt, int fa){
    for(auto son : g[rt]){
        if(son == fa) continue;
        move(rt, son);
        dfs1(son, rt);
        move(son, rt);
    }
}

int main() {
    scanf("%d", &T);
    while(T--){
        scanf("%d", &n);
        memset(mx, 0, sizeof(mx));
        for(int i=1; i<=n; i++) g[i].clear();
        for(int i=1, u, v; i<n; i++){
            scanf("%d%d", &u, &v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        dfs(1, 0);
        dfs1(1, 0);
        ll ans = 0;
        for(int i=n; i; i--) mx[i] = max(mx[i], mx[i+1]), ans += mx[i];
        printf("%lld\n", ans);
    }
}
公開された221元の記事 ウォンの賞賛220 ・は 20000 +を見て

おすすめ

転載: blog.csdn.net/Scar_Halo/article/details/103800139