[bzoj3611][Heoi2014]大工程

Posted 2020-06-13

　　看题目感觉应该就是传说中的虚树？

　　然后跑去学了一发。。。自己YY了一下然后挂飞。。于是就只好抄模板了T_T

　　建完虚树就是个树形dp。。。

　　对于询问总和：每条边对答案的贡献是边权*一端的节点数*另一端的节点数。（这里的节点不包括建虚树时添上去的点）

　　对于询问最小值最大值，每次计算出经过这个节点的最长||最短路径长度就好了。。

　　大概这种题条件都有一个sigma(K)<=n之类的。。而且题目求的东西得符合区间加法。。。不然你把边合在一起也没用>_<

　　链剖求lca果然快。。。速度能进前10.。。然而代码长度实在感人= =

  1 #include<cstdio>
  2 #include<iostream>
  3 #include<cstring>
  4 #include<algorithm>
  5 #define ll long long
  6 using namespace std;
  7 const int maxn=1000023;
  8 const int inf=1002333333;
  9 struct zs{
 10     int too,pre;
 11 }e[maxn<<1];
 12 struct zs1{
 13     int too,pre,dis;
 14 }e1[maxn<<2];
 15 int tot,tot1,last[maxn],last1[maxn];
 16 int sz[maxn],mn[maxn],mx[maxn];
 17 int intree[maxn],poi[maxn],rt;
 18 int dfn[maxn],bel[maxn],size[maxn],dep[maxn],fa[maxn],tim;
 19 int st[maxn],top;
 20 int ansmn,ansmx;
 21 ll anssum;
 22 int i,j,k,n,m,K,a,b,lca;
 23  
 24  
 25 int ra;char rx;
 26 inline int read(){
 27     rx=getchar(),ra=0;
 28     while(rx<‘0‘||rx>‘9‘)rx=getchar();
 29     while(rx>=‘0‘&&rx<=‘9‘)ra*=10,ra+=rx-48,rx=getchar();return ra;
 30 }
 31  
 32  
 33 inline void insert(int a,int b){
 34     e[++tot].too=b,e[tot].pre=last[a],last[a]=tot;
 35     e[++tot].too=a,e[tot].pre=last[b],last[b]=tot;
 36 }
 37 inline void ins(int a,int b){
 38 //  printf("   %d-->%d\n",a,b);
 39     e1[++tot1].too=b,e1[tot1].dis=dep[b]-dep[a],e1[tot1].pre=last1[a],last1[a]=tot1;
 40 }
 41  
 42  
 43 void dfs1(int x){
 44     size[x]=1;
 45     for(int i=last[x];i;i=e[i].pre)
 46         if(e[i].too!=fa[x])
 47             fa[e[i].too]=x,
 48             dfs1(e[i].too),
 49             size[x]+=size[e[i].too];
 50 }
 51 void dfs2(int x,int chain){
 52     bel[x]=chain,dfn[x]=++tim,dep[x]=dep[fa[x]]+1;int mxpos=0,i,to;
 53     for(to=e[i=last[x]].too;i;to=e[i=e[i].pre].too)
 54         if(to!=fa[x]&&size[to]>size[mxpos])mxpos=to;
 55     if(!mxpos)return;
 56     dfs2(mxpos,chain);
 57     for(to=e[i=last[x]].too;i;to=e[i=e[i].pre].too)
 58         if(to!=fa[x]&&to!=mxpos)dfs2(to,to);
 59 }
 60 inline int getlca(int a,int b){
 61     if(dep[bel[a]]<dep[bel[b]])swap(a,b);
 62     while(bel[a]!=bel[b]){
 63         a=fa[bel[a]];
 64         if(dep[bel[a]]<dep[bel[b]])swap(a,b);
 65     }
 66     return dep[a]<dep[b]?a:b;
 67 }
 68  
 69  
 70 bool cmp(int a,int b){return dfn[a]<dfn[b];}
 71  
 72 inline int min(int a,int b){return a<b?a:b;}
 73 inline int max(int a,int b){return a>b?a:b;}
 74 void dp(int x){
 75     register int i,to;
 76     if(intree[x]==m)sz[x]=1,mn[x]=mx[x]=0;
 77         else sz[x]=0,mn[x]=inf,mx[x]=-inf;
 78     for(to=e1[i=last1[x]].too;i;to=e1[i=e1[i].pre].too){
 79         dp(to),sz[x]+=sz[to],mn[to]+=e1[i].dis,mx[to]+=e1[i].dis;
 80         anssum+=(ll)sz[to]*(K-sz[to])*e1[i].dis;
 81         if(mn[x]+mn[to]<ansmn)ansmn=mn[x]+mn[to];
 82         if(mx[to]+mx[x]>ansmx)ansmx=mx[x]+mx[to];
 83         if(mn[to]<mn[x])mn[x]=mn[to];if(mx[to]>mx[x])mx[x]=mx[to];
 84     }
 85     last1[x]=0;
 86 }
 87  
 88  
 89 char s[23];int len;
 90 inline void outll(ll x){
 91     if(!x){putchar(‘0‘);return;}
 92     for(len=0;x;s[++len]=x%10,x/=10);
 93     while(len)putchar(s[len--]+48);
 94 }
 95 inline void outint(int x){
 96     if(!x){putchar(‘0‘);return;}
 97     for(len=0;x;s[++len]=x%10,x/=10);
 98     while(len)putchar(s[len--]+48);
 99 }
100  
101  
102 int main(){
103     register int i;
104     n=read();
105     for(i=1;i<n;i++)a=read(),b=read(),insert(a,b);
106     dfs1(1),dfs2(1,1);
107 //  for(i=1;i<=n;i++)printf("  %d  %d\n",dep[i],dfn[i]);
108 //  for(i=1;i<n;i++)for(j=i+1;j<=n;j++)printf("  %d&&%d %d\n",i,j,getlca(i,j));
109     for(m=read();m;m--){
110         tot1=0;
111         for(K=read(),i=1;i<=K;i++)intree[poi[i]=read()]=m;
112         sort(poi+1,poi+1+K,cmp);top=1,st[1]=poi[1];
113 //      for(i=1;i<=K;i++)printf("  %d\n",poi[i]);
114         for(i=2;i<=K;i++){
115             lca=getlca(poi[i],st[top]);
116             while(dfn[lca]<dfn[st[top]]&&top)
117                 if(dfn[st[top-1]]<=dfn[lca]){
118                     ins(lca,st[top--]);
119                     if(st[top]!=lca)st[++top]=lca;
120                 }else ins(st[top-1],st[top]),top--;
121             st[++top]=poi[i];
122         }
123         while(top>1)ins(st[top-1],st[top]),top--;
124         ansmn=inf,ansmx=anssum=0,rt=st[1],
125         dp(rt);
126         outll(anssum),putchar(‘ ‘),outint(ansmn),putchar(‘ ‘),outint(ansmx),putchar(‘\n‘);
127     }
128     return 0;
129 }

View Code