bzoj 3365 [Usaco2004 Feb]Distance Statistics 路程统计（点分治，单调）

 

【题意】

 

    求树上长度不超过k的点对数目。

 

【思路】

 

    和 Tree 一样一样的。

    就是最后统计的时候别忘把根加上。

 

【代码】

 

#include<set>
#include<cmath>
#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define trav(u,i) for(int i=front[u];i;i=e[i].nxt)
#define FOR(a,b,c) for(int a=(b);a<=(c);a++)
using namespace std;

typedef long long ll;
const int N = 2e5+10; 

ll read() {
    char c=getchar();
    ll f=1,x=0;
    while(!isdigit(c)) {
        if(c=='-') f=-1; c=getchar();
    }
    while(isdigit(c))
        x=x*10+c-'0',c=getchar();
    return x*f;
}

struct Edge {
    int v,w,nxt;
}e[N<<1];
int en=1,front[N];
void adde(int u,int v,int w) 
{
    e[++en]=(Edge){v,w,front[u]}; front[u]=en;
}

int n,m,K,ans;
int rt,size,list[N],vis[N],dis[N],siz[N],f[N],l1,l2;

bool cmp(const int& x,const int& y)
{
    return dis[x]<dis[y];
}
void get_root(int u,int fa) 
{
    siz[u]=1; f[u]=0;
    trav(u,i) if(!vis[e[i].v]&&e[i].v!=fa){
        int v=e[i].v;
        get_root(v,u);
        siz[u]+=siz[v];
        if(f[u]<siz[v]) f[u]=siz[v];
    }
    f[u]=max(f[u],size-siz[u]);
    if(f[u]<f[rt]) rt=u;
}
void dfs(int u,int fa) 
{
    list[++l1]=u;
    trav(u,i) 
    if(!vis[e[i].v]&&e[i].v!=fa) {
        int v=e[i].v;
        dis[v]=dis[u]+e[i].w;
        dfs(v,u);
    }
}
int get_ans(int l,int r) 
{
    sort(list+l,list+r+1,cmp);
    int j=r,ans=0;
    for(int i=l;i<=r;i++) {
        while(j>i&&dis[list[j]]+dis[list[i]]>K) j--;
        ans+=j-i; if(i==j) break; 
    }
    return ans;
}
void solve(int u)
{
    vis[u]=1; l1=l2=0;
    trav(u,i) if(!vis[e[i].v]) {
        int v=e[i].v;
        dis[v]=e[i].w;
        dfs(v,-1);
        ans-=get_ans(l2+1,l1);
        l2=l1;
    }
    list[++l1]=u; dis[u]=0;
    ans+=get_ans(1,l1);
    trav(u,i) if(!vis[e[i].v]) {
        rt=0; get_root(e[i].v,-1);
        size=siz[e[i].v];
        solve(rt);
    }
}

int main()
{
    n=read(),m=read();
    int u,v,w; char s[5];
    FOR(i,1,m) {
        u=read(),v=read(),w=read();
        adde(u,v,w),adde(v,u,w);
        scanf("%s",s);
    }
    K=read();
    size=f[0]=n;
    get_root(1,-1);
    solve(rt);
    printf("%d\n",ans);
    return 0;
}