点分治学习笔记（2） tree

今天做点分治的题，调了1个多小时，发现之前对点分治的理解是有一些偏差的。
dis【v】求出来的是那个点到根的距离。

solve不断分治。cal计算其中经过根节点的所有路径，我们在cal的时候把根节点也统计进去的话，枚举的时候就算上了到根节点的路径。不统计根节点的话，就没有计算到根节点的路径。所以要单独处理。
还有一点是，多组数据我又没有初始化边数组！小ly你能不能长点记性啊，你忘了你NOIP暴力怎么写炸的了？多组数据的题，加进去的边要初始化，head数组和cnt也要初始化。能不能记住啊。。你NOIP犯这么蠢的错误已经留下一次遗憾了，难道你还要把错误继续犯到省选？下次再错出去跑十圈。。

还有就是，对于u节点我们统计了所有他的答案。但这里面有来自一个子树的点贡献的答案。这部分答案，我们再减去就好了。减去的时候直接减就可以，因为整体答案本来是重复的，你减去的部分也是重复的，减完就不重复了（这段估计只有我能看懂）。。

附调了一个多小时的题目：
Description
Give a tree with n vertices,each edge has a length(positive integer less than 1001).
Define dist(u,v)=The min distance between node u and v.
Give an integer k,for every pair (u,v) of vertices is called valid if and only if dist(u,v) not exceed k.
Write a program that will count how many pairs which are valid for a given tree.
Input
The input contains several test cases. The first line of each test case contains two integers n, k. (n<=10000) The following n-1 lines each contains three integers u,v,l, which means there is an edge between node u and v of length l.
The last test case is followed by two zeros.
Output
For each test case output the answer on a single line.

#include<iostream>
#include<cstring>
#include<time.h>
#include<cmath>
#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;

const int MAXN=1e5+5;

struct edge{
    int to,next,w;
}e[MAXN<<1];

int head[MAXN],cnt=0,anss=0;
inline void add(int u,int v,int w){e[++cnt]=(edge){v,head[u],w},head[u]=cnt;}

vector<int>d;
int n,m,k;
int size[MAXN],f[MAXN],dis[MAXN],rt=0,sum;
bool vis[MAXN];
void dfs1(int u,int fa){
    size[u]=1;
    f[u]=0;
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to,w=e[i].w;
        if(vis[v]||v==fa)continue;
        dfs1(v,u);
        size[u]+=size[v];
        f[u]=max(f[u],size[v]);
    }
    f[u]=max(f[u],sum-size[u]);
    if(f[u]<f[rt])rt=u;
}

int dfs2(int u,int fa){
    int ans=0;
    size[u]=1;
//  cout<<dis[u]<<endl;
    d.push_back(dis[u]);
//  if(dis[u]<=k)ans++;
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to,w=e[i].w;
        if(v==fa||vis[v])continue;
        dis[v]=dis[u]+w;
        ans+=dfs2(v,u);
        size[u]+=size[v];
    }
    return ans;
}

int cal(int u,int num){
    int ans=0;
    d.clear();
    dis[u]=num;
    d.push_back(dis[u]);
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to,w=e[i].w;
        if(vis[v])continue;
        dis[v]=dis[u]+w;
        ans+=dfs2(v,u);
    }
//  cout<<d.size()<<" ";    
    int l=0,r=d.size()-1;
    sort(d.begin(),d.end());
    while(l<r){
        while(l<r&&d[l]+d[r]>k)r--;
        ans+=r-l;
        l++;
    }
//  cout<<ans<<endl;
    return ans;
}

void solve(int u){
//  cout<<u<<"ok"<<endl;
    anss+=cal(u,0);
    vis[u]=1;
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to,w=e[i].w;
        if(vis[v])continue;
        anss-=cal(v,w);
        f[0]=sum=size[v];
        dfs1(v,rt=0);
        solve(rt);
    }
}

int main(){
    while(scanf("%d%d",&n,&k)){
        memset(e,0,sizeof(e));
        memset(head,0,sizeof(head));//小ly你能不能长点记性啊。。 
        cnt=0;
        if(!n&&!k) return 0;
        memset(vis,0,sizeof(vis));
        d.clear();
        for(int i=1;i<n;i++){
            int tem1,tem2,tem3;
            scanf("%d%d%d",&tem1,&tem2,&tem3);
            add(tem1,tem2,tem3);add(tem2,tem1,tem3);
        }   
        anss=0;
        f[0]=sum=n;
        dfs1(1,rt=0);  
        solve(rt);
        printf("%d\n",anss);
    }
    return 0;
}