bzoj2286 [Sdoi2011]消耗战

Description

在一场战争中,战场由n个岛屿和n-1个桥梁组成,保证每两个岛屿间有且仅有一条路径可达。现在,我军已经侦查到敌军的总部在编号为1的岛屿,而且他们已经没有足够多的能源维系战斗,我军胜利在望。已知在其他k个岛屿上有丰富能源,为了防止敌军获取能源,我军的任务是炸毁一些桥梁,使得敌军不能到达任何能源丰富的岛屿。由于不同桥梁的材质和结构不同,所以炸毁不同的桥梁有不同的代价,我军希望在满足目标的同时使得总代价最小。
侦查部门还发现,敌军有一台神秘机器。即使我军切断所有能源之后,他们也可以用那台机器。机器产生的效果不仅仅会修复所有我军炸毁的桥梁,而且会重新随机资源分布(但可以保证的是,资源不会分布到1号岛屿上)。不过侦查部门还发现了这台机器只能够使用m次,所以我们只需要把每次任务完成即可。

Input

第一行一个整数n,代表岛屿数量。

接下来n-1行,每行三个整数u,v,w,代表u号岛屿和v号岛屿由一条代价为c的桥梁直接相连,保证1<=u,v<=n且1<=c<=100000。

第n+1行,一个整数m,代表敌方机器能使用的次数。

接下来m行,每行一个整数ki,代表第i次后,有ki个岛屿资源丰富,接下来k个整数h1,h2,…hk,表示资源丰富岛屿的编号。

Output

输出有m行,分别代表每次任务的最小代价。

Sample Input

10
1 5 13
1 9 6
2 1 19
2 4 8
2 3 91
5 6 8
7 5 4
7 8 31
10 7 9
3
2 10 6
4 5 7 8 3
3 9 4 6

Sample Output

12
32
22

HINT

 对于100%的数据,2<=n<=250000,m>=1,sigma(ki)<=500000,1<=ki<=n-1

 

正解:虚树+树形dp。

闲得无聊学了下虚树。虚树就是说对于每个询问有k个点,满足sigma(k)为线性的问题,可以将每次询问的点提出来新建一棵树。对于k个结点,可以证明两两之间本质不同的lca最多只有k-1个。所以再把他们的lca全部加进去,再加一个公共的lca就行了(这道题是加1)。然后把这些结点按照dfs序从小到大排序,找出两个相邻点的lca。再将它们按照dfs序排序,开一个栈,如果栈顶元素是当前待加入结点的祖先,就连一条边,否则弹出栈顶,然后把当前结点加入栈中。这样,我们就能成功地建立一棵虚树了。这题dp还是挺容易的,然而因为邻接表清零出问题了然后调了半天。。

 

 1 //It is made by wfj_2048~
 2 #include <algorithm>
 3 #include <iostream>
 4 #include <cstring>
 5 #include <cstdlib>
 6 #include <cstdio>
 7 #include <vector>
 8 #include <cmath>
 9 #include <queue>
10 #include <stack>
11 #include <map>
12 #include <set>
13 #define inf (1e17)
14 #define N (1000010)
15 #define il inline
16 #define RG  
17 #define ll long long
18 #define File(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)
19 
20 using namespace std;
21 
22 struct edge{ int nt,to,dis; }g[N];
23 
24 int head[N],size[N],dfn[N],ed[N],dep[N],top[N],fa[N],son[N],a[N],st[N],vi[N],n,q,num,cnt;
25 ll dis[N];
26 
27 il int gi(){
28     RG int x=0,q=1; RG char ch=getchar(); while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
29     if (ch=='-') q=-1,ch=getchar(); while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar(); return q*x;
30 }
31 
32 il void insert(RG int from,RG int to,RG int dis){ g[++num]=(edge){head[from],to,dis},head[from]=num; return; }
33 
34 il int cmp(const int &a,const int &b){ return dfn[a]<dfn[b]; }
35 
36 il void dfs1(RG int x,RG int p){
37     dfn[x]=++cnt,dep[x]=dep[p]+1,fa[x]=p,size[x]=1; RG int v;
38     for (RG int i=head[x];i;i=g[i].nt){
39     v=g[i].to; if (v==p) continue;
40     dis[v]=min(dis[x],(ll)g[i].dis);
41     dfs1(v,x); size[x]+=size[v];
42     if (size[son[x]]<=size[v]) son[x]=v;
43     }
44     ed[x]=cnt; return;
45 }
46 
47 il void dfs2(RG int x,RG int p,RG int a){
48     top[x]=a; if (son[x]) dfs2(son[x],x,a);
49     for (RG int i=head[x];i;i=g[i].nt){
50     RG int v=g[i].to;
51     if (v==p || v==son[x]) continue;
52     dfs2(v,x,v);
53     }
54     head[x]=0; return;
55 }
56 
57 il int lca(RG int u,RG int v){
58     while (top[u]!=top[v]){
59     if (dep[top[u]]<dep[top[v]]) swap(u,v);
60     u=fa[top[u]];
61     }
62     return dep[u]<dep[v] ? u : v;
63 }
64 
65 il ll dp(RG int x){
66     if (vi[x]) return dis[x]; RG ll res=0;
67     for (RG int i=head[x];i;i=g[i].nt) res+=dp(g[i].to);
68     return min(res,dis[x]);
69 }
70 
71 il void work(){
72     n=gi(); RG int u,v,w; for (RG int i=1;i<n;++i) u=gi(),v=gi(),w=gi(),insert(u,v,w),insert(v,u,w);
73     dis[1]=inf,dfs1(1,0),dfs2(1,0,1); q=gi();
74     while(q--){
75     RG int k=gi(),M=k; for (RG int i=1;i<=k;++i) vi[a[i]=gi()]=1; sort(a+1,a+k+1,cmp);
76     for (RG int i=2;i<=k;++i) if (ed[a[i-1]]<dfn[a[i]]) a[++M]=lca(a[i-1],a[i]); a[++M]=1;
77     sort(a+1,a+M+1,cmp); M=unique(a+1,a+M+1)-a-1; RG int top=1; st[top]=a[1],num=0;
78     for (RG int i=2;i<=M;++i){
79         while (top && ed[st[top]]<dfn[a[i]]) top--;
80         insert(st[top],a[i],0),st[++top]=a[i];
81     }
82     printf("%lld\n",dp(1)); for (RG int i=1;i<=M;++i) vi[a[i]]=head[a[i]]=0;
83     }
84     return;
85 }
86 
87 int main(){
88     File("war");
89     work();
90     return 0;
91 }

 

posted @ 2017-02-19 08:46  wfj_2048  阅读(153)  评论(0编辑  收藏  举报