学一下虚树
单次dp是O(N)的,总复杂度O(NM)
但实际用不到O(N)个节点,只有O(K)个节点是有用的,若能建出一棵O(K)个节点的树,那么复杂度会变成O(∑K)
把给定节点按照dfs序排序,维护一个栈,栈中的节点是根到栈顶元素的链上的节点
加入节点x,设栈顶元素为p,t=LCA(p,x)
若t==p,则直接加入x
若t!=p,则找到栈中最后一个深度小于t的节点y,在y到p之间的节点中,不会再有中间节点插入了,将这些点删掉,对应从st[top-1]向st[top]连边,最后将点x加入
比较有趣的东西
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<vector>
#define maxn 250010
using namespace std;
int head[maxn],to[2*maxn],next[2*maxn],len[2*maxn];
long long dp[maxn],val[maxn];
vector<int> v[maxn];
bool flag[maxn];
int in[maxn],st[maxn],dep[maxn],fa[20][maxn],a[maxn],q[maxn];
int n,m,tot,num,top,T,k,now;
bool cmp(int x,int y) {return in[x]<in[y];}
void addedge(int x,int y,int z)
{
num++;to[num]=y;len[num]=z;next[num]=head[x];head[x]=num;
}
void dfs(int x)
{
in[x]=++tot;
for (int p=head[x];p;p=next[p])
if (to[p]!=fa[0][x]) fa[0][to[p]]=x,dep[to[p]]=dep[x]+1,val[to[p]]=min(val[x],(long long)len[p]),dfs(to[p]);
}
void DP(int x)
{
if (flag[x]) dp[x]=val[x];
else
{
dp[x]=0;
for (int i=0;i<v[x].size();i++) DP(v[x][i]),dp[x]+=dp[v[x][i]];
dp[x]=min(dp[x],val[x]);
}
}
void DFS(int x)
{
for (int i=0;i<v[x].size();i++) DFS(v[x][i]);
v[x].clear();flag[x]=0;
}
int go_up(int x,int d)
{
for (int i=18;i>=0;i--)
if (d&(1<<i)) x=fa[i][x];
return x;
}
int LCA(int x,int y)
{
if (dep[x]>dep[y]) x=go_up(x,dep[x]-dep[y]);
else y=go_up(y,dep[y]-dep[x]);
if (x==y) return x;
for (int i=18;i>=0;i--)
if (fa[i][x]!=fa[i][y]) x=fa[i][x],y=fa[i][y];
return fa[0][x];
}
int main()
{
scanf("%d",&n);
for (int i=1;i<n;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
addedge(x,y,z);addedge(y,x,z);
}
val[1]=1e18;dfs(1);
for (int j=1;j<=18;j++)
for (int i=1;i<=n;i++)
fa[j][i]=fa[j-1][fa[j-1][i]];
scanf("%d",&T);
while (T--)
{
int x;
top=1;st[top]=1;
scanf("%d",&k);
for (int i=1;i<=k;i++) scanf("%d",&a[i]),flag[a[i]]=1;
sort(a+1,a+k+1,cmp);
for (int i=1;i<=k;i++)
{
int lca=LCA(a[i],st[top]);
while (dep[st[top]]>dep[lca])
if (dep[st[top-1]]<dep[lca]) v[lca].push_back(st[top]),st[top]=lca;
else v[st[top-1]].push_back(st[top]),top--;
if (st[top]!=a[i]) st[++top]=a[i];
}
while (top>1) v[st[top-1]].push_back(st[top]),top--;
DP(1);
DFS(1);
printf("%lld\n",dp[1]);
}
return 0;
}