CF516DDrazil and Morning Exercise

Posted asuldb

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了CF516DDrazil and Morning Exercise相关的知识,希望对你有一定的参考价值。

题目

首先我们知道,在树上距离一个点最远的点一定是直径的两个端点之一

首先两遍( m dfs)把直径求出来,定义(d(u))表示点(u)距离其最远点的距离,有了直径我们就能求出(d)数组了

当然可以树形( m dp),设(f_{x,i,j})表示在(x)子树内部选择一个最大值为(i)最小值为(j)的最大联通块是多少,显然这样的复杂度奇高无比

考虑把求出的直径来好好利用一下

首先感性的发现一下,设(x)为直径中点,那么(d(x))肯定是最小的点权,又发现我们钦定(x)为根,那么一个点的点权肯定比其子树内部的点都要小

那么我们就钦定(x)为根,之后枚举一个联通块的最高点(i),也就是点权的最小值

由于深度越大点权越大,于是我们就把这个子树内部所有点权不超过(d(i)+L)的点都选上,这显然构成一个联通块

于是问题转化成了求一个子树内部有多少个点的点权小于某个给定值,不难发现这是主席树裸题

说的好,于是我选择( m dsu on tree),复杂度(O(nqlog n+nlog^2n)),莫名其妙跑得比并查集老哥们快

代码

#include<bits/stdc++.h>
#define re register
#define LL long long
#define lb(x) ((x)&(-x))
#define max(a,b) ((a)>(b)?(a):(b))
inline int read() {
    char c=getchar();int x=0;while(c<'0'||c>'9') c=getchar();
    while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x;
}
const int maxn=1e5+5;
struct E{int v,nxt,w;}e[maxn<<1];
int n,m,num,Rt,rt,sz,S;
int head[maxn],sum[maxn],son[maxn],pos[maxn],dep[maxn],d[maxn],Ans[55];
LL pre[maxn],mx[maxn],q[55],c[maxn];
inline void add(int x,int y,int w) {
    e[++num].v=y;e[num].nxt=head[x];head[x]=num;e[num].w=w;
}
inline int find(LL x) {
    int l=1,r=sz,now=0;
    while(l<=r) {
        int mid=l+r>>1;
        if(c[mid]<=x) now=mid,l=mid+1;else r=mid-1;
    }
    return now;
}
void dfs(int x,int fa) {
    for(re int i=head[x];i;i=e[i].nxt) {
        if(e[i].v==fa) continue;
        pre[e[i].v]=pre[x]+e[i].w;
        dfs(e[i].v,x);
    }
}
void dfs1(int x) {
    sum[x]=1;
    for(re int i=head[x];i;i=e[i].nxt) {
        if(dep[e[i].v]) continue;
        dep[e[i].v]=dep[x]+1;dfs1(e[i].v);sum[x]+=sum[e[i].v];
        if(sum[e[i].v]>sum[son[x]]) son[x]=e[i].v;
    }
}
inline void add(int x,int v) {
    for(re int i=x;i<=sz;i+=lb(i)) d[i]+=v;
}
inline int ask(int x) {
    int now=0;
    for(re int i=x;i;i-=lb(i)) now+=d[i];
    return now;
}
void calc(int x,int o) {
    add(pos[x],o);
    for(re int i=head[x];i;i=e[i].nxt)
    if(dep[e[i].v]>dep[x]&&S!=e[i].v) calc(e[i].v,o);
}
void dsu(int x,int k) {
    for(re int i=head[x];i;i=e[i].nxt) 
    if(dep[e[i].v]>dep[x]&&son[x]!=e[i].v) dsu(e[i].v,0);
    if(son[x]) dsu(son[x],1);
    S=son[x],calc(x,1),S=0;
    for(re int i=1;i<=m;i++) {
        int g=ask(find(mx[x]+q[i]));
        Ans[i]=max(Ans[i],g);
    }
    if(!k) calc(x,-1);
}
int main() {
    n=read();
    for(re int x,y,w,i=1;i<n;++i)
        x=read(),y=read(),w=read(),add(x,y,w),add(y,x,w);
    m=read();for(re int i=1;i<=m;i++) scanf("%lld",&q[i]);
    dfs(1,0);rt=1;
    for(re int i=2;i<=n;i++) if(pre[i]>pre[rt]) rt=i;
    pre[rt]=0;dfs(rt,0);Rt=1;
    for(re int i=1;i<=n;i++) mx[i]=pre[i];
    for(re int i=2;i<=n;i++) if(pre[i]>pre[Rt]) Rt=i;
    pre[Rt]=0;dfs(Rt,0);
    for(re int i=1;i<=n;i++) mx[i]=max(mx[i],pre[i]);
    rt=1;for(re int i=2;i<=n;i++) if(mx[i]<mx[rt]) rt=i;
    dep[rt]=1;dfs1(rt);
    for(re int i=1;i<=n;i++) c[i]=mx[i];
    std::sort(c+1,c+n+1);sz=std::unique(c+1,c+n+1)-c-1;
    for(re int i=1;i<=n;i++) pos[i]=find(mx[i]);
    dsu(rt,1);for(re int i=1;i<=m;i++) printf("%d
",Ans[i]);
    return 0;
}

以上是关于CF516DDrazil and Morning Exercise的主要内容,如果未能解决你的问题,请参考以下文章

CF516D Drazil and Morning Exercise

Codeforces 516D Drazil and Morning Exercise (栈二分)

CF516D

cf 1063D Round 516 Candies for Children

[Codeforces 816A]Karen and Morning

CodeForces - 816A Karen and Morning 解题