CH Round#58 OrzCC杯noip模拟赛day2-树的问题

【背景】
lll6924 小时候特别喜欢树！这一天他在小时候的玩具中发现了一个本子，上面画着好多树，
这时 lll6924 突发奇想，yy 了一道水题分享给大家。
【描述】
有一个由 n 个点构成的无根树，把这 n 个点分别作为根节点，我们可以造出 n 个有根树。有若
干询问，每次询问是两个点 a 和 b(a≠b)，我们想知道在这 n 个有根树中，有多少个树满足 a 是 b
的祖先或者 b 是 a 的祖先。
【输入格式】
输入有 n+q 行，第一行有两个用空格隔开的正整数 n 和 q，q 表示询问数，接下来的 n-1 行，
每一行有两个用空格隔开的正整数，分别表示一条边所连的两个点。接下来的 q 行，每行是一组询
问。输入保证能构成一棵树。
【输出格式】
输出有 q 行，对应每组询问。

 

 

比赛的时候看到这题还是很开心的。。貌似要比第一题来的显然很多？......

对于两个点a，b，我们可以找到它们的一条路径，现在我们把这条路径拉直来看，就像这样：

注意到以图中所有的红点为根a，b之间的路径都不会被拦腰截断，所以我们只需要计算红点的数量即可。。

任选一点建树，记录下每个点子树的节点数size。对于查询a，b，如果如果a，b之间不存在祖先关系，那么显然答案就是size[a]+size[b]；

若存在祖先关系，不妨设b是a的祖先，c是a的祖先中是b的子节点的那一个，那么可以得到答案就是n-size[c]+size[a]，size[c]-size[a]便是a，b路径上橙点的个数。。

现在问题变成了如何计算c点的问题，我们很容易会想到树上路径倍增，于是这题就可以在O（（n+q）logn）的时间内解决。。a，b是否是祖先关系可以在dfs过程中用时间戳搞定，对于每个节点记录一个访问它的时间和离开它的时间get和leave（访问第一个节点的时间是1，以后依次+1），那么若b是a的祖先显然会存在get[b]leave[a]。。

lll6924给出了一种更神的O(n+q)的算法，基本思想就是dfs离线做。。

∆

∆ 对于每个c结点的计算，我们可以离线做，对a与b，a是b的祖先，我们离线时dfs遍历整棵树，访问某个结点则将其加入栈中，离开时将其弹出，访问到b时，a一定在栈中，则c栈中为a顶上的那个结点。时间复杂度O(n+q)。

 

 

好了现在贴一下我的代码（倍增）：

<pre name="code" class="cpp">#include
#include
#include
#include
using namespace std;
const int maxn=200005;
struct edge{
 int e,next;
};
inti,j,n,q,s,e,x,y,nedge,cnt,a[maxn],d[maxn],u[maxn],size[maxn],first[maxn],last[maxn],z[maxn][19];
edge ed[maxn*2];
void addedge(int s,int e)
{
 ed[++nedge].e=e;
 ed[nedge].next=a[s];
 a[s]=nedge;
}
void dfs(int x,int k)
{
// cout<<k<<endl;
 u[x]=1;
 d[x]=k;
 size[x]=1;
 first[x]=++cnt;
 
 for (int j=a[x];j;j=ed[j].next)
 {
  edge i=ed[j];
  if (!u[i.e])
  {//cout<<j<<endl;
   z[i.e][0]=x;
   dfs(i.e,k+1);
   size[x]+=size[i.e];
  }
 }
 last[x]=++cnt;
}
int query(int x,int y)
{
 if (d[x]>d[y]) swap(x,y);
 if (!(first[x]last[y])) returnsize[x]+size[y];
 int j,c=y;
 for (j=18;j>=0;j--)
  if(z[c][j]&&d[z[c][j]]>d[x]) c=z[c][j];//printf("%d %d %d%d*",x,y,c,z[c][0]);
 return n-size[c]+size[y];
}
int main()
{
 freopen("tree.in","r",stdin);
 freopen("my.out","w",stdout);
 scanf("%d%d",&n,&q);
 memset(a,0,sizeof(a));
 nedge=0;
 memset(ed,0,sizeof(ed));
 for (i=1;i<=n-1;i++)
 {
  scanf("%d%d",&s,&e);
  addedge(s,e);
  addedge(e,s);
 }cnt=0;
 memset(z,0,sizeof(z));
 memset(u,0,sizeof(u));
 memset(d,0,sizeof(d));
 dfs(1,1);

 for (j=1;(1<<j)<=n;j++)
  for (i=1;i<=n;i++)
   z[i][j]=z[z[i][j-1]][j-1];

 for (i=1;i<=q;i++)
 {
  scanf("%d%d",&x,&y);
  printf("%d\n",query(x,y));
 }
 fclose(stdin);
 fclose(stdout);
}