codeforces 293E Close Vertices

题目链接

正解：点分治+树状数组。

点分治板子题，直接点分以后按照$w$排序，扫指针的时候把$w$合法的路径以$l$为下标加入树状数组统计就行了。

写这道题只是想看看我要写多久。。事实证明我确实是老年选手了，这种傻逼题写+调竟然用了$40min$。。

 #include <bits/stdc++.h>
 #define il inline
 #define RG register
 #define ll long long
 #define lb(x) (x & -x)
 #define N (300005)

 using namespace std;

 struct edge{ int nt,to,dis; }g[N];
 struct data{ int l,w; }st[N];

 int head[N],vis[N],dis[N],len[N],son[N],sz[N],c[N],l,w,n,num,top;
 ll ans;

 il int gi(){
   RG int x=,q=; RG char ch=getchar();
   while ((ch<'' || ch>'') && ch!='-') ch=getchar();
   if (ch=='-') q=-,ch=getchar();
   while (ch>='' && ch<='') x=x*+ch-'',ch=getchar();
   return q*x;
 }

 il void insert(RG int from,RG int to,RG int dis){
   g[++num]=(edge){head[from],to,dis},head[from]=num; return;
 }

 il int cmp(const data &a,const data &b){ return a.w<b.w; }

 il void add(RG int x,RG int v){
   if (!x) c[]+=v; else for (;x<=n;x+=lb(x)) c[x]+=v; return;
 }

 il int query(RG int x){
   RG int res=c[]; for (;x;x^=lb(x)) res+=c[x]; return res;
 }

 il void getrt(RG int x,RG int p,RG int &rt){
   son[x]=,sz[x]=;
   for (RG int i=head[x],v;i;i=g[i].nt){
     v=g[i].to; if (v==p || vis[v]) continue;
     getrt(v,x,rt),sz[x]+=sz[v],son[x]=max(son[x],sz[v]);
   }
   son[x]=max(son[x],son[]-sz[x]);
   if (son[rt]>=son[x]) rt=x; return;
 }

 il void getdis(RG int x,RG int p){
   st[++top]=(data){len[x],dis[x]},sz[x]=;
   for (RG int i=head[x],v;i;i=g[i].nt){
     v=g[i].to; if (v==p || vis[v]) continue;
     len[v]=len[x]+,dis[v]=dis[x]+g[i].dis;
     getdis(v,x),sz[x]+=sz[v];
   }
   return;
 }

 il void calc(RG int rt,RG int llim,RG int wlim,RG int fg){
   len[rt]=llim,dis[rt]=wlim,getdis(rt,top=);
   sort(st+,st+top+,cmp); RG int p=;
   for (RG int i=top;i;--i){
     while (p<=top && st[p].w+st[i].w<=w) add(st[p++].l,);
     if (l>=st[i].l) ans+=fg*query(l-st[i].l);
   }
   for (RG int i=;i<p;++i) add(st[i].l,-); return;
 }

 il void solve(RG int x,RG int S){
   RG int rt=; son[]=S,getrt(x,,rt),vis[rt]=,calc(rt,,,);
   for (RG int i=head[rt];i;i=g[i].nt)
     if (!vis[g[i].to]) calc(g[i].to,,g[i].dis,-);
   for (RG int i=head[rt];i;i=g[i].nt)
     if (!vis[g[i].to]) solve(g[i].to,sz[g[i].to]);
   return;
 }

 int main(){
 #ifndef ONLINE_JUDGE
   freopen("vertices.in","r",stdin);
   freopen("vertices.out","w",stdout);
 #endif
   n=gi(),l=gi(),w=gi();
   for (RG int i=,p,d;i<=n;++i)
     p=gi(),d=gi(),insert(p,i,d),insert(i,p,d);
   solve(,n),cout<<(ans-n)/; return ;
 }