标签：长链 剖分 LOJ top son int dep mx define

LOJ#3014. 「JOI 2019 Final」独特的城市（长链剖分）

显然我们画一条直径，容易发现被统计的只可能是直径某个距离较远的端点到这个点的路径上的值

用一个栈统计可以被统计的点，然后我们把这棵树长链剖分，每次在所有轻儿子中找深度最大的，去掉距离u小于这个深度的栈里的点，然后去计算u的重儿子

然后去掉距离u小于重儿子深度栈里的点，但是要再把u加进去，再遍历u的其他儿子

最后重新去掉u，计算答案，用直径两端当根都做一遍，取深度较大的那个

统计的话直接在外面开一个数组，弹出弹入的时候判断以下就好了

#include <bits/stdc++.h>
#define fi first
#define se second
#define pii pair<int,int>
#define mp make_pair
#define pb push_back
#define space putchar(' ')
#define enter putchar('\n')
#define eps 1e-10
#define MAXN 200005
#define ba 47
//#define ivorysi
using namespace std;
typedef long long int64;
typedef unsigned int u32;
typedef double db;
template<class T>
void read(T &res) {
    res = 0;T f = 1;char c = getchar();
    while(c < '0' || c > '9') {
    if(c == '-') f = -1;
    c = getchar();
    }
    while(c >= '0' && c <= '9') {
    res = res * 10 +c - '0';
    c = getchar();
    }
    res *= f;
}
template<class T>
void out(T x) {
    if(x < 0) {x = -x;putchar('-');}
    if(x >= 10) {
    out(x / 10);
    }
    putchar('0' + x % 10);
}
struct node {
    int to,next;
}E[MAXN * 2];
int N,M,dep[MAXN],mx[MAXN],son[MAXN],head[MAXN],sumE,S,T,c[MAXN];
int sta[MAXN],top,all;
int cnt[MAXN],ans[MAXN];
void add(int u,int v) {
    E[++sumE].to = v;
    E[sumE].next = head[u];
    head[u] = sumE;
}
void dfs(int u,int fa) {
    son[u] = 0;
    mx[u] = dep[u];
    for(int i = head[u] ; i ; i = E[i].next) {
    int v = E[i].to;
    if(v != fa) {
        dep[v] = dep[u] + 1;
        dfs(v,u);
        if(mx[v] > mx[son[u]]) son[u] = v;
        mx[u] = max(mx[u],mx[v]);
    }
    }
}
void ins(int x) {
    sta[++top] = x;
    x = c[x];
    if(cnt[x] == 0) ++all;
    ++cnt[x];
}
void del() {
    int x = sta[top--];
    x = c[x];
    if(cnt[x] == 1) --all;
    --cnt[x];
}
void calc(int u,int fa) {
    if(!son[u]) {
    ans[u] = max(ans[u],all);return;
    }
    int ol = 0;
    for(int i = head[u] ; i ; i = E[i].next) {
    int v = E[i].to;
    if(v != fa && v != son[u]) {
        ol = max(ol,mx[v] - dep[u]);
    }
    }
    while(top && dep[sta[top]] + ol >= dep[u]) del();
    ins(u);calc(son[u],u);
    while(top && dep[sta[top]] + mx[son[u]] - dep[u] >= dep[u]) del();
    for(int i = head[u] ; i ; i = E[i].next) {
    int v = E[i].to;
    if(v != fa && v != son[u]) {
        if(sta[top] != u) ins(u);
        calc(v,u);
    }
    }
    while(top && dep[sta[top]] + mx[son[u]] - dep[u] >= dep[u]) del();
    ans[u] = max(ans[u],all);return;
}
void Solve() {
    read(N);read(M);
    int a,b;
    for(int i = 1 ; i < N ; ++i) {
    read(a);read(b);
    add(a,b);add(b,a);
    }
    for(int i = 1 ; i <= N ; ++i) read(c[i]);
    dep[1] = 0;dfs(1,0);
    S = 1;
    for(int i = 2 ; i <= N ; ++i) {
    if(dep[i] > dep[S]) S = i;
    }
    dep[S] = 0;dfs(S,0);
    T = 1;
    for(int i = 2 ; i <= N ; ++i) {
    if(dep[i] > dep[T]) T = i;
    }
    dep[S] = 0;dfs(S,0);top = 0;
    calc(S,0);
    memset(cnt,0,sizeof(cnt));all = 0;
    dep[T] = 0;dfs(T,0);top = 0;
    calc(T,0);
    for(int i = 1 ; i <= N ; ++i) {
    out(ans[i]);enter;
    }
}
int main() {
#ifdef ivorysi
    freopen("f1.in","r",stdin);
#endif
    Solve();
}

标签：长链,剖分,LOJ,top,son,int,dep,mx,define
来源： https://www.cnblogs.com/ivorysi/p/10943426.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。