다중 필드 보정 아홉 번째 HDU 여행과 함께 1007 Rikka - DP + 트리 루트 변경

주제 링크 : 포인트 I 아 ╭ (╯ ^ ╰) ╮

효과에 따라 :

    루팅 해제 트리 정의 $랩)$ 트리에서입니다 $ㅏ$에 $비$ 경로 점
    계산 $쌍 (l_1, l_2)$ , $L (a, b) = l_1, L (c, d) = l_2$
    그리고에서 $ㅏ$에 $비$ 경로 $씨$ 에 $디$ 경로 이산

문제 해결 아이디어 :

    계산 $쌍 (l_1, l_2)$ , 만약이 대회에 발견 $쌍 (X, Y)$
    도 만족 $쌍 (I, Y)에서 I \ (1, X)$
    단순히 각 것을 요구한다 $엑스$ , 최대 $과$ 수
    후 $MX [I] = 최대 (MX [I], MX [I + 1])$ 일 수있다 $O (N)$ 계산 않음

    경로가 교차, 가장자리가 나무를 분할이 가장자리, 열거 할 수없는 고려
    와 두 개의 서브 트리의 직경을 계산 $엑스$ 및 $과$ 는 상기에 상응하는
    직경하여 직접 계산 될 수 트리의 서브 트리 $DP$ , $DP [RT] [0]$ 나타낸다 $RT$ 최대 깊이
     $DP [RT] [1]$ 나타낸다 $RT$ 최대 직경은 두개의 서브 트리 내의 최대 선택 $DP [아들] [0]$ 을 업데이트

---

    그런 다음 루트를 변경하는 생각 가장자리를 열거하는 방법을 고려 $DP$ 아버지로부터의 $RT$ 로 변경 $아들$ 시간이
    먼저 삭제 $RT$ 에 $아들$ 측 업데이트 $RT$ 열거 측의 목적을 달성하기 위해 현재 정보가
    할 수있는 대답의 루트를 변경하는 과정에서 직접 계산
    루트 비슷한 질문을 변경HDU 멀티 학교 여덟 번째 필드를 1006 Acesrc 및 여행 - 나무 DP + 변경 뿌리

코어 : 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);
    }
}