[51nod1709]复杂度分析

　　给出一棵n个点的树(以1号点为根)，定义dep[i]为点i到根路径上点的个数。众所周知，树上最近公共祖先问题可以用倍增算法解决。现在我们需要算出这个算法精确的复杂度。我们定义计算点i和点j最近公共组先的精确复杂度为bit[dep[i]-dep[lca(i,j)]]+bit[dep[j]-dep[lca(i,j)]](bit[i]表示i在二进制表示下有多少个1，lca(i,j)表示点i和点j的最近公共祖先)。为了计算平均所需的复杂度为多少，请你帮忙计算任意两点计算最近公共组先所需复杂度的总和。
即计算 sum{ bit[dep[i]-dep[lca(i,j)]]+bit[dep[j]-dep[lca(i,j)]] } ,1<=i<n,i+1<=j<=n;
　Input
　　第一行一个数n表示点数(1<=n<=100,000)
　　接下来n-1行每行两个数x,y表示一条边(1<=x,y<=n)
　Output
　　一个数表示答案

 

　　抱sxt大腿系列。。大概思路就是统计每个点往上跳每一步对答案的贡献。。先倍增预处理出每个点的那些父亲还有到父亲的那条边。

　　大概就是统计一下能从子树里跳到当前点的节点数，那些点就能当前父亲其他儿子里能跳到父亲的节点一起贡献。。。

　　当然不能每次直接跑。。就把查询都存到到父亲的边里面，最后再dfs一遍统计。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<cmath>
#include<cstdlib>
#include<bitset>
//#include<ctime>
#define ll long long
#define ull unsigned long long
#define ui unsigned int
#define d double
//#define ld long double
using namespace std;
const int maxn=100233,mxnode=maxn<<1;
struct zs{int too,pre;}e[maxn<<1];int tot,last[maxn];
int fa[maxn][18],fae[maxn][18],num[maxn],pos[maxn],TIM,sz[maxn];
ll sum[maxn],sume[maxn<<1];
int i,j,k,n,m;
ll ans;

int ra,fh;char rx;
inline int read(){
    rx=getchar(),ra=0,fh=1;
    while((rx<'0'||rx>'9')&&rx!='-')rx=getchar();
    if(rx=='-')fh=-1,rx=getchar();
    while(rx>='0'&&rx<='9')ra=ra*10+rx-48,rx=getchar();return ra*fh;
}

inline void dfs(int x){
    register int i,to;pos[++TIM]=x,sz[x]=num[x]=1;
    for(i=1;i<18;i++)fa[x][i]=fa[fa[x][i-1]][i-1],fae[x][i]=fae[fa[x][i-1]][i-1];
    for(i=last[x];i;i=e[i].pre)if((to=e[i].too)!=fa[x][0])
        fa[to][0]=x,fae[to][0]=i,dfs(to),sz[x]+=sz[to];
}
inline void DFS(int x){
    register int i,to;
    for(i=last[x];i;i=e[i].pre)if((to=e[i].too)!=fa[x][0])
        ans+=1ll*sume[i]*(sz[x]-sz[to]),DFS(to);
}
inline void insert(int a,int b){
    e[++tot].too=b,e[tot].pre=last[a],last[a]=tot,
    e[++tot].too=a,e[tot].pre=last[b],last[b]=tot;
}
int main(){
    n=read();register int i,j;
    for(i=1;i<n;i++)insert(read(),read());
    dfs(1);
    int f;
    for(j=0;j<18;j++)for(i=1;i<=n;i++)if((f=fa[k=pos[i]][j]))
        num[f]+=num[k],sum[f]+=num[k]+sum[k],sume[fae[k][j]]+=num[k]+sum[k];
    DFS(1),printf("%lld\n",ans);
}