TopCoder SRM 690 Div1 500 TreeWalker

一题磨了一晚上，其实根本不难，就是我太菜了。

我们稍加考虑，就会发现题目求的实际上是$2\times \sum 2^{n-1-dis(i,j)},(i<j)$，稍加变形，就是$2^n\times \sum 2^{-dis(i,j)},(i<j)$，假设$2$的逆元为$inv$，就是$2^n\times \sum in{v}^{dis(i,j)},(i<j)$，然后这个就可以一个很基础的树形DP求了。

#include <bits/stdc++.h>
#define ll long long
#define fr(i,x,y) for(int i=x;i<=y;i++)
#define rf(i,x,y) for(int i=x;i>=y;i--)
using namespace std;
const int N=100001;
const int M=N<<1;
const int p=1e9+7;
const int inv2=(p+1)/2;
int cnt,head[N],Next[M],v[M];
int n;
ll ans;
ll f[N];
ll po[N],A[N];

template<class T> void checkmin(T &a,const T &b) { if (b<a) a=b; } 
template<class T> void checkmax(T &a,const T &b) { if (b>a) a=b; }

class TreeWalker {
public:
    int calc( int N, int A0, int B, int C, int MOD ) ;
};

void add(int x,int y){
	Next[++cnt]=head[x];
	head[x]=cnt;
	v[cnt]=y;
}

void Add(int &x,int y){
	x+=y;
	while(x>=p) x-=p;
}

void Add(ll &x,ll y){
	x+=y;
	while(x>=p) x-=p;
}

void dfs(int x,int fa){
	for(int i=head[x];i;i=Next[i])
	 if (v[i]!=fa){
	 	dfs(v[i],x);
	 	Add(ans,(f[x]+1)*(f[v[i]]+1)%p*inv2%p);
	 	Add(f[x],f[v[i]]*inv2%p+inv2);
	 }
}

int TreeWalker::calc(int N, int A0, int B, int C, int MOD) {
    n=N;
    A[0]=A0;
    fr(i,1,N-2) A[i]=(B*A[i-1]+C)%MOD;
    fr(i,1,N-1){
    	int j=A[i-1]%i;
    	add(i+1,j+1);add(j+1,i+1);
    	//cout<<i+1<<' '<<j+1<<endl;
    }
    po[0]=1;
    fr(i,1,n) po[i]=po[i-1]*2%p;
    
    dfs(1,0);
    return ans*2*po[n-1]%p;
}