Poj 1741&&CF 161D 点分治入门

第一次写点分治，基本上是自己手撸的，由于昨天做了下准备工作(求树的重心)，这题就显得很傻逼了。唯一和预想不太一样的是dfs过程记录了father，这样就不用记两个vis了，感觉在代码的简洁性上还是很巧妙的。还有，poj的垃圾多组测试数据要注意初始化，WA了一发。

Poj 1741代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long ll;

struct Edge{
    int to, next, len;
}edge[20500];

int head[20500];
bool vis[20500];
int siz[20500];
int f[20500];
int cnt, rt, sum, n, k;
ll ans;
ll vs[20500];
int vsc;
ll v[20500];
int vc;

void init(){
    memset(head, -1, sizeof(head));
    memset(vis, false, sizeof(vis));
    cnt=0;
    ans=0;
}

void add(int u, int v, int w){
    edge[cnt].to=v;
    edge[cnt].len=w;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}

void getrt(int u, int fa){
    siz[u]=1;
    f[u]=0;
    for(int i=head[u];~i;i=edge[i].next){
        if(edge[i].to!=fa&&!vis[edge[i].to]){
            getrt(edge[i].to, u);
            siz[u]+=siz[edge[i].to];
            f[u]=max(f[u], siz[edge[i].to]);
        }
    }
    f[u]=max(f[u], sum-siz[u]);
    if(f[u]<f[rt])rt=u;
}

void getdeep(int u, int fa, ll len){
    vs[++vsc]=len;
    v[++vc]=len;
    for(int i=head[u];~i;i=edge[i].next){
        if(edge[i].to!=fa&&!vis[edge[i].to]){
            getdeep(edge[i].to, u, len+edge[i].len);
        }
    }
}

void solve(int u){
    vis[u]=true;
    vsc=0;
    for(int i=head[u];~i;i=edge[i].next){
        if(!vis[edge[i].to]){
            vc=0;
            getdeep(edge[i].to, u, edge[i].len);
            sort(v+1, v+1+vc);
            int l=1, r=vc;
            while(l<r){
                if(v[l]+v[r]<=k){
                    ans-=(r-l);
                    l++;
                }
                else r--;
            }
        }
    }
    sort(vs+1, vs+1+vsc);
    int l=1, r=vsc;
    while(l<r){
        if(vs[l]+vs[r]<=k){
            ans+=(r-l);
            l++;
        }
        else r--;
    }
    for(int i=1;i<=vsc;i++){
        if(vs[i]<=k)ans++;
    }
    for(int i=head[u];~i;i=edge[i].next){
        if(!vis[edge[i].to]){
            rt=0, sum=siz[edge[i].to];
            getrt(edge[i].to, 0);
            solve(rt);
        }
    }
}

int main()
{
    while(~scanf("%d%d", &n, &k)){
        if(!n&&!k)break;
        init();
        for(int i=1;i<n;i++){
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            add(u, v, w);
            add(v, u, w);
        }
        f[0]=30000;
        rt=0, sum=n;
        getrt(1, 0);
        solve(rt);
        printf("%lld\n", ans);
    }
}

又写了一个模板题CF 161D，纯手打，记了下时间，大概35min AC，这速度感觉勉强可以接受吧。

CF 161D代码：

#include <bits/stdc++.h>
using namespace std;

struct Edge{
    int to, next;
}e[105000];
int head[105000];
int cnt;

int siz[105000];
int f[105000];
bool vis[105000];
int n, k, sum, rt;
long long ans;
int v1[105000], v2[105000];
int t1, t2;

void init(){
    memset(head, -1, sizeof(head));
    cnt=0;
    f[0]=100000<<1;
    memset(vis, false, sizeof(vis));
}

void add(int u, int v){
    e[cnt].to=v;
    e[cnt].next=head[u];
    head[u]=cnt++;
    ans=0;
}

void getdeep(int u, int fa, int len){
    v1[++t1]=len;
    v2[++t2]=len;
    for(int i=head[u];~i;i=e[i].next){
        if(e[i].to!=fa&&!vis[e[i].to]){
            getdeep(e[i].to, u, len+1);
        }
    }
}

void getrt(int u, int fa){
    siz[u]=1;
    f[u]=0;
    for(int i=head[u];~i;i=e[i].next){
        if(e[i].to!=fa&&!vis[e[i].to]){
            getrt(e[i].to, u);
            siz[u]+=siz[e[i].to];
            f[u]=max(f[u], siz[e[i].to]);
        }
    }
    f[u]=max(f[u], sum-siz[u]);
    if(f[u]<f[rt])rt=u;
}

void solve(int u){
    vis[u]=true;
    t1=0;
    for(int i=head[u];~i;i=e[i].next){
        if(!vis[e[i].to]){
            t2=0;
            getdeep(e[i].to, u, 1);
            sort(v2+1, v2+1+t2);
            for(int j=1;j<=t2&&2*v2[j]<=k;j++){
                int p=upper_bound(v2+1, v2+1+t2, k-v2[j])-lower_bound(v2+1, v2+1+t2, k-v2[j]);
                if(2*v2[j]==k){
                    ans-=1LL*p*(p-1)/2;
                    break;
                }
                ans-=p;
            }
        }
    }
    sort(v1+1, v1+1+t1);
    for(int i=1;i<=t1&&2*v1[i]<=k;i++){
        int p=upper_bound(v1+1, v1+1+t1, k-v1[i])-lower_bound(v1+1, v1+1+t1, k-v1[i]);
        if(2*v1[i]==k){
            ans+=1LL*p*(p-1)/2;
            break;
        }
        ans+=p;
    }
    for(int i=1;i<=t1;i++)ans+=(v1[i]==k);
    for(int i=head[u];~i;i=e[i].next){
        if(!vis[e[i].to]){
            rt=0;sum=siz[e[i].to];
            getrt(e[i].to, 0);
            solve(rt);
        }
    }
}

int main()
{
    init();
    cin>>n>>k;
    for(int i=1;i<n;i++){
        int u, v;
        cin>>u>>v;
        add(u, v);
        add(v, u);
    }
    rt=0;sum=n;
    getrt(1, 0);
    solve(rt);
    cout<<ans<<endl;
}