虚树学习笔记

  1 #include<stdio.h>
  2 #include<string.h>
  3 #include<algorithm>
  4 #include<queue>
  5 #include<vector>
  6 #include<stack>
  7 using namespace std;
  8 const int MAXN=100006;
  9 /**
 10 @var d[v]    结点v的深度
 11 @var f[v][i] 结点v的第2^i个祖先
 12 @var id[v]   结点的dfs序
 13 @var e[v]    结点v在原树邻接表
 14 @var vt[v]   结点v在虚树上包含的邻接表
 15 @var que     要查询的结点的集合（虚树必须包含的结点）
 16 @var LN      log2(最大深度)的向下取值
 17 @var st      栈，用于构建ST表
 18 @var z       dfs序的计数器
 19 注意结点标号必须从1开始，因为0拿去当空结点了。
 20 */
 21 int d[MAXN],f[MAXN][18],id[MAXN],LN=0,z,st[MAXN];
 22 vector<int>e[MAXN],vt[MAXN],que;
 23 int init(int n)
 24 {
 25     z=0;
 26     memset(f,0,sizeof(f));
 27     for(int i=1; i<=n; i++)
 28         e[i].clear();
 29     d[0]=-1;
 30     LN=0;
 31 }
 32 void dfs(int v,int deep)
 33 {
 34     d[v]=deep;
 35     st[deep]=v;
 36     id[v]=++z;
 37     if((1<<LN)<deep)
 38         LN++;
 39     int i,j;
 40     for(i=1,j=0; i<=deep; i<<=1,j++)
 41     {
 42         f[v][j]=st[deep-i];
 43     }
 44     for(i=0; i<e[v].size(); i++)
 45     {
 46         if(e[v][i]!=f[v][0])
 47         {
 48             dfs(e[v][i],deep+1);
 49         }
 50     }
 51 }
 52 int lca(int x,int y)
 53 {
 54     if(d[x]<d[y])
 55         swap(x,y);
 56     int i;
 57     for(i=LN; i>=0; i--)
 58     {
 59         if(f[x][i]&&d[f[x][i]]>=d[y])
 60             x=f[x][i];
 61     }
 62     if(x==y)
 63         return x;
 64     for(i=LN; i>=0; i--)
 65     {
 66         if(f[x][i]>0&&f[x][i]!=f[y][i])
 67         {
 68             x=f[x][i];
 69             y=f[y][i];
 70         }
 71     }
 72     return f[x][0];
 73 }
 74 int cmp(const int &x,const int &y)
 75 {
 76     return id[x]<id[y];
 77 }
 78 int build(vector<int> &que)
 79 {
 80     int i,j,k,temp;
 81     sort(que.begin(),que.end(),cmp);
 82     stack<int>st;
 83     st.push(que[0]);
 84     vt[que[0]].clear();
 85     for(i=1; i<que.size(); i++)///每次有出栈的时候开始建边
 86     {
 87         k=lca(que[i],st.top());
 88         temp=0;
 89         while(!st.empty()&&lca(k,st.top())!=st.top())
 90         {
 91             if(temp)
 92                 vt[st.top()].push_back(temp);
 93             temp=st.top();
 94             st.pop();
 95         }
 96         if(st.empty()||st.top()!=k)
 97         {
 98             st.push(k);
 99             vt[k].clear();
100         }
101         if(temp)
102             vt[st.top()].push_back(temp);
103         st.push(que[i]);
104         vt[que[i]].clear();
105     }
106     temp=0;
107     while(!st.empty())
108     {
109         if(temp)
110             vt[st.top()].push_back(temp);
111         temp=st.top();
112         st.pop();
113     }
114     return temp;
115 }