bzoj 2599(点分治)

传送门
题解:点分不用说(我这种水平都都看得出来),至于如何最小化边数,这个方法还真没见过,将最优性转为存在性处理!对于符合路径边权和等于k的点对,如果它们之间有i条边,则就将i条边的合法路径数量++(既然要计数那就一定要在点分去重),最后从小到大枚举边数,找到一个合法路径数量大于0的就输出它即可。这里写图片描述
大佬总结地太好啦%%%zyf2000,真是点睛之笔。
再啰嗦一句,我居然A了一道IOI的题( ̄▽ ̄”)

#include<bits/stdc++.h>
using namespace std;
const int MAXN=2e5+5;
int n,k;
int head[MAXN],edge=0;
struct EDGE {
    int v,nxt,w;
}e[MAXN<<1];
bool vis[MAXN];
int dis[MAXN],dep[MAXN],siz[MAXN],mx[MAXN],sum,root,ans[MAXN],ret=-1;
struct A {
    int dis,dep;
    bool operator <(const A &x) const {
        return dis<x.dis;
    }
}a[MAXN];int tot=0;
inline int read() {
    int x=0,f=1;char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*f;
}
inline void adde(int u,int v,int w) {
    e[edge].nxt=head[u],e[edge].v=v,e[edge].w=w,head[u]=edge++;
    e[edge].nxt=head[v],e[edge].v=u,e[edge].w=w,head[v]=edge++;
}
void getroot(int p,int fa) {
    siz[p]=1,mx[p]=0;
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (vis[v]||v==fa) continue;
        getroot(v,p);
        siz[p]+=siz[v];
        mx[p]=max(mx[p],siz[v]);
    }
    mx[p]=max(mx[p],sum-siz[p]);
    if (mx[p]<mx[root]) root=p;
}
void getdis(int p,int fa) {
    siz[p]=1,a[++tot]=(A){dis[p],dep[p]};
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (vis[v]||v==fa) continue;
        dis[v]=dis[p]+e[i].w;
        dep[v]=dep[p]+1;
        getdis(v,p);
        siz[p]+=siz[v];
    }
}
void cal(int p,int ndis,int ndep,int delta) {
    dis[p]=ndis,dep[p]=ndep;
    tot=0;
    getdis(p,0);
    sort(a+1,a+tot+1);
    for (int l=1,r=tot;l<=r;l++) {//必须"<="???!!!
        while (l<r&&a[l].dis+a[r].dis>k) --r;
        int i=r;
        while (a[l].dis+a[i].dis==k) ans[a[l].dep+a[i].dep]+=delta,--i;
    }
}
void work(int pos) {
    cal(pos,0,0,1);
    vis[pos]=1;
    for (int i=head[pos];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (!vis[v]) {
            cal(v,e[i].w,1,-1);
            sum=siz[v],root=0;
            getroot(v,0);
            work(root);
        }
    }
}
int main() {
//  freopen("bzoj 2599.in","r",stdin);
    memset(ans,0,sizeof(ans));
    memset(vis,0,sizeof(vis));
    memset(head,-1,sizeof(head));
    n=read(),k=read();
    for (int i=1;i<n;++i) {
        int u=read()+1,v=read()+1,w=read();
        adde(u,v,w);
    }
    sum=n,root=0;
    mx[0]=MAXN;
    getroot(1,0);
    work(root);
    for (register int i=1;i<n;++i)
        if (ans[i]) {ret=i;break;}
    printf("%d\n",ret);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值