题目大意
给定n棵树,n<=60,每棵树有五个非负整数参数a[i],b[i],c[i],d[i],l[i],表示:第i棵树由第a[i]与第b[i]棵树组成,其中a[i]的第c[i]个点和b[i]棵树的第d[i]个点连一条长度为l[i]的边。
点的编号情况:a[i]的编号照搬过来,b[i]的编号都加上cnt[a[i]],即第a[i]棵树的点数。
编号由0到cnt[a[i]]-1
特别的,第0棵树是一个点,a[i]可以等于b[i]
现在,我们求每一颗树的美观度,定义为
∑n−1i=0∑nj=i+1dis(i,j)
dis(i,j)为i到j的距离。
所有运算在
109+7
的模意义下进行
分析
一看到这题很容易想到对一棵树进行分治(当然不是树分治-_-!),再合并。我们设第i棵树的美观度为f[i],那么f[i]由两棵构成他的树a[i],b[i]合并而来,那么f[i]首先是要加上f[a[i]],f[b[i]],然后就是两棵树之间的点的路径的美观度了。
我们先忽略新连接的边的长度l[i]。
对于a[i]中的一个点,他可以到b[i]所有点,那么记b[i]所有点集中到d[i]价值为all(b[i],d[i]),则对于整棵树来说,我们知道all(b[i],d[i])有cnt[a[i]]个。
同理all(a[i],c[i])有cnt[b[i]]个。
而新连接的边会被走cnt[a[i]]*cnt[b[i]]次。
那么得出结论,f[i]=f[a[i]]+f[a[j]]+all(b[i],d[i])*cnt[a[i]]+all(a[i],c[i])*cnt[b[i]]+l[i]*cnt[a[i]]*cnt[b[i]]
现在问题转化为如何求all了。既然我们拆开了第i棵树,我们就还可以拆第a[i],b[i]棵树嘛,按照上面的思想,继续拆下去。
这个时候又发现困难了,对于一颗树a1,分成两棵树b1,b2后,设要汇聚到的点是x1,我们暂时把他认为在b2里吧,则all(a1,x1)=all(b2,x1-cnt[b1])+all(b1,d[a1])+(dis(c[a1],x1)+l)*cnt[b1]
b1=a[a1],b2=b[a1]
这个时候不知道怎么求road了。
又是原来的思路,继续递归···
然后弄个哈希或者map记录搜过的东西就好了。
一些问题
有没有想过x1会有多少种情况呢?真的会有
2n
个点吗?
怎样才能提高哈希等判重的效率?
如何才能尽量避免程序出现低级错误如x1没减cnt?
对cnt数组取模会不会出问题?
总结错误
这道题打了我很久,有点复杂吧,也有很多问题。
一开始没有完全想好就开始打了,对c++还不太熟的我一部分精力放在语法上,导致很多地方漏打模啊,减号啊等。一方面是要对c++更熟练,也要权衡好打题前和中的时间分配。否则得不偿失。
我用了map,但是不大会,导致开错,还弄了半天才对。
有时候为了调试方便是需要一些编写技巧,这个我做的不好,调试的时候有点麻烦。比如在打的时候f[i]就一行写出。也不能在调试的时候及时了解all的返回值。
知识不牢固就是不爽···
贴个代码
#include<cstdio>
#include<iostream>
#include<map>
using namespace std;
#define ll long long
#define fo(i,j,k) for(i=j;i<=k;i++)
const int lm=1000000;
const ll mo=1000000007;
ll i,t,tt,m,a[65],b[65],c[65],d[65];
ll z,f[65],cnt[65],l[65],rcnt[65];
map<ll,ll> h[650];
map<pair<ll,ll>,ll> h1[650];
ll road(ll a2,ll x2,ll y2)
{
if (a2==0) return 0;
if (x2==y2) return 0;
if (x2>y2)
{
z=x2;
x2=y2;
y2=z;
}
ll nc1=0;
if (h1[a2][make_pair(x2,y2)]!=0) return h1[a2][make_pair(x2,y2)];
if (x2<rcnt[a[a2]]&&y2<rcnt[a[a2]])
nc1=road(a[a2],x2,y2);
else
if (x2>=rcnt[a[a2]]&&y2>=rcnt[a[a2]])
nc1=road(b[a2],x2-rcnt[a[a2]],y2-rcnt[a[a2]]);
else
if (x2<rcnt[a[a2]])
nc1=(road(a[a2],x2,c[a2])+road(b[a2],d[a2],y2-rcnt[a[a2]])+l[a2])%mo;
h1[a2][make_pair(x2,y2)]=nc1;
return nc1;
}
ll all(ll a1,ll x1)
{
if (a1==0) return 0;
ll nc=0,wat=0;
if(h[a1][x1]!=0) return h[a1][x1];
if (x1<rcnt[a[a1]])
{
nc=all(a[a1],x1);
nc=nc+all(b[a1],d[a1]);
wat=road(a[a1],c[a1],x1);
(nc=nc%mo+(l[a1]+wat)%mo*cnt[b[a1]]%mo)%=mo;
}
else
{
nc=all(b[a1],x1-rcnt[a[a1]]);
nc+=all(a[a1],c[a1]);
wat=road(b[a1],d[a1],x1-rcnt[a[a1]]);//wat就是调试用的
(nc=nc%mo+(l[a1]+wat)%mo*cnt[a[a1]]%mo)%=mo;
}
h[a1][x1]=nc;
return nc;
}
int main()
{
freopen("xmax.in","r",stdin);
freopen("xmax.out","w",stdout);
scanf("%lld",&t);
t++;
while (--t)
{
tt++;
fo(i,0,649) h[i].clear(),h1[i].clear();
scanf("%lld",&m);
fo(i,1,m)
scanf("%lld %lld %lld %lld %lld",&a[i],&b[i],&c[i],&d[i],&l[i]);
cnt[0]=1;
rcnt[0]=1;
f[0]=0;
fo(i,1,m)
{
if (i==15)
{
i=i;
}
f[i]=(f[a[i]]+f[b[i]])%mo;
f[i]=(f[i]+all(a[i],c[i])*cnt[b[i]]%mo)%mo;
f[i]=(f[i]+all(b[i],d[i])*cnt[a[i]]%mo)%mo;
f[i]=(f[i]+l[i]*cnt[a[i]]%mo*cnt[b[i]]%mo)%mo;
cnt[i]=(cnt[a[i]]+cnt[b[i]])%mo;
rcnt[i]=rcnt[a[i]]+rcnt[b[i]];
}
fo(i,1,m)
printf("%lld\n",f[i]);
}
}