bzoj 3611: [Heoi2014]大工程 && bzoj 2286: [Sdoi2011消耗战

放波建虚树的模板。

大概是用一个栈维护根节点到当前关键点的一条链，把其他深度大于lca的都弹出去。

每次做完记得复原。

还有sort的时候一定要加cmp！！！

bzoj 3611

 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #define N 1000005
 #define ll long long
 #define inf 0x3f3f3f3f
 using namespace std;
 inline int read()
 {
     int p=;char c=getchar();
     while(c<''||c>'')c=getchar();
     while(c>=''&&c<='')p=p*+c-'',c=getchar();
     return p;
 }
 int n;
 int dfn[N];
 int fa[N][];
 int dep[N],z;
 int head[N],ver[N*],nxt[N*],tot;
 void add(int a,int b)
 {
     tot++;nxt[tot]=head[a];head[a]=tot;ver[tot]=b;
 }
 void dfs(int x,int f)
 {
     dfn[x]=++z;
     for(int i=head[x];i;i=nxt[i])
     {
         if(ver[i]==f)continue;
         dep[ver[i]]=dep[x]+;
         fa[ver[i]][]=x;
         dfs(ver[i],x);
     }
     return ;
 }
 void lca()
 {
     for(int i=;i<=;i++)
     {
         for(int j=;j<=n;j++)
         {
             fa[j][i]=fa[fa[j][i-]][i-];
         }
     }return ;
 }
 int lca(int x,int y)
 {
     if(dep[x]<dep[y])swap(x,y);
     for(int i=;i>=;i--)if(dep[fa[x][i]]>=dep[y])x=fa[x][i];
     if(x==y)return x;
     for(int i=;i>=;i--)
     {
         if(fa[x][i]!=fa[y][i])
         {
             x=fa[x][i];y=fa[y][i];
         }
     }
     return fa[x][];
 }
 bool cmp(int x,int y)
 {
     return dfn[x]<dfn[y];
 }
 int now[N],cnt;
 int st[N],top;
 int dian[N],num;
 int vis[N];
 void build()
 {
     tot=;
     st[]=;top=;
     for(int i=;i<=cnt;i++)
     {
         if(now[i]==)continue;
         int v=now[i];
         int la=lca(st[top],v);
         if(st[top]!=la)
         {
             while(top>=&&dep[st[top-]]>dep[la])
             {
                 add(st[top-],st[top]);top--;
             }
             add(la,st[top]);top--;
             if(st[top]!=la)st[++top]=la;
         }
         st[++top]=v;
     }
     while(top>=)add(st[top-],st[top]),top--;
 }
 ll ans1;
 int ans2,ans3;
 int size[N];
 int mx1[N],mx2[N],mn1[N],mn2[N];
 void dfs(int x)
 {
     dian[++num]=x;
     if(vis[x])size[x]=;
     else size[x]=;
     mx1[x]=mx2[x]=;
     mn1[x]=mn2[x]=inf;
     for(int i=head[x];i;i=nxt[i])
     {
         int quan=dep[ver[i]]-dep[x];
         dfs(ver[i]);
         if(vis[ver[i]])mn1[ver[i]]=;
         if(mn1[ver[i]]+quan<mn1[x])
         {
             mn2[x]=mn1[x];
             mn1[x]=mn1[ver[i]]+quan;
         }
         else if(mn1[ver[i]]+quan<mn2[x])mn2[x]=mn1[ver[i]]+quan;
         ans1+=(long long)size[ver[i]]*(cnt-size[ver[i]])*quan;
         size[x]+=size[ver[i]];
         if(mx1[ver[i]]+quan>mx1[x])
         {
             mx2[x]=mx1[x];
             mx1[x]=mx1[ver[i]]+quan;
         }
         else if(mx1[ver[i]]+quan>mx2[x])mx2[x]=mx1[ver[i]]+quan;
     }
     if(vis[x])ans2=min(ans2,mn1[x]);
     else ans2=min(ans2,mn1[x]+mn2[x]);
     if(mx2[x])ans3=max(ans3,mx1[x]+mx2[x]);
     else if(vis[x])ans3=max(ans3,mx1[x]);
 }
 int main()
 {
     n=read();
     int t1,t2,t3,t4;
     for(int i=;i<n;i++)
     {
         t1=read();t2=read();
         add(t1,t2);add(t2,t1);
     }
     dep[]=;
     dfs(,-);
     lca();
     int q,k;
     q=read();
     memset(head,,sizeof(head));
     for(int i=;i<=q;i++)
     {
         ans1=;ans2=inf;ans3=;
         k=read();cnt=k;num=;
         for(int j=;j<=k;j++)now[j]=read(),vis[now[j]]=;
         sort(now+,now+k+,cmp);
         build();dfs();
         for(int j=;j<=cnt;j++)vis[now[j]]=;
         for(int j=;j<=num;j++)head[dian[j]]=;num=;
         printf("%lld %d %d\n",ans1,ans2,ans3);
     }
     return ;
 }

bzoj2286

 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #define N 500005
 #define inf 0x3f3f3f3f
 using namespace std;
 int n;
 int head[N],ver[N*],nxt[N*],quan[N*],tot;
 inline int read()
 {
     int p=;char c=getchar();
     while(c<''||c>'')c=getchar();
     while(c>=''&&c<='')p=p*+c-'',c=getchar();
     return p;
 }
 void add(int a,int b,int c)
 {
     tot++;nxt[tot]=head[a];head[a]=tot;ver[tot]=b;quan[tot]=c;
 }
 int fa[N][],mn[N][],dep[N];
 void lca()
 {
     for(int i=;i<=;i++)
     {
         for(int j=;j<=n;j++)
         {
             fa[j][i]=fa[fa[j][i-]][i-];
             mn[j][i]=min(mn[j][i-],mn[fa[j][i-]][i-]);
         }
     }
     return ;
 }
 int lca(int x,int y)
 {
     if(dep[x]<dep[y])swap(x,y);
     for(int i=;i>=;i--)if(dep[fa[x][i]]>=dep[y])x=fa[x][i];
     if(x==y)return x;
     for(int i=;i>=;i--)
     {
         if(fa[x][i]!=fa[y][i])
         {
             x=fa[x][i];y=fa[y][i];
         }
     }
     return fa[x][];
 }
 int z,dfn[N];
 void dfs(int x,int f)
 {
     dfn[x]=++z;
     for(int i=head[x];i;i=nxt[i])
     {
         if(ver[i]==f)continue;
         dep[ver[i]]=dep[x]+;
         fa[ver[i]][]=x;
         mn[ver[i]][]=quan[i];
         dfs(ver[i],x);
     }return ;
 }
 int now[N],cnt;
 int vis[N],st[N],top;
 int dian[N],num;
 int qur(int x,int y)
 {
     int ans=inf;
     for(int i=;i>=;i--)
     {
         if(dep[fa[x][i]]>=dep[y])
         {
             ans=min(ans,mn[x][i]);
             x=fa[x][i];
         }
     }
     return ans;
 }
 void build()
 {
     tot=;
     st[]=;top=;
     for(int i=;i<=cnt;i++)
     {
         if(now[i]==)continue;
         int v=now[i];
         int la=lca(st[top],v);
         if(la!=st[top])
         {
             while(top>=&&dep[st[top-]]>dep[la])
             {
                 add(st[top-],st[top],qur(st[top],st[top-])),top--;
             }
             add(la,st[top],qur(st[top],la));top--;
             if(la!=st[top])st[++top]=la;
         }
         st[++top]=v;
     }
     while(top>=)
     {
         add(st[top-],st[top],qur(st[top],st[top-])),top--;
     }
     return ;
 }
 long long f[N];
 bool cmp(int x,int y)//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 {
     return dfn[x]<dfn[y];
 }
 void dp(int x)
 {
     dian[++num]=x;f[x]=;
     for(int i=head[x];i;i=nxt[i])
     {
         dp(ver[i]);
         if(vis[ver[i]])f[x]+=quan[i];
         else f[x]+=min((long long)quan[i],f[ver[i]]);
     }
     return ;
 }
 int main()
 {
     n=read();
     int t1,t2,t3,t4;
     for(int i=;i<n;i++)
     {
         t1=read();t2=read();t3=read();
         add(t1,t2,t3);add(t2,t1,t3);
     }
     mn[][]=inf;dep[]=;
     dfs(,-);
     lca();
     memset(head,,sizeof(head));
     int q,k;
     q=read();
     for(int i=;i<=q;i++)
     {
         k=read();cnt=k;num=;
         for(int j=;j<=k;j++)
         {
             now[j]=read();vis[now[j]]=;
         }
         sort(now+,now+k+,cmp);build();
         dp();
         printf("%lld\n",f[]);
         for(int j=;j<=k;j++)vis[now[j]]=;
         for(int j=;j<=num;j++)head[dian[j]]=;
     }
     return ;
 }