2018年第十届ACMICPC四川省大学程序设计竞赛补题

2018年第十届ACMICPC四川省大学程序设计竞赛补题

A:

咕咕，明天写

D:

题意：
在一棵树上存在m个点对，现在需要选一个点作为根节点，使得这m个点对中，每对点对一个节点为另一个节点的父节点，求根节点有多少种选法

思路：
画图可发现，首先以任意节点为根节点，对于一个点对，若其两个点不在一条链上，则这两个点对的所有子树的节点 +1 ；若在一条链上，则先把树上全部的节点+1，再把父亲到这个儿子的子树(不包括父亲节点) -1 再把儿子的子树+1；最后看节点权值为m的点有多少个，即为答案。
考虑树上差分维护权值，考虑dfs序方便处理

#include<bits/stdc++.h>
using namespace std;
const int maxn = 4e5+5;
int n,m,t,l[maxn],r[maxn],deep[maxn],f[maxn][20],val[maxn],ans,tim,lg[maxn];
vector<int>mp[maxn];
void dfs(int x,int fa){
	l[x]=++tim;//记录当前节点第一次访问的编号，及时间节点，dfs序 
	f[x][0]=fa;deep[x]=deep[fa]+1;
    for(int i=0;i<mp[x].size();i++){
		int y=mp[x][i];
		if(y==fa)continue;
		dfs(y,x);
	}    	
	r[x]=tim;//记录当前节点最后一次访问的编号，l[x]---r[x]表示x及x的子树的编号，为递增+1的数列 
}
int LCA(int x,int y){
	if(deep[x]<deep[y])swap(x,y);//让x深度大于y 
	while(deep[x]>deep[y]){
		x=f[x][lg[deep[x]-deep[y]]-1];
	}
	if(x==y)return x;
	for(int k=lg[deep[x]]-1;k>=0;k--)
		if(f[x][k]!=f[y][k])
		   x=f[x][k],y=f[y][k];
    return f[x][0];
}
void init(){
	memset(val,0,sizeof(val));
	memset(r,0,sizeof(r));
	memset(l,0,sizeof(l));
	for(int i=1;i<=n;i++)mp[i].clear();
	ans=tim=0;
}
int main(){
	for(int i=1;i<=n;i++)//预先算出log_2(i)+1的值
	    lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	scanf("%d",&t);
	while(t--){
		scanf("%d%d",&n,&m);
		init();
		for(int i=1,x,y;i<n;i++){
			scanf("%d%d",&x,&y);
			mp[x].push_back(y);
			mp[y].push_back(x);
		}
		dfs(1,0);
		for(int i=1,x,y;i<=m;i++){
			scanf("%d%d",&x,&y);
			int lca=LCA(x,y);
			if(deep[x]>deep[y])swap(x,y);
			if(lca==x){//在一条链上 
				val[1]++;
				val[l[x]]--,val[r[x]+1]++;
				val[l[y]]++,val[r[y]+1]--;
			}
			else{//不在一条链上 
				val[l[x]]++,val[r[x]+1]--;
				val[l[y]]++,val[r[y]+1]--;
			}
		}
		for(int i=1;i<=n;i++){
			val[i]+=val[i-1];
			if(val[i]==m)ans++;
		}
		printf("%d\n",ans);
	} 
}