test20181024 zi

题意

分析

这种题一般是推公式，发现必须求得的量，然后定义函数记忆化。

然后那些函数里面又是递归处理，合并。

代码

为了不爆空间，用map存记忆化内容。

#include<bits/stdc++.h>

using namespace std;

#define gc c=getchar()
#define r(x) read(x)
#define ll long long

template<typename T>
inline void read(T&x)
{
	x=0;
	T k=1;
	char gc;
	while(!isdigit(c))
	{
		if(c=='-')k=-1;
		gc;
	}
	while(isdigit(c))
	{
		x=x*10+c-'0';
		gc;
	}
	x*=k;
}

const int p=1e9+7;
const int N=65;

struct Tree
{
	ll a,b,c,d,l,siz,ans;
} A[N];

map<pair<pair<int,ll>,ll>,ll>mp;

ll dfs2(int rt,ll u,ll v)
{
	if(!rt)
		return 0;
	pair<pair<int,int>,int> Hash=make_pair(make_pair(rt,u),v);
	if(mp[Hash])
		return mp[Hash];
	Tree &t=A[rt];
	ll &ans=mp[Hash];
	if(u==v)
		return 0;
	if(u<A[t.a].siz&&v<A[t.a].siz)
		return ans=dfs2(t.a,u,v);
	if(u<A[t.a].siz&&v>=A[t.a].siz)
		return ans=(dfs2(t.a,t.c,u)+dfs2(t.b,t.d,v-A[t.a].siz)+t.l)%p;
	if(u>=A[t.a].siz&&v<A[t.a].siz)
		return ans=(dfs2(t.a,t.c,v)+dfs2(t.b,t.d,u-A[t.a].siz)+t.l)%p;
	return ans=dfs2(t.b,u-A[t.a].siz,v-A[t.a].siz);
}

map<pair<int,ll>,ll>mpp;
ll dfs1(int rt,ll x) //T_rt 以 x 为根,所以点的(子树大小*边权)之和
{
	if(!rt)
		return 0;
	Tree &t=A[rt];
	pair<int,ll> Hash=make_pair(rt,x);
	if(mpp[Hash])
		return mpp[Hash];
	ll &ans=mpp[Hash];
	if(x<A[t.a].siz)
		return ans=(dfs1(t.b,t.d)+dfs1(t.a,x)+A[t.b].siz%p*(dfs2(t.a,t.c,x)+t.l)%p)%p;
	else
		return ans=(dfs1(t.a,t.c)+dfs1(t.b,x-A[t.a].siz)+A[t.a].siz%p*(dfs2(t.b,t.d,x-A[t.a].siz)+t.l)%p)%p;
}

int main()
{
	freopen("zi.in","r",stdin);
	freopen("zi.out","w",stdout);
	int m;
	r(m);
	A[0].siz=1;
	for(int i=1; i<=m; ++i)
	{
		Tree &x=A[i];
		r(x.a),r(x.b),r(x.c),r(x.d),r(x.l);
		x.siz=A[x.a].siz+A[x.b].siz;
		x.ans=(A[x.a].ans+A[x.b].ans
		       +(ll)(A[x.a].siz%p)*(A[x.b].siz%p)%p*x.l%p
		       +A[x.b].siz%p*dfs1(x.a,x.c)%p
		       +A[x.a].siz%p*dfs1(x.b,x.d)%p)%p;
		printf("%lld\n",x.ans);
	}
}