算法:树上差分+倍增LCA
难度:NOIP-
好吧,这道题就是 树上差分模板&&倍增LCA模板。结束了...
代码如下:
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <queue>
#include <algorithm>
#define ll long long
#define N 100005
using namespace std;
struct node
{
int next;
int to;
}edge[N<<1];
int head[N],dis[N],vis[N];
int cnt=1;
int dep[N],f[N][35];
void init()
{
memset(head,-1,sizeof(head));
cnt=1;
}
void add(int u,int v)
{
edge[cnt].next=head[u];
edge[cnt].to=v;
head[u]=cnt++;
}
void dfs(int rx,int fx,int de)
{
dep[rx]=de;
f[rx][0]=fx;
for(int i = head[rx];i != -1;i=edge[i].next)
{
int to=edge[i].to;
if(to==fx) continue;
dfs(to,rx,de+1);
}
}
int LCA(int x,int y)
{
if(dep[x]>dep[y])
{
swap(x,y);
}
for(int i = 25;i >= 0;i--)
{
if(dep[f[y][i]]>=dep[x])
{
y=f[y][i];
}
}
if(x==y) return x;
int ret;
for(int i = 25;i >= 0;i--)
{
if(f[y][i]!=f[x][i])
{
x=f[x][i];
y=f[y][i];
}else
{
ret=f[x][i];
}
}
return f[x][0];
}
int ans=0;
void get_ans(int rt,int fa)
{
for (int i = head[rt];i != -1;i=edge[i].next)
{
int to=edge[i].to;
if (to==fa) continue;
get_ans(to,rt);
vis[rt]+=vis[to];
}
ans=max(ans,vis[rt]);
}
int main()
{
init();
int n,m;
scanf("%d%d",&n,&m);
for(int i = 1;i < n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
add(x,y);
add(y,x);
}
dfs(1,1,1);//刚刚把dfs的位置放错了,居然苟过了样例 ......
for(int i = 1;i <= 25;i++)
{
for(int j = 1;j <= n;j++)
{
f[j][i]=f[f[j][i-1]][i-1];
}
}
//刚刚写到了这里...
while(m--)
{
int a,b;
scanf("%d%d",&a,&b);
int lac=LCA(a,b);
++vis[a],++vis[b],--vis[lac],--vis[f[lac][0]];
}
get_ans(1,-1);
printf("%d\n",ans);
return 0 ;
}