码迷,mamicode.com
首页 > 其他好文 > 详细

【模板】虚树

时间:2019-12-13 23:55:51      阅读:159      评论:0      收藏:0      [点我收藏+]

标签:包含   tor   amp   清空   这一   wap   else   如何   printf   

核心思想:

(听名字高大上,实际上没什么东西……虚树的题主要难在如何操作虚树)

给出$k$个关键点,我们要建出一棵只包含这些关键点和他们$lca$的点数最少的树,以实现$dp$等操作。

标志性的数据范围是$\sum{k}\leq 10^{5}$之类的。

 

建树方法:

1.将所有关键点按$dfs$序排序。

2.开一个栈表示根到当前点的虚树路径,并把根丢进去。

3.对于每一个关键点$u$:

若栈中只有根这一个元素,则把$u$丢进去。

否则,我们需要弹出栈中所有不在根到$u$路径上的点。

我们用$lca$表示$lca(u,s[top])$,由于关键点是按照$dfs$序排序的,那么栈中所有$dep$大于$dep[lca]$的点都不在这条路径上。(容易验证它们一定在$lca$的另外一颗子树上)

于是一直弹到最后一个不满足要求的点,每弹出一个点之后就在它和栈顶间连一条边。

由于我们要把$lca$丢进栈里,所以最后一个点应当与$lca$连边,然后弹掉。

然后把$lca$和$u$丢进栈里。容易验证此时栈中的点一定形成一条根到$u$的路径。

4.最后把栈里所有点弹掉,每弹出一个点之后就在它和栈顶间连一条边。

没了。丢进栈里的时候注意判重。

 

复杂度分析:

每加一个点最多产生一个$lca$,那么虚树中的总点数是$O(2k)$的。

清空数组的时候千万要计算好复杂度。

 

代码(CF613D Kingdom and its Cities):

技术图片
#include<bits/stdc++.h>
#define maxn 200005
#define maxm 500005
#define inf 0x7fffffff
#define ll long long
#define debug(x) cerr<<#x<<": "<<x<<endl
#define fgx cerr<<"--------------"<<endl
#define dgx cerr<<"=============="<<endl

using namespace std;
int f[maxn][20],dep[maxn],dfn[maxn],tot;
int hd[maxn],to[maxn<<1],nxt[maxn<<1],cnt;
int st[maxn],po[maxn],vis[maxn],ans,flag; 
vector<int> vc[maxn];

inline int read(){
    int x=0,f=1; char c=getchar();
    for(;!isdigit(c);c=getchar()) if(c==-) f=-1;
    for(;isdigit(c);c=getchar()) x=x*10+c-0;
    return x*f;
}

inline bool cmp(int a,int b){return dfn[a]<dfn[b];}
inline void add(int a,int b){vc[a].push_back(b);}

inline void addedge(int u,int v){
    to[++cnt]=v,nxt[cnt]=hd[u],hd[u]=cnt;
    to[++cnt]=u,nxt[cnt]=hd[v],hd[v]=cnt;
}

inline void dfs(int u,int fa){
    dfn[u]=++tot,dep[u]=dep[fa]+1,f[u][0]=fa;
    for(int i=1;i<20;i++) f[u][i]=f[f[u][i-1]][i-1];
    for(int i=hd[u];i;i=nxt[i]) if(to[i]!=fa) dfs(to[i],u);
}

inline int lca(int u,int v){
    if(dep[u]<dep[v]) swap(u,v);
    for(int i=19;i>=0;i--)
        if(dep[f[u][i]]>=dep[v]) u=f[u][i];
    if(u==v) return u;
    for(int i=19;i>=0;i--)
        if(f[u][i]!=f[v][i]) 
            u=f[u][i],v=f[v][i];
    return f[u][0];
}

inline int solve(int u){
    int num=0;
    for(int i=0;i<vc[u].size();i++){
        int v=vc[u][i]; num+=solve(v);
        if(vis[v]&&vis[u]&&dep[v]==dep[u]+1) flag=1;
    }
    vc[u].clear();
    if(vis[u]){ans+=num;return 1;}
    else{
        if(num<=1) return num;
        else {ans++;return 0;}
    }
}

int main(){
    int n=read();
    for(int i=1;i<=n-1;i++)
        addedge(read(),read());
    dfs(1,0);
    int Q=read();
    while(Q--){
        int k=read(); ans=0,flag=0;
        for(int i=1;i<=k;i++) po[i]=read(),vis[po[i]]=1;
        sort(po+1,po+1+k,cmp);
        st[0]=0,st[++st[0]]=1;
        for(int i=1;i<=k;i++){
            if(st[0]==1){if(po[i]!=1)st[++st[0]]=po[i];continue;}
            int lt=lca(po[i],st[st[0]]);
            while(st[0]>1 && dep[lt]<dep[st[st[0]-1]])
                add(st[st[0]-1],st[st[0]]),st[0]--;
            if(dep[lt]<dep[st[st[0]]]) add(lt,st[st[0]]),st[0]--;
            if(st[st[0]]!=lt) st[++st[0]]=lt; st[++st[0]]=po[i];
        }
        while(st[0]>1) add(st[st[0]-1],st[st[0]]),st[0]--;
        solve(1),printf("%d\n",(flag)?-1:ans);
        for(int i=1;i<=k;i++) vis[po[i]]=0;
    }
    return 0;
}
虚树

【模板】虚树

标签:包含   tor   amp   清空   这一   wap   else   如何   printf   

原文地址:https://www.cnblogs.com/YSFAC/p/12037564.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!