bzoj3677: [Apio2014]连珠线

2021-04-23 16:25

阅读:349

数据范围满足1≤n≤200000。 

 

题解:

看到这道题,可以从两个角度入手考虑:

1)对于这棵树最终的形态,考虑哪些边可能是蓝边,并计算最大值

2)考虑这棵树建树的过程。在一开始只有根节点,后来不断加点过程中有哪些边是蓝边,并计算最大值

 

我一开始是从第一个角度考虑的

通过手动根据样例画图发现,蓝边一定是两条两条地连在一起(因为每次Insert操作都是两边一起变蓝,且这两边是相邻的)

考虑这两条蓝边(指同时出现的那两条蓝边)可能的连法:

技术分享图片

连法一如两条蓝色边所示,连法二如两条绿色边所示

一个点若是两条蓝边的中心点,那两条蓝边要么是与自己两个孩子的连边,要么是与自己一个孩子的连边加上自己与父节点的连边

然后就可以树形dp了

f[0][i]表示i号点不做两条蓝边中心点:

f[0][i]=它每一个son可得到的最大值 的和

f[1][i]表示i号点做两条蓝边中心点,且是连法一:

f[1][i]=(选两个son,每个son不向i连边可得最大值 的和)+(这两个son到点i边的长度)+(其他son可得到的最大值 的和)

f[2][i]表示i号点做两条蓝边中心点,是连法二:

f[2][i]=(选一个son,该son不向i连边可得到的最大值)+(这个son到点i边的长度)+(点i到父节点边的长度)+(其他son可得到的最大值 的和)

大体思路就是这样,看起来很对吧……

然而是有问题的!我一开始就以为这样是对的,但发现WA,洛谷上也只得了20分……

 

一组反例:

技术分享图片

按照上面的算法,最大值时蓝边应这么连

但注意两个黄框

必须先存在4、6才能出现5,必须先存在3、8才能出现2

对于7,必须2、5都存在且连上了一条边才可以出现

假设先以9-4-6-5的顺序把5及其子树都出现了,要想再出现2,需要让3、8出现

但3、8都是叶节点,唯一与2的连边还是蓝边,说明他们只能是一开始连了一条边,之后插入2

但这时5及其子树已插入完了,3、8没有与已存在的这棵树连在一起

2想出现的话只能3、8单成一棵树,与题目不符!

 

为了保证符合题意,我们考虑第二个角度

假设已确定根节点了,那么由题意,所有“两条蓝边”都是上面所说的连法二,不可能为连法一

这样同样可以树形dp,而且比上面的还简单

(转移方法与上面类似,不详细说了)

但是这是在确定一个根节点的情况下,而每个点都可能是根节点。

暴力每次树形dp的话O(n2)伤不起……

于是考虑换根。

从上到下换根。每次选当前“根节点”的一个未当过根的子节点作为新的根节点,把当前这个点作为它的子节点,O(1)维护各种信息

这样就很快了,只是细节需要多多多多多多注意一下

 

代码:

(注:为了方便计算,我多用了几个数组计算不同的东西,细节有点多)

技术分享图片技术分享图片
 1 #include 2 #include 3 #include
 4 #define INF 2100000000
 5 using namespace std;
 6 
 7 const int N = 200005;
 8 int f[4][N],ma[N][2];
 9 
10 struct node{
11     int v,len;
12     node *next;       
13 }pool[2*N],*h[N];
14 int cnt;
15 void addedge(int u,int v,int len){
16     node *p=&pool[++cnt],*q=&pool[++cnt];
17     p->v=v;p->next=h[u];h[u]=p;p->len=len;
18     q->v=u;q->next=h[v];h[v]=q;q->len=len;     
19 }
20 
21 int fa[N],dis[N];
22 int n;
23 void dfs(int u){
24     int v,num1=-INF,num2=-INF,m1=0,m2=0;
25     f[0][u]=f[1][u]=f[2][u]=f[3][u]=0;
26     for(node *p=h[u];p;p=p->next)
27         if(!fa[v=p->v]){
28             fa[v]=u;
29             dis[v]=p->len;
30             dfs(v);
31             f[0][u]+=f[2][v];
32             if(f[3][v]>num1) num2=num1,m2=m1,num1=f[3][v],m1=v;
33             else if(f[3][v]>num2) num2=f[3][v],m2=v;
34         }
35     if(m1) f[1][u]=dis[u]+num1+f[0][u];
36     ma[u][0]=m1;ma[u][1]=m2;
37     f[2][u]=max(f[1][u],f[0][u]);
38     f[3][u]=f[0][u]-f[2][u]+dis[u];
39 }
40 
41 int ans=0;
42 void dfs2(int u){
43     int v;
44     int pre[4],mm[2],ff,d;
45     for(int i=0;i4;i++) pre[i]=f[i][u];
46     d=dis[u];
47     
48     ans=max(ans,f[0][u]);
49     for(node *p=h[u];p;p=p->next)
50         if(fa[v=p->v]==u){
51             f[0][u]=f[0][u]-f[2][v];
52             dis[u]=p->len;
53             if(ma[u][0]==v)
54                 if(ma[u][1]) f[1][u]=f[0][u]+f[3][ma[u][1]]+dis[u];
55                 else f[1][u]=0;/**/
56             else if(ma[u][0]) 
57                 f[1][u]=f[0][u]+f[3][ma[u][0]]+dis[u];
58             f[2][u]=max(f[1][u],f[0][u]);
59             f[3][u]=f[0][u]-f[2][u]+dis[u];
60             mm[0]=ma[v][0];mm[1]=ma[v][1];
61             if((mm[0] && f[3][u]>f[3][mm[0]]) || !mm[0]) ma[v][1]=ma[v][0],ma[v][0]=u;
62             else if((mm[1] && f[3][u]>f[3][mm[1]]) || !mm[1]) ma[v][1]=u;
63             
64             ff=f[0][v];
65             f[0][v]+=f[2][u];
66             
67             dfs2(v);
68             
69             for(int i=0;i4;i++) f[i][u]=pre[i];
70             dis[u]=d;f[0][v]=ff;
71             ma[v][0]=mm[0];ma[v][1]=mm[1];
72         }
73 }
74 
75 int main()
76 {
77     int i,a,b,c;
78     scanf("%d",&n);
79     for(i=1;i)
80         scanf("%d%d%d",&a,&b,&c),addedge(a,b,c);
81         
82     fa[1]=-1;dis[1]=0;
83     dfs(1);
84     dfs2(1);
85     
86     printf("%d\n",ans);
87     
88     return 0;    
89 }
View Code

 


评论


亲,登录后才可以留言!