传送门:牛客
题目描述:
题目较长,此处省略
输入:
7
1 2
1 3
2 4
2 5
3 6
3 7
2
2 6 7
1 4
3
5 1 2
1 1 1
5 2 1 2
输出:
2
1
1
一道比较复杂的树题.需要一些复杂的讨论以及LCA知识
对于LCA,可以使用树链剖分进行解决
然后我们看一下题目,我们会发现有这样一个简单的结论,那就是我们的两个点<u,v>被占领的时候,那么他们的祖先 L C A ( u , v ) LCA(u,v) LCA(u,v)也必然是被占领的.(这个很好理解,因为我们的,uv肯定是穿过他们的祖先节点的).所以我们可以预处理出每一个帮派的 L C A LCA LCA
同样的对于我们后来的联盟,我们可以求出这些联盟共同的 L C A LCA LCA
那么对于一个首都 v v v来说,此时的 v v v有以下几种情况:
- 当 v v v不是我们 L C A LCA LCA的子节点.因为我们的被占领的地方肯定是一个联通图,所以此时我们的最短距离显然是 v v v到第一个联通图上的一点,那么此时的那一个点就是我们的LCA(可以自己画图理解理解)
- 我们的
v
v
v是我们的
L
C
A
LCA
LCA的子节点.
此时我们可以预处理出所有节点的 d f s dfs dfs序,那么对于此时我们的 v v v来说,又有这几种情况:.
1.比v d f s dfs dfs序小的最近节点的是v的父亲,此时显然最短距离就是这个点与v的距离,注意肯定是一个联通图,而我们的v处于这个位置,所以在v的节点中不可能存在一个点是被占领的
2.比v d f s dfs dfs序小的最近节点是 v v v的兄弟,设为u1,此时我们需要看一下 d f s dfs dfs大的最近节点(设为u2)的情况,根据dfs遍历的性质,dfs序大的节点不可能是v的父亲了,此时必然是v的兄弟.所以此时我们的LCA(u1,v)和LCA(u2,v)的一个点到v的距离必然有一个是最短的.这里感觉很难说明清楚,因为用到了dfs遍历的性质,感觉使用到了dfs遍历中的夹逼性质,但是通过画图还是很好理解的
对于树上两点之间距离,可以使用到根节点的距离来进行解决
d
i
s
[
u
]
[
v
]
=
d
i
s
t
[
u
]
+
d
i
s
t
[
v
]
−
2
∗
d
i
s
t
[
L
C
A
(
u
,
v
)
]
dis[u][v]=dist[u]+dist[v]-2*dist[LCA(u,v)]
dis[u][v]=dist[u]+dist[v]−2∗dist[LCA(u,v)]
下面是具体的代码部分:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
ll x=0,w=1;char ch=getchar();
for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
return x*w;
}
#define maxn 1000000
const double eps=1e-8;
#define int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
int fa[maxn],Size[maxn],max_son[maxn],dep[maxn];
vector<int>edge[maxn];
void dfs1(int u,int per_u) {
Size[u]=1;
for(int i=0;i<edge[u].size();i++) {
int v=edge[u][i];
if(v==per_u) continue;
dep[v]=dep[u]+1;
dfs1(v,u);
fa[v]=u;Size[u]+=Size[v];
if(Size[v]>Size[max_son[u]]) {
max_son[u]=v;
}
}
}
int top[maxn];
void dfs2(int u,int t) {
top[u]=t;
if(!max_son[u]) return ;
dfs2(max_son[u],t);
for(int i=0;i<edge[u].size();i++) {
int v=edge[u][i];
if(v==fa[u]||v==max_son[u]) continue;
dfs2(v,v);
}
}
int LCA(int u,int v) {
while(top[u]!=top[v]) {
if(dep[top[u]]<dep[top[v]]) swap(u,v);
u=fa[top[u]];
}
if(dep[u]>dep[v]) swap(u,v);
return u;
}
/*============================================================*/
int n,k;int c[maxn],lc[maxn],t[maxn];//lc记录每一个帮派管的点的LCA
vector<int>g[maxn];//记录每一个帮派管的点的dfs序
int dfsn[maxn],dfscnt=0,pos[maxn];//记录dfs序
void dfs(int u,int per_u) {
dfsn[u]=++dfscnt;
pos[dfscnt]=u;
for(int i=0;i<edge[u].size();i++) {
int v=edge[u][i];
if(v==per_u) continue;
dfs(v,u);
}
}
int get_dist(int u,int v) {
return dep[u]+dep[v]-2*dep[LCA(u,v)];
}
/*============================================================*/
int main() {
n=read();
for(int i=1;i<=n-1;i++) {
int u=read(),v=read();
edge[u].push_back(v);
edge[v].push_back(u);
}
dfs1(1,0);dfs2(1,1);dfs(1,1);
int k=read();
for(int i=1;i<=k;i++) {
int num=read();
for(int j=1;j<=num;j++) {
c[j]=read();
if(j==1) lc[i]=c[j];
else lc[i]=LCA(lc[i],c[j]);
g[i].push_back(dfsn[c[j]]);
}
sort(g[i].begin(),g[i].end());
}
int q=read();
for(int i=1;i<=q;i++) {
int v=read(),num=read();
for(int j=1;j<=num;j++) t[j]=read();
int lca=lc[t[1]];
for(int j=2;j<=num;j++) lca=LCA(lca,lc[t[j]]);
if(LCA(lca,v)!=lca) {
printf("%d\n",get_dist(lca,v));
continue;
}
int ans=int_INF;
for(int j=1;j<=num;j++) {
int tmp=t[j];
auto p=lower_bound(g[tmp].begin(),g[tmp].end(),dfsn[v]);
if(p!=g[tmp].end()) {
ans=min(ans,get_dist(v,LCA(v,pos[*p])));
}
if(p!=g[tmp].begin()) {
ans=min(ans,get_dist(v,LCA(v,pos[*prev(p)])));
}
}
cout<<ans<<endl;
}
return 0;
}