码迷,mamicode.com
首页 > 其他好文 > 详细

[dp][递归] Jzoj P4211 送你一棵圣诞树

时间:2019-01-24 22:53:48      阅读:206      评论:0      收藏:0      [点我收藏+]

标签:tpcc   end   osi   分享图片   选择   cstring   col   border   nta   

Description

再过三个多月就是圣诞节了,小R 想送小Y 一棵圣诞树作为节日礼物。因为他想让这棵圣诞树越大越好,所以当然是买不到能够让他满意的树的,因此他打算自己把这棵树拼出来。
现在,小R 开始画这棵树的设计图纸了。因为这棵树实在太大,所以他采用了一种比较方便的方法。首先他定义了m+ 1 棵树T0 到Tm。最开始他只画好了T0 的图纸:就只有一个点,编号为0。
接着,对于每一棵树Ti,他在第Tai 棵树的第ci 个点和第Tbi 棵树的第di 个点之间连上了一条长度为li 的边。在Ti 中,他保持Tai 中的所有节点编号不变,然后如果Tai 中有s 个节点,他会把Tbi 中的所有节点的编号加上s。
终于,他画好了所有的树。现在他定义一颗大小为n 的树的美观度为技术分享图片,其中d(i; j) 为这棵树中i 到j 的最短距离。
为了方便小R 选择等究竟拼哪一棵树,你可以分别告诉他T1 到Tm 的美观度吗?答案可能很大,请对10^9 + 7 取模后输出。
 

Input

第一行输入一个正整数T 表示数据组数。每组数据的第一行是一个整数m,接下来m 行每行五个整数ai, bi, ci, di, li,保证0 <= ai, bi < i, 0<= li<= 10^9,ci, di 存在。

Output

对于每组询问输出m 行。第i 行输出Ti 的权值
 

Sample Input

1
2
0 0 0 0 2
1 1 0 0 4

Sample Output

2
28
 

Data Constraint

对于30% 的数据,m <= 8
对于60% 的数据,m <= 16
对于100% 的数据,1 <= m<= 60,T<= 100

 

题解

  • 题目大意:给定了n棵树,每次会奖两棵树中的x,y之间连一条len的边,也就是将两棵树合并,问每棵树上两两点对的最短路径和
  • 设calc1(x,y)为x这棵树里,其他点到y的最短路径和,calc2(x,y,k)为x这棵树里,y到k的最短路径长度
  • 设f[i]为第i棵数的答案,f[i]=f[i.left]+f[i.right]+calc1(i.left,i.c)*i.right.size+calc1(i.right,i.d)*i.left.size+i.len*i.left.size*i.right.size
  • calc1(x,y)中可以分成两种情况,①在x的左子树②在x的右子树
  • 当在x的左子树时calc1(x,y)=calc1(x.left,y)+(calc2(y,x.left,x.c)+x.len)*x.right.size+calc1(x.right,x.d),但在右子树里也是同理
  • calc2(x,y,z)也可以想上面分成两种情况,①y和k在同一棵子树里②y和k不在同一棵子树里
  • 当在它们在同一颗子树里,直接往下递归;当它们不在同一棵子树里时,calc2(x,y,z)=calc2(x.left,y,x.c)+calc2(x.right,z,x.d)+x.len
  • 这题比较恶心,还要用map优化

代码

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<cstring>
 4 #include<map>
 5 #define ll long long
 6 using namespace std;
 7 const ll N=70,mo=1e9+7;
 8 ll T,n,f[N];
 9 typedef pair<ll,ll> node;
10 struct edge{ ll a,b,size,c,d,v; }p[N];
11 map <node,ll> Q[N];
12 ll calc2(ll k,ll x,ll y)
13 {
14     if (x>y) swap(x,y);
15     if (k==0||x==y) return 0;
16     if (x<p[p[k].a].size&&y<p[p[k].a].size) return calc2(p[k].a,x,y);
17     else if (x>=p[p[k].a].size&&y>=p[p[k].a].size) return calc2(p[k].b,x-p[p[k].a].size,y-p[p[k].a].size);
18     else 
19     {
20         node q=make_pair(x,y);
21         if (Q[k].find(q)!=Q[k].end()) return Q[k][q];
22         return Q[k][q]=(calc2(p[k].a,p[k].c,x)+calc2(p[k].b,p[k].d,y-p[p[k].a].size)+p[k].v)%mo;
23     }
24 }
25 ll calc1(ll k,ll x)
26 {
27     ll r=0; node q=make_pair(x,0);
28     if (Q[k].find(q)!=Q[k].end()) return Q[k][q];
29     if (k==0) return 0;
30     if (x<p[p[k].a].size) r=(calc1(p[k].a,x)+calc1(p[k].b,p[k].d))%mo+(((p[k].v+calc2(p[k].a,p[k].c,x))%mo)*(p[p[k].b].size%mo))%mo;
31     else r=(calc1(p[k].a,p[k].c)+calc1(p[k].b,x-p[p[k].a].size))%mo+(((p[k].v+calc2(p[k].b,p[k].d,x-p[p[k].a].size))%mo)*(p[p[k].a].size%mo)%mo);
32     return (Q[k][q]=r)%mo;
33 }
34 int main()
35 {
36     freopen("data.in","r",stdin);
37     scanf("%d",&T);
38     while (T--)
39     {
40         scanf("%d",&n),memset(p,0,sizeof(p)),p[0].size=1;
41         for (int i=1;i<=n;i++) scanf("%lld%lld%lld%lld%lld",&p[i].a,&p[i].b,&p[i].c,&p[i].d,&p[i].v),p[i].size=p[p[i].a].size+p[p[i].b].size,Q[i].clear();
42         for (int i=1;i<=n;i++)
43             f[i]=((f[p[i].a]+f[p[i].b])%mo+(calc1(p[i].a,p[i].c)*(p[p[i].b].size%mo)+(calc1(p[i].b,p[i].d)*(p[p[i].a].size%mo))%mo
44                 +(((p[i].v%mo)*(p[p[i].a].size%mo))%mo*(p[p[i].b].size%mo))%mo)%mo)%mo,
45             printf("%lld\n",f[i]);
46     }
47 }

 

[dp][递归] Jzoj P4211 送你一棵圣诞树

标签:tpcc   end   osi   分享图片   选择   cstring   col   border   nta   

原文地址:https://www.cnblogs.com/Comfortable/p/10317133.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!